PyTorch Developer Podcast
PyTorch Developer Podcast
Edward Yang, Team PyTorch
AOTAutograd
19 minutes Posted May 9, 2022 at 4:00 pm.
0:00
19:12
Download MP3
Show notes

AOTAutograd is a cool new feature in functorch for capturing both forward and backward traces of PyTorch operators, letting you run them through a compiler and then drop the compiled kernels back into a normal PyTorch eager program. Today, Horace joins me to tell me how it works, what it is good to use for, and what our future plans for it are.