The Array Cast cover image

PyTorch and Dex with Adam Paszke

The Array Cast

00:00

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.

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