Merge pull request #1694 from Saransh-cpp/fix-docs

valentinsulzer · web-flow · commit 025b4d11767b · 2021-09-28T07:44:53.000-04:00
Pin `jax` and `jaxlib` versions - fix docs deployment
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -6,8 +6,8 @@ anytree >= 2.4.3
 autograd >= 1.2
 scikit-fem >= 0.2.0
 casadi >= 3.5.0
-jax>=0.1.68
-jaxlib>=0.1.47
+jax==0.2.12
+jaxlib==0.1.70
 jupyter  # For example notebooks
 pybtex
 # Note: Matplotlib is loaded for debug plots but to ensure pybamm runs
