Marco Rossi. Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet. AIMLR 2026, 7 (2), 148-163. https://doi.org/10.69987/AIMLR.2026.70211.