github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-02-11 06:55:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

docs(jax-maxtext training): remove single-node for llama 3.1 405b (#5939)

Browse Source 
(cherry picked from commit a3a4440909)
This commit is contained in:
peterjunpark
2026-02-06 13:50:03 -05:00
committed by GitHub
parent b62e0546fd
commit 208443edec
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/data/how-to/rocm-for-ai/training/jax-maxtext-benchmark-models.yaml
View File

@@ -49,13 +49,13 @@ model_groups:
model_repo: Llama-3.1-70B
precision: bf16
doc_options: ["single-node"]
- model: Llama 3.1 405B
- model: Llama 3.1 405B (multi-node)
mad_tag: jax_maxtext_train_llama-3.1-405b
model_repo: Llama-3.1-405B
precision: bf16
multinode_config:
gfx950: env_scripts/gfx950_llama3_405b.yml
doc_options: ["single-node", "multi-node"]
doc_options: ["multi-node"]
- model: Llama 3.3 70B
mad_tag: jax_maxtext_train_llama-3.3-70b
model_repo: Llama-3.3-70B