Skip to content

Commit 1c96d12

Browse files
docs(aws): add redirection to Optimum Neuron (#1851)
* add redirection * spelling * Update docs/sagemaker/source/dlcs/available.md Co-authored-by: Jeff Boudier <[email protected]> --------- Co-authored-by: Jeff Boudier <[email protected]>
1 parent f666b4a commit 1c96d12

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

docs/sagemaker/source/dlcs/available.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@ Below you can find a listing of all the Deep Learning Containers (DLCs) availabl
44

55
For each supported combination of use-case (training, inference), accelerator type (CPU, GPU, Neuron), and framework (PyTorch, TGI, TEI) containers are created.
66

7+
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).
8+
79
## Training
810

911
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.

0 commit comments

Comments
 (0)