Extend the training job `export_model` function to also export as ONNX (`model.onnx`). Also update `store_model_artifacts` accordingly. _Note: in the future, we want to create the ONNX model on-demand rather than by default_