Registry / ai-ml / adam-atan2-pytorch

adam-atan2-pytorch

JSON →
library0.3.4pypypiunverified

An optimizer with Adam-atan2 update rule for PyTorch. Current version 0.3.4, released 2025. Active development with periodic updates.

ai-ml
pip install adam-atan2-pytorch
github ↗homepage ↗
INSTALL
IMPORT
SIG · ADAM-ATAN2-PYTORCH
A
adam-atan2-pytorch
ai-mlpythonv0.3.4
harness data pending
Install & Compatibility
Where this runs

No compatibility data collected yet for this library.

Code
Verified usage

Verified import paths — ran on the pinned version, not inferred.

AdamAtan2
✓ from adam_atan2_pytorch import AdamAtan2
Correct import path. Note the camelCase class name.

Minimal usage example: create model, optimizer, run one step.

import torch from adam_atan2_pytorch import AdamAtan2 model = torch.nn.Linear(10, 2) optimizer = AdamAtan2(model.parameters(), lr=1e-3) loss_fn = torch.nn.CrossEntropyLoss() input = torch.randn(4, 10) target = torch.randint(0, 2, (4,)) for _ in range(10): optimizer.zero_grad() output = model(input) loss = loss_fn(output, target) loss.backward() optimizer.step() print("Success")
Debug
Known issues
gotchaThe optimizer is named AdamAtan2 (camelCase), not Adam_atan2 or AdamAtan2Pytorch. Use exact class name.
fix
from adam_atan2_pytorch import AdamAtan2
affects: all
gotchaRequires PyTorch >= 1.13.0 due to use of accelerate and newer optimizer features. Older versions will fail.
fix
Upgrade PyTorch to >=1.13.0
affects: >=0.3.0
deprecatedThe 'betas' parameter now expects a tuple (b1, b2) instead of two separate keyword arguments. Old usage may break.
fix
Use optimizer = AdamAtan2(params, lr=1e-3, betas=(0.9, 0.999)) instead of betas1 and betas2.
affects: >=0.3.0
Upgrade
Version history
0.3.4latest on PyPI · released May 5, 2026
Audit
Dependencies
torchrequiredRuntime dependency
einopsrequiredUsed internally for tensor operations
Agent activity
53 hits · last 30 days
node
46
Resources
githubhttps://github.com/lucidrains/adam_atan2_pytorch ↗
homepagehttps://github.com/lucidrains/adam_atan2_pytorch ↗
adam-atan2-pytorch — pip install adam-atan2-pytorch · libregistry