Update Modal PyTorch and CuTe DSL versions (#488)

Author: msaroufim

src/runners/modal_runner.py

@@ -59,20 +59,19 @@
     # nvidia cuda packages
     .uv_pip_install(
         "nvidia-cupynumeric~=25.3",
-        "nvidia-cutlass-dsl==4.3.5",
+        "nvidia-cutlass-dsl==4.5.2",
         "cuda-core[cu13]",
         "cuda-python[all]==13.0",
         # "nvmath-python[cu13]~=0.4",
         # "numba-cuda[cu13]~=0.15",
     )
     # Install torch last so its CUDA/NCCL dependency set wins over broader CUDA Python packages.
     .uv_pip_install(
-        "torch==2.11.0",
-        index_url="https://download.pytorch.org/whl/cu129",
+        "torch==2.12.0",
     )
     # CUTLASS C++ headers for #include <cutlass/...>
     .run_commands(
-        "git clone --depth 1 --branch v4.3.5 https://github.com/NVIDIA/cutlass.git /opt/cutlass",
+        "git clone --depth 1 --branch v4.5.2 https://github.com/NVIDIA/cutlass.git /opt/cutlass",
     )
     .env({
         "CUTLASS_PATH": "/opt/cutlass",