/jax-healpy

Healpy with JAX

Primary LanguagePython

Healpy with JAX

This project intends to assess the interest of implementing healpy functions using JAX.

WARNING: BETA STAGE!!!

Benchmark

Installation

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/s2fft@0.0.1"