TORAX: Tokamak transport simulation in JAX
Open weights language model from Google DeepMind, based on Griffin.
Open on GitHub