Create requirements.txt

requirements.txt

@@ -0,0 +1,14 @@
+jax~=0.4.13
+flax~=0.7.1
+fjutils~=0.0.15
+numpy==1.25.2
+typing~=3.7.4.3
+transformers~=4.31.0
+einops~=0.6.1
+optax~=0.1.7
+msgpack~=1.0.5
+ipython~=8.14.0
+tqdm==4.65.0
+datasets==2.14.3
+setuptools~=60.0.0
+easydel