PyTorch Developer Podcast
PyTorch Developer Podcast
Edward Yang, Team PyTorch
Backend extensibility
15 minutes Posted May 14, 2021 at 1:00 pm.
0:00
15:16
Download MP3
Show notes

What's the current state of backend extensibility? How did PyTorch evolve from being a CPU and CUDA only framework to also support AMD ROCm and XLA? What are some problems with adding an out-of-tree backend, and what's some work to make it better?

Further reading: