Backend extensibility

PyTorch Developer Podcast - A podcast by Edward Yang, Team PyTorch

Categories:

Technology

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?