Optimization and Parallelization of A Numerical Gravitational-Wave Model


Next-generation gravitational-wave (GW) detectors, such as the Laser Interferometer Space Antenna (LISA), will detect GW signals from extreme mass-ratio inspirals. High fidelity and fast GW models are essential for achieving the full scientific potential of LISA. We have developed a high-accuracy, data-driven (surrogate) model for LISA-type sources. The code is currently in a Jupyter notebook, but to enable data analysis studies, we require the model to operate as an optimized, stand-alone library. This project aims to accomplish this goal by porting the model into two publicly available, community-driven packages GWSurrogate and the Black Hole Perturbation Toolkit. In this project, the student will port the model to these existing codebases before optimizing. The model data will be stored in HDF5 file format. One of the main computational bottlenecks is likely to be the large matrix-vector multiplication required to compute each harmonic mode. The student will explore offloading this cost to a GPU through the cupy package and parallelization over mode computations. Code profiling will also be carried out to identify other parts of the code that could benefit from further optimizations.