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