pin pygrain to 0.0.10 (#160)

Signed-off-by: kunjan <kunjanp@google.com>

Author: coolkp

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