expsolve - a PyTorch compatible Differentiable Numerical Algorithms package for computational quantum mechanics