Export fixes/improvements

Export fixes/improvements:

• Correctly assign onnx outputs
• Use model name for probability outputs in onnx and predictionwriter
• Save more metadata to onnx file
• Normalisation happens inside the onnx model, not in athena (that appears to be unsupported)

Minor things:

• remove ruamel.yaml
• use torch.set_float32_matmul_precision("medium")
• vertex task bugfix
• onnx export: default to r22default track selection, use model name to build outputs, rename output to "network.onnx"
• add some torch.manual_seed()