Never lose your progress: Checkpointing with OrbaxPicture this. You have spent six hours training a model. The loss curve looks beautiful, accuracy is climbing, and you are one epoch away from a result worth writing home about. Then the power goes ouAug 7, 2026·16 min read
Custom training loop from scratch in JAXWriting a training loop that you can actually controlMar 4, 2026·10 min read
Building Neural Networks with Flax NNX Flax NNX: Neural Networks That Feel Like PyTorchFeb 23, 2026·11 min read
Transformations That Change EverythingAutomatic vectorization with vmap and gradients with grad in JaxFeb 9, 2026·11 min read
Why JAX? The NumPy You Know, But FasterHow JAX uses XLA to strap a rocket engine onto your math operations.Feb 2, 2026·10 min read
The reverse turing test: We must now prove we are “dumb” to beat AIThe pressure to appear less polished, less articulate, or even "dumber", simply to avoid being mistaken for a machineDec 10, 2025·4 min read
Google Colab in VS Code: A Deep Dive into the New ExtensionA comprehensive guide to leveraging Google's cloud computing prowess directly within your favorite local editor.Nov 16, 2025·7 min read
