Skip to content

DeepSpeed bf16 Gradient Norm Underflow

DeepSpeed's bf16 training can trigger 'assert all_groups_norm > 0' because bf16 gradient norms can underflow to zero. This is a numerical precision issue specific to bf16's limited range. Denpex detects the underflow from DeepSpeed error logs and recommends the fix.

Quick answer

DeepSpeed's bf16 training can trigger 'assert all_groups_norm > 0' because bf16 gradient norms can underflow to zero.

Symptom
AssertionError: all_groups_norm > 0 in DeepSpeed gradient norm computation
Root cause
bf16 has fewer mantissa bits (7) than fp16 (10), reducing the smallest representable positive number. When gradient norms are very small (common with large models, small LR, or gradient accumulation), they can underflow to zero in bf16. DeepSpeed's gradient norm assertion assumes non-zero norms, which is not guaranteed with bf16 precision.
Recommended fix
Upgrade DeepSpeed to v0.13.5 or later which fixes the bf16 norm underflow
How Denpex helps
Denpex investigates DeepSpeed bf16 Gradient Norm Underflow using the evidence you provide or your connected workload collects. Earlier rank, host or application evidence is needed to distinguish an initiating failure from a downstream report.
Training Stability#deepspeed#bf16#gradient-norm#underflow#precision#numerical-stability

What this failure is

DeepSpeed bf16 Gradient Norm Underflow is a Training Stability failure seen during ML training runs. DeepSpeed's bf16 training can trigger 'assert all_groups_norm > 0' because bf16 gradient norms can underflow to zero. This is a numerical precision issue specific to bf16's limited range. Denpex detects the underflow from DeepSpeed error logs and recommends the fix. Common tags: Deepspeed, Bf16, Gradient Norm, Underflow.

Live diagnosis, no signup

Is this what broke your run? Paste your log.

You're reading about DeepSpeed bf16 Gradient Norm Underflow. Paste your own traceback and relevant evidence for an investigation of your workload, with a next action or a specific missing fact. A reference entry does not establish your cause. No account or card for the free diagnosis. Review data handling before submitting sensitive logs.

Before uploading, review cloud data handling and local options.

training_logs.txt
No log to hand? Try one:

Upload your log file. Large files are processed in your browser. Only selected evidence is submitted when you run diagnosis.

3 free diagnoses/day

Why it happens (the mechanism)

Bf16 has fewer mantissa bits (7) than fp16 (10), reducing the smallest representable positive number. When gradient norms are very small (common with large models, small LR, or gradient accumulation), they can underflow to zero in bf16. DeepSpeed's gradient norm assertion assumes non-zero norms, which is not guaranteed with bf16 precision. The gradient norm is computed in bf16 precision instead of being upcast to fp32 for the norm calculation. Taken together, these mechanisms explain why the failure is reproducible, why it tends to surface on specific workloads or scales, and why generic mitigation attempts often fall short without addressing the underlying cause.

What you'll observe

  • Training crashes with 'assert all_groups_norm > 0' when using bf16 precision with DeepSpeed
  • The error occurs even though gradients are non-zero (they're just too small for bf16 to represent)
  • Switching to fp16 resolves the issue but bf16 is preferred for training stability

Common symptoms and what they mean

SymptomWhy it happens
AssertionError: all_groups_norm > 0 in DeepSpeed gradient norm computationbf16 has fewer mantissa bits (7) than fp16 (10), reducing the smallest representable positive number
Training runs fine with fp16 but crashes with bf16When gradient norms are very small (common with large models, small LR, or gradient accumulation), they can underflow to zero in bf16
The error occurs at the same step consistentlyDeepSpeed's gradient norm assertion assumes non-zero norms, which is not guaranteed with bf16 precision
Gradient norms reported as 0.0 in DeepSpeed logs even though the model is learningThe gradient norm is computed in bf16 precision instead of being upcast to fp32 for the norm calculation

Which systems are affected

  • DeepSpeed ZeRO stage 2/3 with bf16 precision
  • Training with small learning rates that produce very small gradients
  • Models with many layers where gradients diminish through backpropagation
  • DeepSpeed versions before 0.13.5

How to confirm this is the problem

Use this checklist to test the hypothesis against a small reproduction. No single line proves the root cause, so preserve the preceding events and compare one variable at a time.

  • ✓Reproduce the failure from a clean checkpoint/seed: the symptom must appear without warm-up state from a previous run.
  • ✓Verified signal present: AssertionError: all_groups_norm > 0 in DeepSpeed gradient norm computation
  • ✓Verified signal present: Training runs fine with fp16 but crashes with bf16
  • ✓Verified signal present: The error occurs at the same step consistently
  • ✓Verified signal present: Gradient norms reported as 0.0 in DeepSpeed logs even though the model is learning
  • ✓A targeted fix from the "How to fix it" section eliminates or substantially reduces the symptom within one validation pass.

Root cause

  • bf16 has fewer mantissa bits (7) than fp16 (10), reducing the smallest representable positive number
  • When gradient norms are very small (common with large models, small LR, or gradient accumulation), they can underflow to zero in bf16
  • DeepSpeed's gradient norm assertion assumes non-zero norms, which is not guaranteed with bf16 precision
  • The gradient norm is computed in bf16 precision instead of being upcast to fp32 for the norm calculation

The fix and how to prevent it

Searchable error signature

search key
AssertionError: all_groups_norm > 0 in DeepSpeed gradient norm computation

Use this text as a lookup key in logs and upstream issue trackers. It is not presented as a captured customer log. Confirm the cause from your own preceding events, versions, configuration and the cited references.

The fix and the prevention pattern

The root cause is on this page and stays free. A free account adds the exact remediation steps, saved history, and the fix on every entry in the encyclopedia.

Sign up free. Unlock the full analysis

No credit card. Daily allowance follows verified trust tier. Instant access.

Diagnose this failure in VS Code

Select the traceback or open the failed terminal, then run Denpex locally to see the initiating rank, collateral failures, exact fix, and verification command without uploading the log.

Install the free VS Code extension

DeepSpeed errors in context

DeepSpeed changes when parameters, gradients and optimizer state are created, partitioned, gathered and offloaded. The hub separates ZeRO, memory, checkpoint and pipeline failures by lifecycle phase.

Compare every deepspeed error side by side

Don't just read the fix, diagnose your run

The encyclopedia tells you what went wrong. Denpex tells you what went wrong in YOUR training run. With your logs, your config, and your stack.