Lyra Aug 25, 2026 A performant and compact LLM pretraining stack in pure JAX JAX · TPU · Pallas Chroma Jan 9, 2026 An AOT fixed-shape compiler and runtime for PyTorch graph exports PyTorch · C++ · CUDA · Metal