@@ -52,6 +52,7 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
5252| gpt_oss_20b | [ small_llm_moe_pretraining] ( https://github.com/mlcommons/training/tree/master/small_llm_moe_pretraining/primus ) | Primus | C4 | 20B |
5353| deepseekv3 | [ llm_moe_pretraining] ( https://github.com/mlcommons/training/tree/master/llm_moe_pretraining/nemo ) | NeMo | C4 | 671B |
5454| qwen35_397b_grpo | [ llm_post_training] ( https://github.com/mlcommons/training/tree/master/llm_post_training ) | NeMo-RL / NeMo-Gym | SWE tasks | 397B
55+ | dlrm_v4_hstu | [ recommendation] ( https://github.com/mlcommons/training/tree/master/recommendation ) | pytorch | Yambda-5b | HSTU
5556
5657* Framework here is given for the reference implementation. Submitters are free to use their own frameworks to run the benchmark.
5758
@@ -64,8 +65,8 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
6465| flux.1 | [ text_to_image] ( https://github.com/mlcommons/training/tree/master/text_to_image ) | torchtitan | CC12M subset | 11.9B
6566| llama3.1_8b | [ small_llm_pretraining] ( https://github.com/mlcommons/training/tree/master/small_llm_pretraining ) | NeMo | C4 | 8b
6667| llama2_70b_lora | [ llama2_70b_lora] ( https://github.com/mlcommons/training/tree/master/llama2_70b_lora ) | pytorch | SCROLLS GovReport | 70B
67- | llama3.1_405b | [ large_language_model_pretraining] ( https://github.com/mlcommons/training/tree/master/large_language_model_pretraining ) | NeMo | C4 | 405B
68- | dlrm_dcnv2 | [ recommendation_v2] ( https://github.com/mlcommons/training/tree/master/recommendation_v2/torchrec_dlrm ) | torchrec | Criteo 3.5TB multi-hot | 167M
68+ | llama3.1_405b | [ large_language_model_pretraining] ( https://github.com/mlcommons/training/tree/master/retired_benchmarks/ large_language_model_pretraining ) | NeMo | C4 | 405B
69+ | dlrm_dcnv2 | [ recommendation_v2] ( https://github.com/mlcommons/training/tree/master/retired_benchmarks/ recommendation_v2/torchrec_dlrm ) | torchrec | Criteo 3.5TB multi-hot | 167M
6970| gpt_oss_20b | [ small_llm_moe_pretraining] ( https://github.com/mlcommons/training/tree/master/small_llm_moe_pretraining/primus ) | Primus | C4 | 20B |
7071| deepseekv3 | [ llm_moe_pretraining] ( https://github.com/mlcommons/training/tree/master/llm_moe_pretraining/nemo ) | NeMo | C4 | 671B |
7172
@@ -81,8 +82,8 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
8182| flux.1 | [ text_to_image] ( https://github.com/mlcommons/training/tree/master/text_to_image ) | torchtitan | CC12M subset | 11.9B
8283| llama3.1_8b | [ small_llm_pretraining] ( https://github.com/mlcommons/training/tree/master/small_llm_pretraining ) | NeMo | C4 | 8b
8384| llama2_70b_lora | [ llama2_70b_lora] ( https://github.com/mlcommons/training/tree/master/llama2_70b_lora ) | pytorch | SCROLLS GovReport | 70B
84- | llama3.1_405b | [ large_language_model_pretraining] ( https://github.com/mlcommons/training/tree/master/large_language_model_pretraining ) | NeMo | C4 | 405B
85- | dlrm_dcnv2 | [ recommendation_v2] ( https://github.com/mlcommons/training/tree/master/recommendation_v2/torchrec_dlrm ) | torchrec | Criteo 3.5TB multi-hot | 167M
85+ | llama3.1_405b | [ large_language_model_pretraining] ( https://github.com/mlcommons/training/tree/master/retired_benchmarks/ large_language_model_pretraining ) | NeMo | C4 | 405B
86+ | dlrm_dcnv2 | [ recommendation_v2] ( https://github.com/mlcommons/training/tree/master/retired_benchmarks/ recommendation_v2/torchrec_dlrm ) | torchrec | Criteo 3.5TB multi-hot | 167M
8687| rgat | [ graph_neural_network] ( https://github.com/mlcommons/training/tree/master/graph_neural_network ) | GLT | IGBH-Full | 25M
8788
8889* Framework here is given for the reference implementation. Submitters are free to use their own frameworks to run the benchmark.
@@ -99,9 +100,9 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
99100| retinanet | [ single_stage_detector] ( https://github.com/mlcommons/training/tree/master/single_stage_detector ) | pytorch | OpenImages | 37M
100101| Stable Diffusionv2 | [ image generation] ( https://github.com/mlcommons/training/tree/master/stable_diffusion ) | pytorch | LAION-400M-filtered | 865M
101102| bert | [ language_model] ( https://github.com/mlcommons/training/tree/master/language_model/tensorflow/bert ) | tensorflow | Wikipedia 2020/01/01 | 340M
102- | llama3.1_405b | [ large_language_model_pretraining] ( https://github.com/mlcommons/training/tree/master/large_language_model_pretraining ) | NeMo | C4 | 405B
103+ | llama3.1_405b | [ large_language_model_pretraining] ( https://github.com/mlcommons/training/tree/master/retired_benchmarks/ large_language_model_pretraining ) | NeMo | C4 | 405B
103104| llama2_70b_lora | [ llama2_70b_lora] ( https://github.com/mlcommons/training/tree/master/llama2_70b_lora ) | pytorch | SCROLLS GovReport | 70B
104- | dlrm_dcnv2 | [ recommendation_v2] ( https://github.com/mlcommons/training/tree/master/recommendation_v2/torchrec_dlrm ) | torchrec | Criteo 3.5TB multi-hot | 167M
105+ | dlrm_dcnv2 | [ recommendation_v2] ( https://github.com/mlcommons/training/tree/master/retired_benchmarks/ recommendation_v2/torchrec_dlrm ) | torchrec | Criteo 3.5TB multi-hot | 167M
105106| rgat | [ graph_neural_network] ( https://github.com/mlcommons/training/tree/master/graph_neural_network ) | GLT | IGBH-Full | 25M
106107
107108# MLPerf Training v4.1 (Submission Deadline Oct 11, 2024)
@@ -114,7 +115,7 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
114115| BERT-large | [ language/nlp] ( https://github.com/mlcommons/training/tree/master/language_model/tensorflow/bert ) | tensorflow | Wikipedia 2020/01/01 | 340M
115116| GPT3 | [ language/llm] ( https://github.com/mlcommons/training/tree/master/large_language_model ) | paxml,megatron-lm | C4 | 175B
116117| LLama2 70B-LoRA | [ language/LLM fine-tuning] ( https://github.com/mlcommons/training/tree/master/llama2_70b_lora ) | pytorch | SCROLLS GovReport | 70B
117- | DLRMv2 | [ recommendation] ( https://github.com/mlcommons/training/tree/master/recommendation_v2/torchrec_dlrm ) | torchrec | Criteo 3.5TB multi-hot | 167M
118+ | DLRMv2 | [ recommendation] ( https://github.com/mlcommons/training/tree/master/retired_benchmarks/ recommendation_v2/torchrec_dlrm ) | torchrec | Criteo 3.5TB multi-hot | 167M
118119| RGAT | [ GNN] ( https://github.com/mlcommons/training/tree/master/graph_neural_network ) | pytorch | IGBH-Full | 25M
119120
120121* Note model parameter count is not the same as active parameter that are being trained in the benchmark.
@@ -131,7 +132,7 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
131132| BERT-large | [ language/nlp] ( https://github.com/mlcommons/training/tree/master/language_model/tensorflow/bert ) | tensorflow | Wikipedia 2020/01/01
132133| GPT3 | [ language/llm] ( https://github.com/mlcommons/training/tree/master/large_language_model ) | paxml,megatron-lm | C4
133134| LLama2 70B-LoRA | [ language/LLM fine-tuning] ( https://github.com/mlcommons/training/tree/master/llama2_70b_lora ) | pytorch | SCROLLS GovReport
134- | DLRMv2 | [ recommendation] ( https://github.com/mlcommons/training/tree/master/recommendation_v2/torchrec_dlrm ) | torchrec | Criteo 3.5TB multi-hot
135+ | DLRMv2 | [ recommendation] ( https://github.com/mlcommons/training/tree/master/retired_benchmarks/ recommendation_v2/torchrec_dlrm ) | torchrec | Criteo 3.5TB multi-hot
135136| RGAT | [ GNN] ( https://github.com/mlcommons/training/tree/master/graph_neural_network ) | pytorch | IGBFull
136137
137138
0 commit comments