seniorPyTorch

How does gradient scaling prevent underflow in mixed precision training?

Updated May 17, 2026

Short answer

Gradient scaling multiplies loss to prevent small gradients from underflowing in FP16.

Deep explanation

In FP16, small gradient values may become zero due to limited precision. GradScaler scales loss before backpropagation and unscales gradients before optimizer step to maintain numerical stability.

Unlock with a Pro subscription to view this section.

View pricing

Real-world example

No real-world example available yet.

Unlock with a Pro subscription to view this section.

Upgrade to Pro

Common mistakes

No common mistakes listed yet.

Unlock with a Pro subscription to view this section.

Upgrade to Pro

Follow-up questions

No follow-up questions available yet.

Unlock with a Pro subscription to view this section.

Upgrade to Pro

More PyTorch interview questions

View all →