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

The AI-powered Podcast Player

Save insights by tapping your headphones, chat with episodes, discover the best highlights - and more!
App store bannerPlay store banner
Get the app