4min chapter

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

Get the Snipd
podcast app

Unlock the knowledge in podcasts with the podcast player of the future.
App store bannerPlay store banner

AI-powered
podcast player

Listen to all your favourite podcasts with AI-powered features

Discover
highlights

Listen to the best highlights from the podcasts you love and dive into the full episode

Save any
moment

Hear something you like? Tap your headphones to save it with AI-generated key takeaways

Share
& Export

Send highlights to Twitter, WhatsApp or export them to Notion, Readwise & more

AI-powered
podcast player

Listen to all your favourite podcasts with AI-powered features

Discover
highlights

Listen to the best highlights from the podcasts you love and dive into the full episode