The AWS Deep Learning Containers are available today with the latest framework versions of TensorFlow 2.1.0 & 1.15.2, PyTorch 1.4.0, and MXNet 1.6.0 . The release includes the addition of Amazon SageMaker Python SDK in the containers, and updates to the Amazon SageMaker Experiments package. Amazon SageMaker Experiments is a feature in Amazon SageMaker that lets you organize, track, compare, and evaluate machine learning (ML) experiments and model versions. The TensorFlow 2.1.0 python3 training containers now also include SageMaker Debugger , which allow data scientists to save and inspect the model tensors during training jobs.