Hacker News with Generative AI: Jax

Shape Rotation 101: An Intro to Einsum and Jax Transformers (bearblog.dev)
Maxtext: A simple, performant and scalable Jax LLM (github.com/google)