PyTorch Developer Podcast
PyTorch Developer Podcast
Edward Yang, Team PyTorch
Torch vs ATen APIs
15 minutes Posted Apr 11, 2022 at 4:00 pm.
0:00
15:03
Download MP3
Show notes

PyTorch’s torch API is the Python API everyone knows and loves, but there’s also another API, the ATen API, which most of PyTorch’s internal subsystems are built on. How to tell them apart? What implications do these have on our graph mode IR design? Also, a plug for PrimTorch, a new set of operators, not designed for eager mode, that is supposed to be even lower level than ATen.