heavyball 1.6.2__tar.gz → 1.6.3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (30) hide show
  1. {heavyball-1.6.2 → heavyball-1.6.3}/PKG-INFO +9 -3
  2. {heavyball-1.6.2 → heavyball-1.6.3}/heavyball.egg-info/PKG-INFO +9 -3
  3. heavyball-1.6.3/heavyball.egg-info/requires.txt +13 -0
  4. {heavyball-1.6.2 → heavyball-1.6.3}/pyproject.toml +18 -26
  5. heavyball-1.6.2/heavyball.egg-info/requires.txt +0 -7
  6. {heavyball-1.6.2 → heavyball-1.6.3}/LICENSE +0 -0
  7. {heavyball-1.6.2 → heavyball-1.6.3}/README.md +0 -0
  8. {heavyball-1.6.2 → heavyball-1.6.3}/heavyball/__init__.py +0 -0
  9. {heavyball-1.6.2 → heavyball-1.6.3}/heavyball/chainable.py +0 -0
  10. {heavyball-1.6.2 → heavyball-1.6.3}/heavyball/utils.py +0 -0
  11. {heavyball-1.6.2 → heavyball-1.6.3}/heavyball.egg-info/SOURCES.txt +0 -0
  12. {heavyball-1.6.2 → heavyball-1.6.3}/heavyball.egg-info/dependency_links.txt +0 -0
  13. {heavyball-1.6.2 → heavyball-1.6.3}/heavyball.egg-info/top_level.txt +0 -0
  14. {heavyball-1.6.2 → heavyball-1.6.3}/setup.cfg +0 -0
  15. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_bf16_params.py +0 -0
  16. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_bf16_q.py +0 -0
  17. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_bf16_storage.py +0 -0
  18. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_caution.py +0 -0
  19. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_channels_last.py +0 -0
  20. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_closure.py +0 -0
  21. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_ema.py +0 -0
  22. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_foreach.py +0 -0
  23. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_hook.py +0 -0
  24. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_mars.py +0 -0
  25. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_memory.py +0 -0
  26. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_merge.py +0 -0
  27. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_no_grad.py +0 -0
  28. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_psgd.py +0 -0
  29. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_soap.py +0 -0
  30. {heavyball-1.6.2 → heavyball-1.6.3}/test/test_stochastic_updates.py +0 -0
{heavyball-1.6.2 → heavyball-1.6.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: heavyball
3
- Version: 1.6.2
3
+ Version: 1.6.3
4
4
  Summary: Efficient Optimizers
5
5
  Author-email: HeavyBall Authors <github.heavyball@nestler.sh>
6
6
  Project-URL: source, https://github.com/HomebrewML/HeavyBall
@@ -15,12 +15,18 @@ Classifier: Programming Language :: Python :: 3
15
15
  Requires-Python: >=3.9
16
16
  Description-Content-Type: text/markdown
17
17
  License-File: LICENSE
18
- Requires-Dist: opt-einsum>=3.0.0
19
- Requires-Dist: torch>=2.0.0
18
+ Requires-Dist: opt-einsum>=3.4.0
19
+ Requires-Dist: torch>=2.1.0
20
+ Requires-Dist: numpy
20
21
  Provides-Extra: dev
21
22
  Requires-Dist: pre-commit; extra == "dev"
22
23
  Requires-Dist: pytest; extra == "dev"
23
24
  Requires-Dist: ruff; extra == "dev"
25
+ Requires-Dist: matplotlib; extra == "dev"
26
+ Requires-Dist: seaborn; extra == "dev"
27
+ Requires-Dist: hyperopt; extra == "dev"
28
+ Requires-Dist: pandas; extra == "dev"
29
+ Requires-Dist: typer; extra == "dev"
24
30
 
25
31
  # `heavyball`: Efficient Optimizers
26
32
 
{heavyball-1.6.2 → heavyball-1.6.3}/heavyball.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: heavyball
3
- Version: 1.6.2
3
+ Version: 1.6.3
4
4
  Summary: Efficient Optimizers
5
5
  Author-email: HeavyBall Authors <github.heavyball@nestler.sh>
6
6
  Project-URL: source, https://github.com/HomebrewML/HeavyBall
@@ -15,12 +15,18 @@ Classifier: Programming Language :: Python :: 3
15
15
  Requires-Python: >=3.9
16
16
  Description-Content-Type: text/markdown
17
17
  License-File: LICENSE
18
- Requires-Dist: opt-einsum>=3.0.0
19
- Requires-Dist: torch>=2.0.0
18
+ Requires-Dist: opt-einsum>=3.4.0
19
+ Requires-Dist: torch>=2.1.0
20
+ Requires-Dist: numpy
20
21
  Provides-Extra: dev
21
22
  Requires-Dist: pre-commit; extra == "dev"
22
23
  Requires-Dist: pytest; extra == "dev"
23
24
  Requires-Dist: ruff; extra == "dev"
25
+ Requires-Dist: matplotlib; extra == "dev"
26
+ Requires-Dist: seaborn; extra == "dev"
27
+ Requires-Dist: hyperopt; extra == "dev"
28
+ Requires-Dist: pandas; extra == "dev"
29
+ Requires-Dist: typer; extra == "dev"
24
30
 
25
31
  # `heavyball`: Efficient Optimizers
26
32
 
heavyball-1.6.3/heavyball.egg-info/requires.txt ADDED
@@ -0,0 +1,13 @@
1
+ opt-einsum>=3.4.0
2
+ torch>=2.1.0
3
+ numpy
4
+
5
+ [dev]
6
+ pre-commit
7
+ pytest
8
+ ruff
9
+ matplotlib
10
+ seaborn
11
+ hyperopt
12
+ pandas
13
+ typer
{heavyball-1.6.2 → heavyball-1.6.3}/pyproject.toml RENAMED
@@ -5,45 +5,37 @@ build-backend = "setuptools.build_meta"
5
5
  [project]
6
6
  name = "heavyball"
7
7
  description = "Efficient Optimizers"
8
- version = "1.6.2"
9
- authors = [
10
- {name = "HeavyBall Authors", email = "github.heavyball@nestler.sh"}
8
+ version = "1.6.3"
9
+ authors = [{ name = "HeavyBall Authors", email = "github.heavyball@nestler.sh" }]
10
+ classifiers = ["Intended Audience :: Developers",
11
+ "Intended Audience :: Science/Research",
12
+ "License :: OSI Approved :: BSD License",
13
+ "Natural Language :: English",
14
+ "Operating System :: OS Independent",
15
+ "Programming Language :: Python :: 3",
11
16
  ]
12
- classifiers = [
13
- "Intended Audience :: Developers",
14
- "Intended Audience :: Science/Research",
15
- "License :: OSI Approved :: BSD License",
16
- "Natural Language :: English",
17
- "Operating System :: OS Independent",
18
- "Programming Language :: Python :: 3",
17
+ dependencies = ["opt-einsum>=3.4.0",
18
+ "torch>=2.1.0",
19
+ "numpy",
19
20
  ]
20
- dependencies = [
21
- "opt-einsum>=3.0.0",
22
- "torch>=2.0.0",
23
- ]
24
- keywords = [
25
- "torch",
26
- "optimizer",
27
- "muon",
28
- "soap",
29
- "psgd",
21
+ keywords = ["torch",
22
+ "optimizer",
23
+ "muon",
24
+ "soap",
25
+ "psgd",
30
26
  ]
31
27
  readme = "README.md"
32
28
  requires-python = ">=3.9"
33
29
 
34
30
  [project.optional-dependencies]
35
- dev = [
36
- "pre-commit",
37
- "pytest",
38
- "ruff",
39
- ]
31
+ dev = ["pre-commit", "pytest", "ruff", "matplotlib", "seaborn", "hyperopt", "pandas", "typer"]
40
32
 
41
33
  [project.urls]
42
34
  source = "https://github.com/HomebrewML/HeavyBall"
43
35
  tracker = "https://github.com/HomebrewML/HeavyBall/issues"
44
36
 
45
37
  [tool.ruff]
46
- line-length = 99
38
+ line-length = 120
47
39
 
48
40
  [tool.ruff.lint]
49
41
  extend-select = ["I", "W"]
heavyball-1.6.2/heavyball.egg-info/requires.txt DELETED
@@ -1,7 +0,0 @@
1
- opt-einsum>=3.0.0
2
- torch>=2.0.0
3
-
4
- [dev]
5
- pre-commit
6
- pytest
7
- ruff
{heavyball-1.6.2 → heavyball-1.6.3}/LICENSE RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/README.md RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/heavyball/utils.py RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/setup.cfg RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/test/test_bf16_q.py RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/test/test_ema.py RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/test/test_hook.py RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/test/test_mars.py RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/test/test_memory.py RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/test/test_merge.py RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/test/test_psgd.py RENAMED
File without changes
{heavyball-1.6.2 → heavyball-1.6.3}/test/test_soap.py RENAMED
File without changes