pip install expsolve
A PyTorch compatible Differentiable Numerical Algorithms package for computational quantum mechanics.
Includes time-propagators based on splitting and composition methods, exponential integrators, Magnus expansion etc.
Current functionality is limited to Fourier discretisation in space.
Please see examples directory for tutorial notebooks.