Jul 1, 2026Jul 1, 2026 · 11 min read · 196 viewsOptax: Optimizers You Can Compose Like LEGOIn our last article on this blog in March, we built a complete training loop. We used optax.adamw() to create an optimizer, wrapped it in nnx.Optimizer, and watched our model learn. But we barely scra11 min read · 196 views
