WIAS Preprint No. 2996, (2023)

WaveTrain: A Python package for numerical quantum mechanics of chain-like systems based on tensor trains


  • Riedel, Jerome
    ORCID: 0000-0003-1361-715X
  • Gelß, Patrick
    ORCID: 0000-0003-0120-1413
  • Klein, Rupert
  • Schmidt, Burkhard
    ORCID: 0000-0002-9658-499X

2020 Mathematics Subject Classification

  • 35-04 65-04 65M70 65P10 81Q05

2008 Physics and Astronomy Classification Scheme

  • 02.30.Jr 02.60.Cb 03.65.Ge 07.05.Tp 63.20.D- 63.20.kd 71.35.Aa


  • Numerical quantum mechanics, Schrödinger equations, Python software, tensor train decompositions, alternating linear scheme, Wielandt deflation, symplectic integrators




WaveTrain is an open-source software for numerical simulations of chain-like quantum systems with nearest-neighbor (NN) interactions only. The Python package is centered around tensor train (TT, or matrix product) format representations of Hamiltonian operators and (stationary or time-evolving) state vectors. It builds on the Python tensor train toolbox Scikit_tt, which provides efficient construction methods and storage schemes for the TT format. Its solvers for eigenvalue problems and linear differential equations are used in WaveTrain for the time-independent and time-dependent Schrödinger equations, respectively. Employing efficient decompositions to construct low-rank representations, the tensor-train ranks of state vectors are often found to depend only marginally on the chain length N. This results in the computational effort growing only slightly more than linearly with N, thus mitigating the curse of dimensionality. As a complement to the classes for full quantum mechanics, WaveTrain also contains classes for fully classical and mixed quantum-classical (Ehrenfest or mean field) dynamics of bipartite systems. The graphical capabilities allow visualization of quantum dynamics on the fly, with a choice of several different representations based on reduced density matrices. Even though developed for treating quasi one-dimensional excitonic energy transport in molecular solids or conjugated organic polymers, including coupling to phonons, WaveTrain can be used for any kind of chain-like quantum systems, with or without periodic boundary conditions, and with NN interactions only. The present work describes version 1.0 of our WaveTrain software, based on version 1.2 of scikit_tt, both of which are freely available from the GitHub platform where they will also be further developed. Moreover, WaveTrain is mirrored at SourceForge, within the framework of the WavePacket project for numerical quantum dynamics. Worked-out demonstration examples with complete input and output, including animated graphics, are available.

Appeared in

Download Documents