From ab5de6795f9dcb77888696efdc0a4f26ac24008f Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Sun, 12 Oct 2025 09:35:06 -0700
Subject: [PATCH] bring in muon

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

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",