PyTorch 2 Internals

· January 8, 2024

(Updated: Nov 24, 2024)

A wide ranging talk that includes a lot of the new work on ExecuTorch for on-device support, including the torch.export stack. The talk really goes through the whole flow from graph capture, lowering and optimizations so its a great way to get a sense for what is happening within PyTorch!

https://blog.christianperone.com/2023/12/pytorch-2-internals-talk/