Merge pull request #230 from astro-informatics/map/risbo-precompute-transform

Add stable high-spin transforms (precompute, standard)

Author: CosmoMatt

README.md

@@ -5,10 +5,9 @@
 [![MIT License](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 [![PyPI package](https://badge.fury.io/py/s2fft.svg)](https://badge.fury.io/py/s2fft)
 [![arXiv](http://img.shields.io/badge/arXiv-2311.14670-orange.svg?style=flat)](https://arxiv.org/abs/2311.14670)<!-- ALL-CONTRIBUTORS-BADGE:START - Do not remove or modify this section -->
-[![All Contributors](https://img.shields.io/badge/all_contributors-11-orange.svg?style=flat-square)](#contributors-)
-<!-- ALL-CONTRIBUTORS-BADGE:END --> 
+[![All Contributors](https://img.shields.io/badge/all_contributors-11-orange.svg?style=flat-square)](#contributors-)<!-- ALL-CONTRIBUTORS-BADGE:END --> 
 [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/astro-informatics/s2fft/blob/main/notebooks/spherical_harmonic_transform.ipynb)
-<!-- [![image](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black) -->
+[![Linter](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
 
 <img align="left" height="85" width="98" src="./docs/assets/sax_logo.png">
 

docs/api/precompute_transforms/index.rst

@@ -60,6 +60,10 @@ Precompute Functions
      - Builds a kernel including quadrature weights and Wigner-D coefficients for spherical harmonic transform.
    * - :func:`~s2fft.precompute_transforms.construct.wigner_kernel`
      - Builds a kernel including quadrature weights and Wigner-D coefficients for Wigner transform.
+   * - :func:`~s2fft.precompute_transforms.construct.spin_spherical_kernel_jax`
+     - Builds a kernel including quadrature weights and Wigner-D coefficients for spherical harmonic transform (JAX).
+   * - :func:`~s2fft.precompute_transforms.construct.wigner_kernel_jax`
+     - Builds a kernel including quadrature weights and Wigner-D coefficients for Wigner transform (JAX).
    * - :func:`~s2fft.precompute_transforms.construct.healpix_phase_shifts`
      - Builds a vector of corresponding phase shifts for each HEALPix latitudinal ring.
 
@@ -69,6 +73,7 @@ Precompute Functions
    :caption: Precompute Transforms
 
    construct
+   alt_construct
    spin_spherical 
    wigner