Abdullah

Currently there is no official Pytorch template for version 2.5+ and requires manual torch update or using an uncached container image. Pytorch 2.5 introduced some good features (for e.g. FlexAttention)...