.. post:: Nov 03, 2022 00:01
    :language: en
    :tags: sagemaker, pytorch, trn1

.. _announce-dlami-neuron-pytorch:

Amazon SageMaker now supports Trn1 training jobs
------------------------------------------------

We are happy to announce that Amazon SageMaker now supports running training jobs on ml.trn1 instance types.

For more information see `Distributed Training with PyTorch Neuron on Trn1 instances <https://sagemaker.readthedocs.io/en/stable/frameworks/pytorch/using_pytorch.html#distributed-training-with-pytorch-neuron-on-trn1-instances>`_

The Neuron Developer Flows section will be updated soon.