RBA gives LLaVA-OneVision-2 a 3x lossless training speed-up, merged upstream

liuziwei7 · x · 2026-11-10

The team used RBA to fully automatically optimize LLaVA-OneVision-2's training, achieving a 3x lossless speed-up: one full epoch dropped from 9.64h to 3.19h on 8×Blackwell with matching loss curves. On 4×Hopper 80GB it went from 21.94h to 9.84h (2.2x). MFU jumped from 11.6% to 45.5% on Blackwell and 22.1% to 45.5% on Hopper, with eval loss essentially unchanged. Verified across 2 nodes × 8 GPUs with multi-node launch and load balancing, under both TP 2 / PP 4 presets. Merged into the main repo via PR #205.

Original post →

More from Infra

Infra channel →