Pin specific revisions for JAX and Haiku.

PiperOrigin-RevId: 318034576 Change-Id: I5ab0d89a6906784004bfff8c21452873ea02a9d7
giogix2 · Jun 24, 2020 · 63ede7d · 63ede7d
1 parent bc2cf69
commit 63ede7d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -12,6 +12,6 @@ matplotlib == 3.1.1
 nashpy == 0.0.19
 scipy == 1.4.1
 attrs == 19.1.0
-jax >= 0.1.67
-jaxlib >= 0.1.47
-git+git://github.com/deepmind/dm-haiku@master
+jax == 0.1.67
+jaxlib == 0.1.47
+dm-haiku == 0.0.1