Search for a command to run...
Series
In this series, I will be sharing how to kickstart a career in Machine Learning with a concentration on TensorFlow
Here's a scenario every ML engineer eventually lives through: you kick off a training run before leaving the office. Three hours later, you check back in. The loss curve looks like a flat line at NaN.
In our last article on this blog in March, we built a complete training loop. We used optax.adamw() to create an optimizer, wrapped it in nnx.Optimizer, and watched our model learn. But we barely scra
Writing a training loop that you can actually control
Automatic vectorization with vmap and gradients with grad in Jax
How JAX uses XLA to strap a rocket engine onto your math operations.
A comprehensive guide to leveraging Google's cloud computing prowess directly within your favorite local editor.
