writing
Writing about ML implementations and the occasional non-technical stuff.
More coming soon.
Linear-Time Sequential Modeling: Mamba from Scratch
coming soon
August 2026
Implementing FlashAttention using Pallas
coming soon
July 2026
Implementing GRPO from Scratch in JAX
coming soon
June 2026
Proximal Policy Optimization, in JAX
coming soon
May 2026