Healpy with JAX

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

WARNING: BETA STAGE!!!

Installation

First install JAX following the documentation.

Then install the package with:

pip install jax-healpy

To use the spherical harmonics functions, you will need s2fft, part of the recommended dependencies:

pip install jax-healpy[recommended]

Execution time measured on the Jean Zay supercomputer.

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
benchmarks		benchmarks
docs/benchmarks		docs/benchmarks
jax_healpy		jax_healpy
tests		tests
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
README.md		README.md
pyproject.toml		pyproject.toml