Simple example project that builds a PyTorch CUDA extension - View it on GitHub