diff --git a/benchmarks/requirements.txt b/benchmarks/requirements.txt
index 07ab7d5..4fdd02b 100644
--- a/benchmarks/requirements.txt
+++ b/benchmarks/requirements.txt
@@ -1,4 +1,4 @@
-jax[cpu]==0.4.14
+jax[cpu]==0.5.0
 brainpy==2.4.6.post5
 taichi==1.7.0
 brainpylib==0.2.6