diff --git a/requirements-dev.txt b/requirements-dev.txt
index 4795fd7..cb33b92 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -2,8 +2,8 @@ pre-commit
 mypy==0.910
 mkdocs-material
 mkdocstrings-python
+jaxlib==0.4.16
 jax==0.4.16
 jax-cosmo==0.1.0
-jaxlib==0.4.14
 pytest==7.4.3
 pytest-cov==4.1.0
diff --git a/requirements.txt b/requirements.txt
index 5fdac37..9fe926c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+jaxlib==0.4.16
 jax==0.4.16
 jax-cosmo==0.1.0
-jaxlib==0.4.14
+