Skip to navigation
PerformancePerformance Summary

Performance Summary

View as Markdown

Training performance measurements for language, vision-language, and image-generation models using NeMo AutoModel with the PyTorch backend. Model names link to pinned YAML recipes.

Fine-Tuning Performance

Full Fine-Tuning: NVIDIA H100 80GB, Precision: BF16

ModelWorkload#GPUsGBSMBSLBSGASeq LengthTPPPCPEPVPFSDPKernel OptimizationsTime per Global Step (s)Model TFLOPs/sec/GPUTokens/secTokens/sec/GPU
Qwen3.8-27BMedPix VLM88111Variable1111-NRSDPA + torch1.732NR1,865233.07
GLM-5.3Tulu-3 packed256256141409614164NRNRcuDNN + torch + HybridEP20.584NR41,615162.56
GLM-5.3-FlashMedPix packed72144114204811272-NRcuDNN + torch + torch_mm + HybridEP17.094NR9,480131.66
DeepSeek-V4-Flash-Vision-ExpMedPix VLM128128141409614132NRNRTileLang + torch + torch_mm + HybridEP11.178NR3,29925.77
MiMo-V2.5-ProHellaSwag1442881361Variable11818NRNRSDPA + torch + torch_mm + DeepEP6.028NR1,81612.61
MiMo-V2.6-Flash-RLTulu-3 packed6464111409611164-NRTE attention + torch + torch_mm + HybridEP5.221NR41,011640.80
Nemotron-3.5-Lightning-30B-A3B-BF16Tulu-3 packed8811181921118-NRTE attention/linear + torch_mm + DeepEP1.025NR55,1676,895.85
Nemotron-3.5-Super-VL-120B-A12B-BF16Tulu-3 packed (text only)6432114819211864-NRTE attention/linear12.355NR14,737230.26
Kimi-Linear-48B-A3B-InstructHellaSwag packed8811140961118-NReager attention + torch2.862NR11,3171,414.63
Laguna-XS-2.1HellaSwag packed8811110241118-NRSDPA + torch + torch_mm1.323NR6,174771.73
  • Qwen3.8-27B freezes vision; GLM-5.3-Flash and Nemotron-3.5-Super-VL freeze vision and embeddings.
  • Nemotron-3.5-Super-VL uses text-only checkpoint nvidia/NVIDIA-Nemotron-3.5-Super-midtrain-67B-vision-pretrained.
  • Nemotron-3.5-Lightning uses all 52 layers with MTP disabled.

Full Fine-Tuning: NVIDIA GB200, Precision: BF16

ModelWorkload#GPUsGBSMBSLBSGASeq LengthTPPPCPEPVPFSDPKernel OptimizationsTime per Global Step (s)Model TFLOPs/sec/GPUTokens/secTokens/sec/GPU
DeepSeek-V4.1-FlashMedPix VLM6464111409611164-NRTileLang + torch + torch_mm + HybridEP3.585NR5,47385.52
MiMo-V2.6-Pro-RLTulu-3 packed128161216553612864NRNRTE attention/linear + torch_mm + HybridEP16.973NR58,391456.18

FSDP2; MiMo-V2.6-Pro-RL includes profiler overhead.

LoRA Performance

LoRA: NVIDIA H100 80GB, Precision: BF16

ModelWorkload#GPUsGBSMBSLBSGASeq LengthTPPPCPEPVPFSDPKernel OptimizationsTime per Global Step (s)Model TFLOPs/sec/GPUTokens/secTokens/sec/GPU
Llama3 8BMock13222164096111-11TE + triton10.5140212,47312472.87
Qwen2.5 7BMock13222164096111-11TE + triton9.2942314,11014110.05
Llama3 70BMock8322244096211-14TE + triton + fsdp2_prefetch15.003168,7351091.85
Qwen2.5 32BMock8322244096211-14TE + triton + fsdp2_prefetch7.2830118,0022250.31
Llama3 70B 2-nodeMock16322224096211-18TE + triton + fsdp2_prefetch8.3228515,758984.85
Qwen2.5 32B 2-nodeMock16322224096211-18TE + triton + fsdp2_prefetch3.9527733,1662072.89

Full-length mock sequences; no padding.

Pre-Training Performance

GPU: NVIDIA H100 80GB, Precision: BF16

ModelWorkload#GPUsGBSMBSLBSGASeq LengthTPPPCPEPVPFSDPKernel OptimizationsTime per Global Step (s)Model TFLOPs/sec/GPUTokens/secTokens/sec/GPU
Nemotron V3 Super 120BMock64512224409611164-64TE + DeepEP + TorchSDPA7.286334287,8084,497
Nemotron V3 Nano 30BMock8512441640961118-8TE + DeepEP + TorchSDPA15.614328134,31216,789
Nemotron V3 Ultra 550BMock128256221409611164-128TE + DeepEP + torch_mm10.05293104,320815
DeepSeek V3 671BMock102481921844096141648256TE + DeepEP37.87216885,760865
DeepSeek V3 671BMock256512181409614164864TE + DeepEP8.18250256,5121,002
Kimi K2Mock256512182409618132432TE + DeepEP8.86189236,544924
Qwen3 MoE 30BMock8512441640961118-8TE + DeepEP21.77327796,32012,040
GPT-OSS 20BMock825622164096111--8TE + DeepEP + FlexAttn10.04279104,46413,058
GPT-OSS 120BMock645122244096111--64TE + DeepEP + FlexAttn4.30231488,0647,626
Llama3 70BMock641281148192112--32TE + fsdp2_prefetch18.9038955,473866.77
Qwen3.8-Flash-NextMock, balanced routing6464111819211164-64FA4 QSA (cute) + torch + gmm + HybridEP3.93186.85133,3642,083.81
  • Full-length mock sequences; balanced MoE routing.
  • Qwen3.8-Flash-Next: 20 warmup and 30 measured steps; label-token rate; TFLOPs derived from MFU.

GPU: NVIDIA H100 80GB, Precision: FP8

ModelWorkload#GPUsGBSMBSLBSGASeq LengthTPPPCPEPVPFSDPKernel OptimizationsTime per Global Step (s)Model TFLOPs/sec/GPUTokens/secTokens/sec/GPU
Moonlight 16BMock, balanced routing825644820481118-8TE FP8 + DeepEP3.006324.94174,41421,801.73

TE current scaling; 10 warmup and 20 measured steps.

GPU: NVIDIA GB200, Precision: MXFP8

ModelWorkload#GPUsGBSMBSLBSGASeq LengthTPPPCPEPVPFSDPKernel OptimizationsTime per Global Step (s)Model TFLOPs/sec/GPUTokens/secTokens/sec/GPU
DeepSeek V3 671BMock, balanced routing64128221409611164-64TE MXFP8 + HybridEP2.986685.45175,5822,743.47
DeepSeek V3 671B (compiled MLA)Mock, balanced routing64128221409611164-64SDPA + compiled MLA + TE MXFP8 + HybridEP6.044338.5786,7451,355.39
Qwen3 MoE 30BMock, balanced routing85121616440961118-8TE MXFP8 + HybridEP9.164658.08228,84728,605.85
Qwen3 MoE 235BMock, balanced routing3281924464409611132-32TE MXFP8 + HybridEP194.098799.73172,8745,402.30
GPT-OSS 120BMock, balanced routing321280885409611116-32TE MXFP8 + HybridEP7.840681.95668,73520,897.96
  • TE MXFP8 block scaling; 10 warmup and 20 measured steps, except Qwen3 MoE 235B (two warmup and three measured steps).
  • FP8/MXFP8 token rates count full sequences; TFLOPs derive from MFU and the configured peak.

Image Generation Performance

Image Generation, Full Fine-Tuning: NVIDIA H100 80GB, Precision: BF16

ModelMode#GPUsGBSMBSLBSGATPPPCPEPOptimizerTime per Global Step (s)Images/sec/GPU
Qwen-Image-2.1Full fine-tuning816221111-TE FusedAdam, FP32 master weights~1.38~1.45

Naruto BLIP, 1024 x 1024, FSDP2. Author-reported 1,000-step run; warmup exclusion unspecified.

Image Generation, LoRA: NVIDIA H100 80GB, Precision: BF16

ModelMode#GPUsGBSMBSLBSGATPPPCPEPOptimizerTime per Global Step (s)Images/sec/GPU
HunyuanImage-3.0LoRA, rank 64881111118TE FusedAdam, 16-bit remainders2.2260.449

FSDP2; 100 steps, averaging steps 10-99. Cached 1024 x 1024 Naruto BLIP inputs; synchronized maximum-rank timing excludes setup and checkpoint I/O.

Glossary

  • NR: Not recorded or not verified
  • MFU: Model FLOPs Utilization - ratio of achieved compute to peak hardware capability
  • TP: Tensor Parallelism - splits individual layers across GPUs
  • PP: Pipeline Parallelism - splits model layers into stages
  • CP: Context Parallelism - splits the sequence across GPUs
  • FSDP: Fully Sharded Data Parallelism - the table column gives the shard degree when verified
  • EP: Expert Parallelism - distributes MoE experts across GPUs
  • DP: Data Parallelism - replicates model and splits data
  • VP: Virtual Pipeline - number of pipeline stages per GPU for interleaving
  • MBS: Micro-Batch Size - size of one forward pass in pipeline
  • LBS: Local Batch Size - size of one step per GPU
  • GBS: Global Batch Size - total batch size across all GPUs
  • GA: Gradient Accumulation - number of local-batches before optimizer step
  • TE: Transformer Engine kernel optimizations - RMSNorm, Linear and DotProductAttention
  • DeepEP: Expert-parallel communication library for MoE token dispatch and combine
  • FlexAttn: PyTorch’s Flex Attention

Configuration Files

Model names in the tables link directly to their YAML recipes. Mock-data benchmarks live in examples/llm_benchmark/; dataset-based recipes live in examples/llm_finetune/, examples/vlm_finetune/, examples/retrieval/, and examples/diffusion/finetune/. Use the recorded table settings and pinned recipes when reproducing a historical result.

Version Information

  • Last Updated: 2026-10-14
  • Default Recipe Snapshot: b203f3f9
  • Measurements: Historical development and benchmark runs; source commits vary.