A Python GPU-accelerated solver for the Gross-Pitaevskii equation and applications to many-body cavity QED

Lorenzo Fioroni, Luca Gravina, Justyna Stefaniak +4

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 lin…