diff --git a/doc/requirements.txt b/doc/requirements.txt
index 732b82e..d344e3b 100644
--- a/doc/requirements.txt
+++ b/doc/requirements.txt
@@ -1,6 +1,6 @@
 numpy==1.22.4
 scipy==1.7.3
-qutip==5.0.0a1
+qutip==5.0.0a2
 cython==0.29.30
 sphinx==5.0.2
 sphinx_rtd_theme==1.0.0
@@ -10,5 +10,5 @@ matplotlib==3.5.2
 docutils==0.17.1
 pyqir-generator==0.6.2
 pyqir-parser==0.6.2
-jax==0.4.11
-jaxlib==0.4.11
+jax
+jaxlib