MARCO ROSSI. Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet. Journal of Artificial Intelligence Review, [S. l.], v. 7, n. 2, p. 148–163, 2026. DOI: 10.69987/. Disponível em: https://learnedvertex.com/index.php/JAIR/article/view/421. Acesso em: 11 oct. 2026.