A Python GPU-accelerated solver for the Gross-Pitaevskii equation and applications to many-body cavity QED
arXiv:2404.14401 · doi:10.21468/SciPostPhysCodeb.38
Abstract
TorchGPE is a general-purpose Python package developed for solving the Gross-Pitaevskii equation (GPE). This solver is designed to integrate wave functions across a spectrum of linear and non-linear potentials. A distinctive aspect of TorchGPE is its modular approach, which allows the incorporation of arbitrary self-consistent and time-dependent potentials, e.g., those relevant in many-body cavity QED models. The package employs a symmetric split-step Fourier propagation method, effective in both real and imaginary time. In our work, we demonstrate a significant improvement in computational efficiency by leveraging GPU computing capabilities. With the integration of the latter technology, TorchGPE achieves a substantial speed-up with respect to conventional CPU-based methods, greatly expanding the scope and potential of research in this field.
References in corpus (20)
- Theory of Bose-Einstein condensation in trapped gases
- The Dicke Quantum Phase Transition with a Superfluid Gas in an Optical Cavity
- Effective wave-equations for the dynamics of cigar-shaped and disc-shaped Bose condensates
- Supersolid formation in a quantum gas breaking continuous translational symmetry
- Fortran programs for the time-dependent Gross-Pitaevskii equation in a fully anisotropic trap
- A Bose-Einstein condensate in an optical lattice
- Cavity QED with Quantum Gases: New Paradigms in Many-Body Physics
- Roton-type mode softening in a quantum gas with cavity-mediated long-range interactions
- Observation of Roton Mode Population in a Dipolar Quantum Gas
- Monitoring and manipulating Higgs and Goldstone modes in a supersolid quantum gas
- A primer on quantum fluids
- Split-step Fourier methods for the Gross-Pitaevskii equation
- Self-oscillating pump in a topological dissipative atom-cavity system
- Density Fluctuations across the Superfluid-Supersolid Phase Transition in a Dipolar Quantum Gas
- Dynamic structure factor of ultracold Bose and Fermi gases in optical lattices
- Measuring the dynamics of a first order structural phase transition between two configurations of a superradiant crystal
- A Second-Order Distributed Trotter-Suzuki Solver with a Hybrid Kernel
- Vortex lattice in a uniform Bose-Einstein condensate in a box trap
- GPU-accelerated solutions of the nonlinear Schrödinger equation for simulating 2D spinor BECs
- Dimensional reduction and localization of a Bose-Einstein condensate in a quasi-1D bichromatic optical lattice
Cited by in corpus (7)
- Numerical solution of nonlinear Schrödinger equation by a hybrid pseudospectral-variational quantum algorithm
- PHOENIX -- Paderborn highly optimized and energy efficient solver for two-dimensional nonlinear Schrödinger equations with integrated extensions
- Solving the Gross-Pitaevskii equation on multiple different scales using the quantics tensor train representation
- FFTArray: A Python Library for the Implementation of Discretized Multi-Dimensional Fourier Transforms
- Bandstructure of a coupled BEC-cavity system: effects of dissipation and geometry
- Tunable optical lattices for the creation of matter-wave lattice solitons
- Synchronization of Quasi-Particle Excitations in a Quantum Gas with Cavity-Mediated Interactions