Marco Rossi. 2026. “Multi-Head Attention-Based OOM Risk Prediction for CNN and Transformer Training Tasks on GPUMemNet”. Journal of Artificial Intelligence Review 7 (2): 148-63. https://doi.org/10.69987/.