bump flax version (#14343)

examples/flax/language-modeling/requirements.txt

@@ -1,5 +1,5 @@
 datasets >= 1.1.3
 jax>=0.2.8
 jaxlib>=0.1.59
-flax>=0.3.4
+flax>=0.3.5
 optax>=0.0.9

examples/flax/question-answering/requirements.txt

@@ -1,5 +1,5 @@
 datasets >= 1.8.0
 jax>=0.2.17
 jaxlib>=0.1.68
-flax>=0.3.4
+flax>=0.3.5
 optax>=0.0.8

examples/flax/summarization/requirements.txt

@@ -1,5 +1,5 @@
 datasets >= 1.1.3
 jax>=0.2.8
 jaxlib>=0.1.59
-flax>=0.3.4
+flax>=0.3.5
 optax>=0.0.8

examples/flax/text-classification/requirements.txt

@@ -1,5 +1,5 @@
 datasets >= 1.1.3
 jax>=0.2.8
 jaxlib>=0.1.59
-flax>=0.3.4
+flax>=0.3.5
 optax>=0.0.8

examples/flax/token-classification/requirements.txt

@@ -1,6 +1,6 @@
 datasets >= 1.8.0
 jax>=0.2.8
 jaxlib>=0.1.59
-flax>=0.3.4
+flax>=0.3.5
 optax>=0.0.8
 seqeval

examples/flax/vision/requirements.txt

@@ -1,6 +1,6 @@
 jax>=0.2.8
 jaxlib>=0.1.59
-flax>=0.3.4
+flax>=0.3.5
 optax>=0.0.8
 -f https://download.pytorch.org/whl/torch_stable.html
 torch==1.9.0+cpu