Optimizing Data Distribution and Kernel Performance for Efficient Training of Chemistry Foundation Models: A Case Study with MACE. - Jesun Firoz; Franco Pellegrini; Mario Geiger; Darren Hsu; Jenna A. Bilbrey; Han-Yi Chou; Maximilian Stadler; Markus Hoehnerbach; Tingyu Wang; Dejun Lin; Emine Küçükbenli; Henry W. Sprueill; et al. (CoRR, 2025)