
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