bump flax version (#14343)

examples/flax/language-modeling/requirements.txt

 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

 datasets >= 1.8.0
 jax>=0.2.17
 jaxlib>=0.1.68
-flax>=0.3.4
+flax>=0.3.5
 optax>=0.0.8
\ No newline at end of file

examples/flax/summarization/requirements.txt

 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

 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

 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
\ No newline at end of file

examples/flax/vision/requirements.txt

 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 

examples/research_projects/jax-projects/hybrid_clip/requirements.txt

 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 

setup.py

@@ -104,7 +104,7 @@ _deps = [
     "fastapi",
     "filelock",
     "flake8>=3.8.3",
-    "flax>=0.3.4",
+    "flax>=0.3.5",
     "fugashi>=1.0",
     "GitPython<3.1.19",
     "huggingface-hub>=0.1.0,<1.0",

src/transformers/dependency_versions_table.py

@@ -15,7 +15,7 @@ deps = {
     "fastapi": "fastapi",
     "filelock": "filelock",
     "flake8": "flake8>=3.8.3",
-    "flax": "flax>=0.3.4",
+    "flax": "flax>=0.3.5",
     "fugashi": "fugashi>=1.0",
     "GitPython": "GitPython<3.1.19",
     "huggingface-hub": "huggingface-hub>=0.1.0,<1.0",