Torch vs ATen APIs

11/04/2022 15 min Episodio 57
Torch vs ATen APIs

Listen "Torch vs ATen APIs"

Episode Synopsis

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.