muloco-1. A simple reference implementation of the single-worker MuLoCo optimizer in Jax & PyTorch. MuLoCo-1 has been shown to outperfrom Muon and have larger critical batch sizes.

github.com/bentherien/muloco-1

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.