Abhinav Goel

Abhinav Goel is a senior deep learning architect and the technical lead for JAX performance at NVIDIA. He has published over twenty peer-reviewed research articles and holds three patents. He earned a Ph.D. from the Elmore Family School of Electrical and Computer Engineering at Purdue University.
Avatar photo

Posts by Abhinav Goel

Decorative image.
Agentic AI / Generative AI

Train Models Faster with JAX and MaxText Using NVFP4 on NVIDIA Blackwell

Pre-training frontier LLMs comes down to throughput. When training spans trillions of tokens across thousands of accelerators, every percentage point of step... 7 MIN READ