High Level Neural Network Modeling with FlaxIn our previous article, we broke down why JAX requires explicit state handling. We saw that wrapping our parameters, optimizer momentum, and PRNG keys into a clean carry pattern lets us use jax.jit, Aug 19, 2026·7 min read·112
Why the JAX AI Stack is built for the Future of Foundation ModelsFoundation models like large-scale transformers, multimodal systems, MoEs, etc are pushing the currently known boundaries of AI. Training and serving them demands extreme scale, hardware efficiency, rJul 3, 2026·4 min read·98
Deploying on Fridays?An article explaining why developers should avoid deployment on FridayDec 26, 2025·3 min read·37
Welcome. Let’s all settle in, shall we?An introductory article on what to expect hereDec 21, 2025·3 min read·21
