From a3a4440909ae0f4f28976598163d4e79d622054e Mon Sep 17 00:00:00 2001 From: peterjunpark Date: Fri, 6 Feb 2026 13:47:55 -0500 Subject: [PATCH] docs(jax-maxtext training): remove single-node for llama 3.1 405b --- .../rocm-for-ai/training/jax-maxtext-benchmark-models.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/data/how-to/rocm-for-ai/training/jax-maxtext-benchmark-models.yaml b/docs/data/how-to/rocm-for-ai/training/jax-maxtext-benchmark-models.yaml index a63eb5019..9180b9d31 100644 --- a/docs/data/how-to/rocm-for-ai/training/jax-maxtext-benchmark-models.yaml +++ b/docs/data/how-to/rocm-for-ai/training/jax-maxtext-benchmark-models.yaml @@ -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