Marco Rossi. Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet. JAIR [Internet]. 2026 May 3 [cited 2026 Oct. 10];7(2):148-63. Available from: https://learnedvertex.com/index.php/JAIR/article/view/421