Convert Julia functions to PyTorch autograd functions - View it on GitHub