Custom training loop from scratch in JAX
Writing a training loop that you can actually control
Mar 4, 202610 min read34
Command Palette
Search for a command to run...
Series
Machine Learning
In this series, I will be sharing how to kickstart a career in Machine Learning with a concentration on TensorFlow
Transformations That Change Everything
Automatic vectorization with vmap and gradients with grad in Jax
Feb 9, 202610 min read78
Why JAX? The NumPy You Know, But Faster
How JAX uses XLA to strap a rocket engine onto your math operations.
Feb 2, 20269 min read124
Google Colab in VS Code: A Deep Dive into the New Extension
A comprehensive guide to leveraging Google's cloud computing prowess directly within your favorite local editor.
Nov 16, 20257 min read1.5K
Build Your First AI Agent with Gemini and LlamaIndex
Building a Tourism AI Assistant with Agentic Workflows
Sep 14, 202511 min read393
Luganda Inference on Gemma 3
Running an inference in Luganda on the Gemma 3 models for image and video tasks
Mar 17, 20257 min read1.3K