Modular Manifolds

Supporting code for the blog post on modular manifolds.

Find the blog post at https://thinkingmachines.ai/blog/modular-manifolds.

Code structure

src/
├── main.py                     # Entry point: training loop and CLI
├── msign.py                    # Matrix sign function via Polar-Express
├── manifold_muon.py            # Manifold Muon update rule
└── hyperspherical_descent.py   # Hyperspherical descent update rule