diff --git a/requirements_travis.txt b/requirements_travis.txt index 2b281ebb7..7d226c1a3 100644 --- a/requirements_travis.txt +++ b/requirements_travis.txt @@ -1,6 +1,6 @@ tensorflow>=2.0.0 pytype==2019.06.21 pytest -torch>=1.1 +torch==1.3.1 jax>=0.1.0 jaxlib>=0.1.27