Amazon SageMaker
Developer Guide

Using Apache MXNet with Amazon SageMaker

You can use Amazon SageMaker to train and deploy a model using custom MXNet code. The Amazon SageMaker Python SDK MXNet estimators and models and the Amazon SageMaker open-source MXNet container make writing a MXNet script and running it in Amazon SageMaker easier. For information about writing MXNet training scripts and using MXNet estimators and models with Amazon SageMaker, see https://github.com/aws/sagemaker-python-sdk#mxnet-sagemaker-estimators.

For information about MXNet versions supported by the Amazon SageMaker MXNet container, see https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/mxnet/README.rst.

For information about how to build and contribute to the Amazon SageMaker MXNet container, see the GitHub repository at https://github.com/aws/sagemaker-mxnet-container.