Update requirements.txt

ashiqimranintel · web-flow · commit 741f4d0d5091 · 2024-12-18T10:20:36.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
