Last released Nov 22, 2024
Scalable Training for Foundation Models with Named Tensors and JAX
Last released Nov 18, 2024
Named Tensors for Legible Deep Learning in JAX
Last released Nov 13, 2024
A slightly opinionated framework for simple dataclass-based configurations based on Pyrallis.
Supported by