Update requirements.txt

ashiqimranintel · web-flow · commit b3dc28823f8e · 2024-12-18T10:19:57.000-08:00
diff --git a/test/requirements.txt b/test/requirements.txt
@@ -1,3 +1,4 @@
 jax==0.4.30
 jaxlib==0.4.30
 flax==0.8.5
+optax==0.2.4
