Ashutoshkumarsingh·Jun 3PyTorch, Keras, and Flax in 2026: Architecture, Performance, and Real-World AdoptionIntroduction
KAKANAKOU Miguel·May 16Transformer from Scratch with JAX and Flax — Part 3: FSDP TrainingThe full code is available on Github.
Youngbin Lee·Apr 29GPT Writes One Token at a Time. This Model Writes Entire Blocks at Once.Building a Diffusion Language Model from Scratch with Flax NNX on TPU
KAKANAKOU Miguel·Apr 20Transformer from scratch with Jax and Flax — Part 2 of 2The full code is available on Github.
KAKANAKOU Miguel·Apr 20Transformer from scratch with Jax and Flax - Part 1The full code is available on Github.
