Add support for the PyTorch C++ API. This can be used for setting up models, autograd, optimization and GPU acceleration