diff --git a/docs/sagemaker/source/dlcs/available.md b/docs/sagemaker/source/dlcs/available.md index 7b326a04c..c24fcf1bc 100644 --- a/docs/sagemaker/source/dlcs/available.md +++ b/docs/sagemaker/source/dlcs/available.md @@ -4,6 +4,8 @@ Below you can find a listing of all the Deep Learning Containers (DLCs) availabl For each supported combination of use-case (training, inference), accelerator type (CPU, GPU, Neuron), and framework (PyTorch, TGI, TEI) containers are created. +Neuron DLCs for training and inference on AWS Trainium and AWS Inferentia instances can be found in the [Optimum Neuron documentation](https://huggingface.co/docs/optimum-neuron/en/containers). + ## Training Pytorch Training DLC: For training, our DLCs are available for PyTorch via Transformers. They include support for training on GPUs and AWS AI chips with libraries such as TRL, Sentence Transformers, or Diffusers.