This project intends to assess the interest of implementing healpy functions using JAX.
WARNING: BETA STAGE!!!
pip install -U "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
pip install jax-healpy
pip install "s2fft @ git+https://github.com/astro-informatics/[email protected]"