I've just had a look through this, and I have to say that I'm really impressed!
A framework for training large models, top-notch software engineering, and best of all -- built using Equinox! :)
Today, I’m excited to announce the release of Levanter 1.0, our new JAX-based framework for training foundation models, which we’ve been working on
@StanfordCRFM
. Levanter is designed to be legible, scalable and reproducible.
@PatrickKidger
Yeah, I'm a big fan of Equinox! I owed you/Equinox a shoutout in the tweets but forgot! Even in places where I'm not directly using Equinox, it was a huge influence.