Added tokamax as a requirement in requirements_with_jax_ai_image.txt

prishajain1 · prishajain1 · commit 6a7e4a2ea8eb · 2025-11-16T22:11:58.000+05:30
diff --git a/requirements_with_jax_ai_image.txt b/requirements_with_jax_ai_image.txt
@@ -30,6 +30,7 @@ orbax-checkpoint
 tokenizers==0.21.0
 huggingface_hub>=0.30.2
 transformers==4.48.1
+tokamax
 einops==0.8.0
 sentencepiece
 aqtp
