How Does Critical Batch Size Scale in Pre-training? - Zhang, Hanlin; Morwani, Depen; Vyas, Nikhil; Wu, Jingfeng; Zou, Difan; Ghai, Udaya; Foster, Dean; Kakade, Sham (International Conference on Learning Representations, 2025)