AWS Deep Learning AMIs now come pre-installed with Open Neural Network Exchange (ONNX), an open source format for neural network computational graph supported by popular deep learning frameworks, including Apache MXNet, TensorFlow, PyTorch, Chainer, and Cognitive Toolkit (CNTK). ONNX gives developers the flexibility to migrate between frameworks. For example, developers can use PyTorch for prototyping, building and training their models, and then use ONNX to migrate their models to MXNet to leverage its scalability for inference. To learn more about using ONNX, see our blog post and tutorials.
