Add tunix to tpu

djherbis · web-flow · commit 4bb016df17bd · 2025-09-17T23:46:03.000-04:00
diff --git a/tpu/requirements.txt b/tpu/requirements.txt
@@ -20,6 +20,7 @@ git+https://github.com/deepmind/dm-haiku
 jraph
 optax
 trax
+git+https://github.com/google/tunix
 # Jupyter packages
 jupyter-lsp==1.5.1
 jupyterlab
