Stochastic rounding lets BF16 RL post-training converge as fast as FP16

heghbalz · x · 2026-11-13

A technical thread on RL post-training numerics: casting FP32 shadow weights to BF16 with stochastic rounding lets BF16 RL post-training converge as fast as FP16. At the same learning rate, vanilla BF16 lags — even when training and inference backends produce bit-exact logprobs.

Key points:

Original post →

More from Infra

Infra channel →