The Array Cast cover image

PyTorch and Dex with Adam Paszke

The Array Cast

CHAPTER

The Evolution of Autograd in PyTorch

Jax is an evolution of autograd, a library built by Matt Johnson and Google MacLaur. Autodiff in PyTorch is actually exposed through a module called torch autograd. Jax was meant to keep things like automatic differentiation but also add better support for running on accelerators.

00:00
Transcript
Play full episode

Remember Everything You Learn from Podcasts

Save insights instantly, chat with episodes, and build lasting knowledge - all powered by AI.
App store bannerPlay store banner