Amazon SageMaker
Developer Guide

Use 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://sagemaker.readthedocs.io/en/stable/using_mxnet.html.

For API reference for the Amazon SageMaker Python SDK MXNet classes, see https://sagemaker.readthedocs.io/en/stable/sagemaker.mxnet.html.

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.