HeavyBall/pyproject.toml at main · HomebrewML/HeavyBall · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
[build-system]
requires = ["setuptools>=75.0"]
build-backend = "setuptools.build_meta"

[project]
name = "heavyball"
description = "Compile-first PyTorch optimizer library - AdamW, Muon, SOAP/Shampoo, PSGD, Schedule-Free, and 30+ more with torch.compile fusion and composable features"
version = "3.1.1"
authors = [{ name = "HeavyBall Authors", email = "github.heavyball@nestler.sh" }]
classifiers = ["Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: BSD License",
    "Natural Language :: English",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = ["opt-einsum>=3.4.0",
    "torch>=2.2,<3.0",
    "numpy>=2.3.4",
]
keywords = [
    "pytorch",
    "torch",
    "optimizer",
    "adam",
    "adamw",
    "sgd",
    "deep-learning",
    "machine-learning",
    "torch-compile",
    "triton",
    "second-order",
    "mixed-precision",
    "memory-efficient",
    "soap",
    "muon",
    "psgd",
    "adopt",
    "shampoo",
    "schedule-free",
    "laprop",
    "ademamix",
    "nadam",
    "rmsprop",
    "sam",
    "preconditioning",
    "kronecker",
    "mars",
    "llm-training",
    "distributed",
    "fsdp",
    "ddp",
]
readme = "README.md"
requires-python = ">=3.9"

[project.optional-dependencies]
dev = ["pre-commit", "pytest", "hypothesis", "ruff", "matplotlib", "seaborn", "pandas", "typer", "optuna", "optunahub", "gpytorch", "scikit-learn", "lightbench"]

[project.urls]
Homepage = "https://github.com/HomebrewML/HeavyBall"
source = "https://github.com/HomebrewML/HeavyBall"
tracker = "https://github.com/HomebrewML/HeavyBall/issues"

[tool.ruff]
line-length = 120

[tool.ruff.lint]
extend-select = ["I", "W"]
ignore = ["E741"]
preview = false

[tool.ruff.lint.isort]
relative-imports-order = "closest-to-furthest"

[tool.ruff.format]
preview = false

[tool.setuptools.packages.find]
include = ["heavyball*"]