bump version for klsoap

Author: ClashLuke

heavyball/utils.py

@@ -1281,7 +1281,7 @@ def compute_correction(self, fp32, narrow):
         ls = (_log_ulp(narrow) - 1).float()
         e_norm = _scale_by_exp2(e, -ls)
         scaled = e_norm.clamp(-1.0, 1.0) * self.smax
-        # SR on the int correction — the bits below the correction's resolution.
+        # SR on the int correction (the bits below the correction's resolution)
         # narrow is RNE so |e| ≤ ULP/2, keeping `scaled` in [-smax, smax]; SR
         # adds at most 1 unit of correction (= ULP/(2*smax)) of error and is
         # unbiased on the lowest representable bits.

pyproject.toml

@@ -5,7 +5,7 @@ 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.0.1"
+version = "3.1.0"
 authors = [{ name = "HeavyBall Authors", email = "github.heavyball@nestler.sh" }]
 classifiers = ["Intended Audience :: Developers",
     "Intended Audience :: Science/Research",