revert requirements.txt to main branch

requirements.txt

@@ -1,19 +1,19 @@
 absl-py==1.0.0
-brax==0.9.2
-chex>=0.1.5
+brax==0.0.15
+chex==0.1.5
 dm-haiku==0.0.5
-flax==0.7.3
+flax==0.6.0
 gym==0.23.1
 ipython
-jax==0.4.16
-jaxlib==0.4.16
-jumanji==0.3.1
+jax==0.3.17
+jaxlib==0.3.15
+jumanji==0.1.3
 jupyter
 numpy==1.22.3
-optax==0.1.7
+optax==0.1.4
 protobuf==3.19.4
 scikit-learn==1.0.2
 scipy==1.8.0
 seaborn==0.11.2
 tensorflow-probability==0.15.0
-typing-extensions==4.3.0
+typing-extensions==4.3.0