Examples of C extensions for PyTorch - View it on GitHub