github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-01-08 22:28:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

docs: Add Primus (Megatron) training Docker documentation (#5218)

Browse Source
This commit is contained in:
Peter Park
2025-08-21 23:50:55 -04:00
committed by GitHub
parent 65ebbaa117
commit 98029db4ee
11 changed files with 1994 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/sphinx/_toc.yml.in
View File

@@ -44,8 +44,8 @@ subtrees:
title: Training
subtrees:
- entries:
- file: how-to/rocm-for-ai/training/benchmark-docker/megatron-lm.rst
title: Train a model with Megatron-LM
- file: how-to/rocm-for-ai/training/benchmark-docker/primus-megatron.rst
title: Train a model with Primus and Megatron-Core
- file: how-to/rocm-for-ai/training/benchmark-docker/pytorch-training.rst
title: Train a model with PyTorch
- file: how-to/rocm-for-ai/training/benchmark-docker/jax-maxtext.rst