diff --git a/pyproject.toml b/pyproject.toml
index 6fc3f9b40f..25638fd714 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -34,7 +34,7 @@ classifiers = [
 dependencies = [
     "docrep>=0.3.2",
     "flax",
-    "lightning>=2.0,<2.2",
+    "lightning>=2.0,<2.3",
     "ml-collections>=0.1.1",
     "mudata>=0.1.2",
     "numpyro>=0.12.1",