TorchBraid: High-Performance Layer-Parallel Training of Deep Neural Networks with MPI and GPU Acceleration. - CYR, ERIC C.; HAHNE, JENS; MOORE, NICHOLAS S.; SCHRODER, JACOB B.; SOUTHWORTH, BEN S.; VARGAS, DAVID A. (ACM Transactions on Mathematical Software, 2025)