diff --git a/templates/local_modal_runner.py b/templates/local_modal_runner.py
index 6157d14..1d9cab9 100644
--- a/templates/local_modal_runner.py
+++ b/templates/local_modal_runner.py
@@ -91,6 +91,7 @@ def _github_ref(repo: str, override_env: str) -> str:
     .uv_pip_install(
         "tinygrad~=0.10",
         "helion",
+        "max",
     )
     .uv_pip_install(
         "nvidia-cutlass-dsl==4.5.2",