Last released Sep 1, 2026
Named Tensors for Legible Deep Learning in JAX
Last released Jun 12, 2026
A slightly opinionated framework for simple dataclass-based configurations based on Pyrallis.
Last released Oct 18, 2025
Scalable Training for Foundation Models with Named Tensors and JAX