TY  - JOUR
PB  - SciPost Foundation
DO  - 10.21468/SciPostPhysCodeb.38-r1.0
TI  - Codebase release 1.0 for TorchGPE
PY  - 2024/11/06
UR  - https://scipost.org/SciPostPhysCodeb.38-r1.0
JF  - SciPost Physics Codebases
JA  - SciPost Phys. Codebases
SP  - 38-r1.0
A1  - Fioroni, Lorenzo
AU  - Gravina, Luca
AU  - Stefaniak, Justyna
AU  - Baumgärtner, Alexander
AU  - Finger, Fabian
AU  - Dreon, Davide
AU  - Donner, Tobias
AB  - 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.
ER  -

@Article{10.21468/SciPostPhysCodeb.38,
	title={{A Python GPU-accelerated solver for the Gross-Pitaevskii equation and applications to many-body cavity QED}},
	author={Lorenzo Fioroni and Luca Gravina and Justyna Stefaniak and Alexander Baumgärtner and Fabian Finger and Davide Dreon and Tobias Donner},
	journal={SciPost Phys. Codebases},
	pages={38},
	year={2024},
	publisher={SciPost},
	doi={10.21468/SciPostPhysCodeb.38},
	url={https://scipost.org/10.21468/SciPostPhysCodeb.38},
}

@Article{10.21468/SciPostPhysCodeb.38-r1.0,
	title={{Codebase release 1.0 for TorchGPE}},
	author={Lorenzo Fioroni and Luca Gravina and Justyna Stefaniak and Alexander Baumgärtner and Fabian Finger and Davide Dreon and Tobias Donner},
	journal={SciPost Phys. Codebases},
	pages={38-r1.0},
	year={2024},
	publisher={SciPost},
	doi={10.21468/SciPostPhysCodeb.38-r1.0},
	url={https://scipost.org/10.21468/SciPostPhysCodeb.38-r1.0},
}