Official Runpod PyTorch 2.5 Template

[AbdullahWasTaken]

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) that have made its way into downstream libraries like Huggingface.

4d05770ce08f5aa6ba184a01f8e9acbdc0c999b4 added Pytorch 2.5 as a target in the docker bake. Is it possible to push a container image for this tag? It will be quite helpful. Thanks