GPU-accelerated solutions of the nonlinear Schrödinger equation for simulating 2D spinor BECs
arXiv:2010.15069 · doi:10.1016/j.cpc.2022.108314
Abstract
As a first approximation beyond linearity, the nonlinear Schrödinger equation (NLSE) reliably describes a broad class of physical systems. Though numerical solutions of this model are well-established, these methods can be computationally complex. In this paper, we showcase a code development approach, demonstrating how computational time can be significantly reduced with readily available graphics processing unit (GPU) hardware and a straightforward code migration using open-source libraries. This process shows how CPU computations with power-law scaling in computation time with grid size can be made linear using GPUs. As a specific case study, we investigate the Gross-Pitaevskii equation, a specific version of the nonlinear Schrödinger model, as it describes in two dimensions a trapped, interacting, two-component Bose-Einstein condensate (BEC) subject to a spatially dependent interspin coupling, resulting in an analog to a spin-Hall system. This computational approach lets us probe high-resolution spatial features - revealing an interaction-dependent phase transition - all in a reasonable amount of time. Our computational approach is particularly relevant for research groups looking to easily accelerate straightforward numerical simulation of physical phenomena.
29 pages, 4 figures
References in corpus (7)
- Array Programming with NumPy
- Spin Hall effects for cold atoms in a light induced gauge potential
- Quantum Elliptic Vortex in a Nematic-Spin Bose-Einstein Condensate
- A Second-Order Distributed Trotter-Suzuki Solver with a Hybrid Kernel
- Spontaneous formation and non-equilibrium dynamics of a soliton-shaped Bose-Einstein condensate in a trap
- The nonlinear Schrödinger equation: A mathematical model with its wide-ranging applications
- Solving the spin-2 Gross-Pitaevskii equation using exact nonlinear dynamics and symplectic composition
Cited by in corpus (5)
- A Python GPU-accelerated solver for the Gross-Pitaevskii equation and applications to many-body cavity QED
- i-SPin 2: An integrator for general spin-s Gross-Pitaevskii systems
- PHOENIX -- Paderborn highly optimized and energy efficient solver for two-dimensional nonlinear Schrödinger equations with integrated extensions
- FFTArray: A Python Library for the Implementation of Discretized Multi-Dimensional Fourier Transforms
- Competitive algorithms for calculating the ground state properties of Bose-Fermi mixtures