Use a released version of optax rather than installing from Git. (#12173)

Use a released version of optax rather than installing from Git
2021-06-15 13:12:51 +02:00
parent 9bc9e59869
commit 9b393240a2
2 changed files with 2 additions and 1 deletions
--- a/examples/flax/language-modeling/requirements.txt
+++ b/examples/flax/language-modeling/requirements.txt
@@ -2,3 +2,4 @@ datasets >= 1.1.3
 jax>=0.2.8
 jaxlib>=0.1.59
 flax>=0.3.4
+optax>=0.0.8