diff --git a/pyproject.toml b/pyproject.toml
index f073f37..272a9ef 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -27,6 +27,7 @@ classifiers=[
 
 dependencies = [
     "accelerate",
+    "adam-atan2-pytorch>=0.2.2",
     "assoc-scan",
     "einx>=0.3.0",
     "einops>=0.8.1",