Today at Stanford, we released Levanter, a Jax-based framework for training foundation models. It is now open-source on Github under Apache 2.0.