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