Contribute Media
A thank you to everyone who makes this possible: Read More

Extending PyTorch with Custom Python/C++/CUDA Operators

Description

In this talk, we'll go over the new recommended APIs to extend PyTorch with custom Python/C++/CUDA operators. Users have been able to extend PyTorch with custom operators for years but we have updated our guidance for creating custom operators that compose with torch.compile, autograd, and other PyTorch subsystems.

Details

Improve this page