implementation detail · filed under optimization
Stochastic Rounding of Gradients
Stochastic rounding applied specifically to gradients in the NVFP4 training context.
Also called stochastic rounding on gradients.
- source
- 1
- model
- 1
- lab adopt it
- 1
- strongest
- used
How sources treat it
One count per evidence span, weakest treatment to strongest.
used 1
Documented in
Evidence
1 span quoted from the sources, strongest treatment first.
stochastic rounding on gradients
usedoptimizationin Nemotron 3 familyNVIDIA
Filed alongside
Other methods under optimization :: quantization-aware training.
Quantization-Aware TrainingFP4 QuantizationFP4 Quantization-Aware TrainingMXFP4 Weights with MXFP8 ActivationsStochastic RoundingINT4 Quantization-Aware TrainingMXFP4 Quantization-Aware Post-TrainingNVFP4 TrainingPer-Block Scalar ScalingQ4_0 Quantization FormatQuantize-Dequantize TrainingRandom Hadamard TransformsStochastic Rounding for Mamba CacheStraight-Through Estimator