Apply pre-commit formatting (isort) [skip-build]

Add MoE uneven shard test with mixed expert and non-expert params [skip-build]

Add uneven shard correctness test [skip-build]

Add optimization docs and update implementation guide [skip-build]

Update tests for MoE and parallel optimizations [skip-build]

Muon optimizer: expert batching, parallel caching, A2A overlap [skip-build]

Optimize pipeline: batched update, zero-copy scatter, prelaunch gather [skip-build]

Cache AdamW placement grouping and tensor lists [skip-build]

Add torch.compile, CUDA graph, and compiled momentum [skip-build]

Apply suggestions from code review

Add mhc_attn, mhc_ffn, lambda_proj to skip_keys

Remove verbose param_groups summary logging

Support multi-component expert_keys (e.g. "experts.w1")

Extract is_expert_param() helper to consolidate expert key matching

Include original (pre-normalize) FQN in is_muon logging

Add info-level logging for param group classification (Muon vs AdamW)

Use component-level matching for expert_keys to avoid shared_experts collision

Normalize parameter FQNs to handle torch.compile / checkpoint wrappers

Merge pull request #17 from MotifTechnologies/optimal-ns-coefficients

Apply pre-commit formatting (yapf) [skip-build]

Add max_iter cap and non-finite checks to _optimal_quintic [skip-build]

Apply pre-commit formatting (yapf, isort) [skip-build]

Add comment explaining _coeffs_list and Polar Express vs former NS [skip-build]

Replace hardcoded NS coefficients with analytically optimal ones [skip-build]

Refactor pipeline to async generator pattern (#16)

Support mHC (#15)

Update arxiv URL

Support param group with various placements (#13)

Update tag (#8)

Merge pull request #14 from MotifTechnologies/fix_bug_in_fsdp

Add built binary [skip-build]

fix bug in fsdp

feat(workflow): add Slack notifications for build start, success, and failure [skip-build] (#12)

Merge pull request #11 from MotifTechnologies/ca1207-patch-1

Add built binary [skip-build]

Update torch-ext/optimizer/muon.py

Update torch-ext/optimizer/muon.py

Update muon.py

Merge pull request #10 from MotifTechnologies/fix_a2a_gs_assert

Add built binary [skip-build]

fix assert in a2a gather scatter

fix: use [skip-build] keyword to prevent build [skip-build]

Merge pull request #9 from MotifTechnologies/all2all_gather_scatter

Add built binary [ci skip]

delete state in split_func

Add built binary [ci skip]

change owner_params to owned_params

modify pre step (overlap step) can get from arsgs

add doc strings + init self rank on init_assign_params

Add built binary [ci skip]