Marco Rossi, “Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet”, JAIR, vol. 7, no. 2, pp. 148–163, May 2026, doi: 10.69987/.