diff --git a/pyproject.toml b/pyproject.toml
index 0da8b6d..baed46f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -17,7 +17,6 @@ keywords = []
 dependencies = [
     "jax==0.4.31",
     "flax==0.8.5",
-    "grain==0.2.0",
     "ml_dtypes==0.4.0",
     "optax==0.2.3",
     "orbax==0.1.9",