Releases: astro-informatics/s2fft
Releases · astro-informatics/s2fft
v1.1.0
This minor release of S2FFT
aims to increase the accessibility of differentiable harmonic transforms to users without easy access to GPU compute resources. We provide custom JAX frontends for existing CPU bound C/C++ spherical harmonic libraries, at this point we capture the functionality of SSHT
and HEALPix
though in principle any spherical harmonics could easily be integrated.
Tip
For details on this approach see the original derivation in section 5.3.1 and 5.3.2 of Price & McEwen 2023.
Main changelog:
- JAX frontend support for HEALPix C++ library
- JAX frontend support for SSHT C library
- Reverse mode gradients for above validated against finite difference
v1.0.2
Incremental version of S2FFT
which adds support for:
PyTorch
precompute transforms- Gauss-Legendre sampling schemes
- JAX harmonic space rotation functions
- JAX Risbo recursions
This version also implements some changes to reduce the compile time of HEALPix Fast Fourier transforms, though this issues is yet to be fully solved.