WaveTrain is an open-source software for numerical simulations of chain-like quantum systems with nearest-neighbor (NN) interactions only (with or without periodic boundary conditions). This Python package is centered around tensor train (TT, or matrix product) representations of quantum-mechanical Hamiltonian operators and (stationary or time-evolving) state vectors. WaveTrain builds on the Python tensor train toolbox scikit_tt, which provides efficient construction methods, storage schemes, as well as solvers for eigenvalue problems and linear differential equations in the TT format.
WaveTrain comprises solvers for time-independent and time-dependent Schrödinger equations employing TT decompositions to construct low-rank representations. Often, the TT ranks of state vectors are found to depend only marginally on the chain length N, which results in the computational effort growing only slightly more than linearly in N, thus mitigating the curse of dimensionality.
Features
- Time-dependent/-independent Schroedinger equation
- Quantum-classical dynamics (Ehrenfest aka mean field)
- Chain-like systems with nearest-neighbor interactions only
- Low-rank tensor train representations of state vectors
- Demonstration examples with animated graphics
- Latest version 1.0.7 released on 11-Feb 2023
- See also our WaveTrain publication at https://doi.org/10.1063/5.0147314