A pure function rides through three stations: grad turns it into its gradient, jit compiles it, and vmap runs it on a whole batch at once. Random numbers come from keys you split by hand. Then reuse a key, change the batch size every call, or go back to a Python loop, and watch the clock. This lesson needs JavaScript.