Last released May 20, 2024
A slightly opinionated framework for simple dataclass-based configurations based on Pyrallis.
Last released May 17, 2024
Named Tensors for Legible Deep Learning in JAX
Last released Oct 24, 2023
Scalable Training for Foundation Models with Named Tensors and JAX
Supported by