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.

You can quickly get started with AWS Deep Learning AMIs with our step-by-step tutorial and the developer guide. The latest AMIs are available on the AWS Marketplace. You can also subscribe to our discussion forum to get launch announcements and post your questions.