Example repository for custom C++/CUDA operators for TorchScript - View it on GitHub