Convolutions and more as einsum for PyTorch
A small automatic differentiation engine, supporting higher-order derivatives