adv-optm 2.dev2__tar.gz → 2.dev3__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.

Potentially problematic release.

This version of adv-optm might be problematic. Click here for more details.

Files changed (28) hide show
  1. {adv_optm-2.dev2 → adv_optm-2.dev3}/PKG-INFO +1 -1
  2. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/__init__.py +1 -1
  3. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/optim/Muon_adv.py +3 -0
  4. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm.egg-info/PKG-INFO +1 -1
  5. {adv_optm-2.dev2 → adv_optm-2.dev3}/setup.py +1 -1
  6. {adv_optm-2.dev2 → adv_optm-2.dev3}/LICENSE +0 -0
  7. {adv_optm-2.dev2 → adv_optm-2.dev3}/README.md +0 -0
  8. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/optim/AdaMuon_adv.py +0 -0
  9. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/optim/AdamW_adv.py +0 -0
  10. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/optim/Adopt_adv.py +0 -0
  11. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/optim/Lion_Prodigy_adv.py +0 -0
  12. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/optim/Lion_adv.py +0 -0
  13. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/optim/Prodigy_adv.py +0 -0
  14. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/optim/Simplified_AdEMAMix.py +0 -0
  15. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/optim/__init__.py +0 -0
  16. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/util/BF16_Stochastic_Rounding.py +0 -0
  17. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/util/Effective_Shape.py +0 -0
  18. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/util/Kourkoutas.py +0 -0
  19. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/util/NNMF.py +0 -0
  20. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/util/Newton_Schulz.py +0 -0
  21. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/util/One_Bit_Boolean.py +0 -0
  22. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/util/OrthoGrad.py +0 -0
  23. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/util/__init__.py +0 -0
  24. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm.egg-info/SOURCES.txt +0 -0
  25. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm.egg-info/dependency_links.txt +0 -0
  26. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm.egg-info/requires.txt +0 -0
  27. {adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm.egg-info/top_level.txt +0 -0
  28. {adv_optm-2.dev2 → adv_optm-2.dev3}/setup.cfg +0 -0
{adv_optm-2.dev2 → adv_optm-2.dev3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: adv_optm
3
- Version: 2.dev2
3
+ Version: 2.dev3
4
4
  Summary: A family of highly efficient, lightweight yet powerful optimizers.
5
5
  Home-page: https://github.com/Koratahiu/Advanced_Optimizers
6
6
  Author: Koratahiu
{adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/__init__.py RENAMED
@@ -20,4 +20,4 @@ __all__ = [
20
20
  "AdaMuon_adv",
21
21
  ]
22
22
 
23
- __version__ = "2.dev2"
23
+ __version__ = "2.dev3"
{adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm/optim/Muon_adv.py RENAMED
@@ -76,6 +76,7 @@ class Muon_adv(torch.optim.Optimizer):
76
76
  adam_beta3_ema (float): Beta3 for AdEMAMix.
77
77
  adam_alpha (float): Alpha for AdEMAMix.
78
78
  adam_kourkoutas_beta (bool): Kourkoutas-β for AdamW.
79
+ adam_nnmf_factor (bool): 1-bit factored for AdamW.
79
80
  """
80
81
 
81
82
  def __init__(
@@ -125,6 +126,7 @@ class Muon_adv(torch.optim.Optimizer):
125
126
  adam_ema_alpha: float = 0.95,
126
127
  adam_tiny_spike: float = 1e-9,
127
128
  adam_k_warmup_steps: int = 0,
129
+ adam_nnmf_factor: bool = False,
128
130
  ):
129
131
  if not (lr >= 0.0):
130
132
  raise ValueError(f"Learning-rate should be >= 0.0. Got {lr}")
@@ -165,6 +167,7 @@ class Muon_adv(torch.optim.Optimizer):
165
167
  "adam_kourkoutas_beta": adam_kourkoutas_beta, "adam_beta2_min": adam_beta2_min,
166
168
  "adam_ema_alpha": adam_ema_alpha, "adam_tiny_spike": adam_tiny_spike,
167
169
  "adam_k_warmup_steps": adam_k_warmup_steps,
170
+ "adam_nnmf_factor":adam_nnmf_factor,
168
171
  }
169
172
  self.stochastic_rounding = stochastic_rounding
170
173
  self.compiled_optimizer = compiled_optimizer
{adv_optm-2.dev2 → adv_optm-2.dev3}/adv_optm.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: adv_optm
3
- Version: 2.dev2
3
+ Version: 2.dev3
4
4
  Summary: A family of highly efficient, lightweight yet powerful optimizers.
5
5
  Home-page: https://github.com/Koratahiu/Advanced_Optimizers
6
6
  Author: Koratahiu
{adv_optm-2.dev2 → adv_optm-2.dev3}/setup.py RENAMED
@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
5
5
 
6
6
  setup(
7
7
  name="adv_optm",
8
- version="2.dev2",
8
+ version="2.dev3",
9
9
  author="Koratahiu",
10
10
  author_email="hiuhonor@gmail.com",
11
11
  license='Apache 2.0',
{adv_optm-2.dev2 → adv_optm-2.dev3}/LICENSE RENAMED
File without changes
{adv_optm-2.dev2 → adv_optm-2.dev3}/README.md RENAMED
File without changes
{adv_optm-2.dev2 → adv_optm-2.dev3}/setup.cfg RENAMED
File without changes