pin optax==0.0.2 (kingoflolz#12)

Author: nostalgebraist

requirements.txt

@@ -6,7 +6,7 @@ wandb~=0.10.22
 einops~=0.3.0
 requests~=2.25.1
 fabric~=2.6.0
-optax~=0.0.2
+optax==0.0.2
 git+https://github.com/deepmind/dm-haiku
 git+https://github.com/EleutherAI/lm-evaluation-harness/
 ray~=1.2.0
@@ -17,4 +17,4 @@ tensorflow-cpu~=2.4.1
 google-cloud-storage~=1.36.2
 smart_open[gcs]
 func_timeout
-ftfy
+ftfy