diff --git a/.ci/docker/ci_commit_pins/jax.txt b/.ci/docker/ci_commit_pins/jax.txt
index 100435be135..6f4eebdf6f6 100644
--- a/.ci/docker/ci_commit_pins/jax.txt
+++ b/.ci/docker/ci_commit_pins/jax.txt
@@ -1 +1 @@
-0.8.2
+0.8.1