Magdy Morshed
(INFN Ferrara)
11/20/24, 9:50 AM
The Healpy package is widely used in cosmology, both for the handling of the HEALPix pixelization and the associated spherical harmonic transforms. The current Healpy package implementation is not JAX friendly, as the spherical harmonics implementation is based in C++, which prevents us to possibly accelerate the associated routines using JIT or to run them on GPUs. In this talk, we will...