diff --git a/requirements_with_jax_stable_stack.txt b/requirements_with_jax_stable_stack.txt
index 869d2d88c..ba21b4774 100644
--- a/requirements_with_jax_stable_stack.txt
+++ b/requirements_with_jax_stable_stack.txt
@@ -7,7 +7,7 @@ flax>=0.10.2
 ftfy
 git+https://github.com/mlperf/logging.git
 google-cloud-storage==2.17.0
-grain-nightly
+grain-nightly==0.0.10
 huggingface_hub==0.24.7
 jax>=0.4.30
 jaxlib>=0.4.30