Tutorial for building a custom CUDA function for Pytorch - View it on GitHub