You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: docs/sagemaker/source/dlcs/available.md
+2Lines changed: 2 additions & 0 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -4,6 +4,8 @@ Below you can find a listing of all the Deep Learning Containers (DLCs) availabl
4
4
5
5
For each supported combination of use-case (training, inference), accelerator type (CPU, GPU, Neuron), and framework (PyTorch, TGI, TEI) containers are created.
6
6
7
+
For Neuron DLCs specificaly, you can browse the Optimum Neuron [documentation](https://huggingface.co/docs/optimum-neuron/en/containers).
8
+
7
9
## Training
8
10
9
11
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