diff --git a/requirements.txt b/requirements.txt
index ba2cd2c..71c8750 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,7 +8,7 @@ absl-py==0.11.0           # via dm-haiku, jax, jaxlib
 dm-haiku==0.0.2           # via pandas_toolkit (setup.py)
 jax-toolkit==0.1.2        # via pandas_toolkit (setup.py)
 jax==0.2.5               # via jax-toolkit, pandas_toolkit (setup.py)
-jaxlib==0.1.56            # via jax-toolkit, pandas_toolkit (setup.py)
+jaxlib==0.4.2            # via jax-toolkit, pandas_toolkit (setup.py)
 joblib==0.17.0            # via scikit-learn
 numpy==1.19.4             # via dm-haiku, jax, jaxlib, opt-einsum, pandas, scikit-learn, scipy
 opt-einsum==3.3.0         # via jax