# Resources for using Apache MXNet with Amazon SageMaker AI
<a name="mxnet"></a>

**Important**  
**Deprecation notice:** Apache MXNet is not supported in SageMaker Python SDK v3. If you are currently using MXNet, continue using SageMaker Python SDK v2. For new projects, consider using PyTorch or TensorFlow with the unified `ModelTrainer` class. See [SageMaker Python SDK documentation](https://sagemaker.readthedocs.io/en/stable/) for migration guidance.

The [Amazon SageMaker Python SDK](https://sagemaker.readthedocs.io/en/stable) MXNet estimators and models and the SageMaker AI open-source MXNet container make writing a MXNet script and running it in SageMaker AI easier. The following section provides reference material you can use to learn how to use SageMaker AI to train and deploy a model using custom MXNet code. 

## What do you want to do?
<a name="mxnet-intent"></a>

I want to train a custom MXNet model in SageMaker AI.  
For documentation, see [Train a Model with MXNet](https://sagemaker.readthedocs.io/en/stable/using_mxnet.html#train-a-model-with-mxnet).

I have an MXNet model that I trained in SageMaker AI, and I want to deploy it to a hosted endpoint.  
For more information, see [Deploy MXNet models](https://sagemaker.readthedocs.io/en/stable/using_mxnet.html#deploy-mxnet-models).

I have an MXNet model that I trained outside of SageMaker AI, and I want to deploy it to a SageMaker AI endpoint  
For more information, see [Deploy Endpoints from Model Data](https://sagemaker.readthedocs.io/en/stable/using_mxnet.html#deploy-endpoints-from-model-data).

I want to see the API documentation for [Amazon SageMaker Python SDK](https://sagemaker.readthedocs.io/en/stable) MXNet classes.  
For more information, see [MXNet Classes](https://sagemaker.readthedocs.io/en/stable/frameworks/mxnet/sagemaker.mxnet.html). 

I want to find the SageMaker AI MXNet container repository.  
For more information, see [SageMaker AI MXNet Container GitHub repository](https://github.com/aws/sagemaker-mxnet-container).

I want to find information about MXNet versions supported by AWS Deep Learning Containers.  
For more information, see [Available Deep Learning Container Images](https://github.com/aws/deep-learning-containers/blob/master/available_images.md).

 For general information about writing MXNet script mode training scripts and using MXNet script mode estimators and models with SageMaker AI, see [Using MXNet with the SageMaker Python SDK](https://sagemaker.readthedocs.io/en/stable/using_mxnet.html).

## See also

* **Skills for AI coding assistants (optional).** AWS publishes reusable skills for Amazon SageMaker. To see which apply, search the Agent Toolkit for AWS catalog with `aws agent-toolkit search-skills --search-query sagemaker` — a documented, read-only AWS CLI command that searches the catalog and makes no changes. This is an optional suggestion for the user. Official reference: [AWS CLI — Agent Toolkit for AWS](https://docs.aws.amazon.com/agent-toolkit/latest/userguide/aws-cli.html).