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