Refactor pipeline to async generator pattern (#16)

Support mHC (#15)

Support param group with various placements (#13)

Add built binary [skip-build]

Add built binary [skip-build]

Add built binary [skip-build]

Add built binary [ci skip]

Add built binary [ci skip]

Add built binary [ci skip]

Add built binary [ci skip]

feat(muon_clip) : add muon clip (#6)

chore: add nix build workflow (#8)

feat(muon) : add tuned-abc-values & blfoat16 communication

feat: update muon to receive paramgroups, not model (#4)

fix(muon): add update_p stage and dealloc tensors properly

Support HSDP (#4)

support torch 2.8 (#3)

fix(muon): handle un-distributed env

refactor(muon): change argument adam_wd to weight_decay and handle params' type

fix(muon): free tensors that are no longer needed

chore(muon): update comment

chore(muon): clean build and update doc

fix(muon): delete intermediate tensors immediately to lower peak mem usage

chore: initial commit