Skip to content

Releases: astro-informatics/s2fft

v1.1.0

09 Apr 07:52
76fa862
Compare
Choose a tag to compare

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

05 Mar 12:37
b3d033c
Compare
Choose a tag to compare

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.

v1.0.1

13 Dec 16:31
c608309
Compare
Choose a tag to compare

This is a relatively small patch which attempts to mitigate the compile time of HEALPix transforms. Though this is unlikely to have solved the overall scaling of the compile time, it has reduced the time by a factor of ~4 which should help.

V1.0.0

27 Nov 10:02
3444a42
Compare
Choose a tag to compare

This is the initial beta release to accompany the associated paper (see badges). In addition this version is deployed on PyPi with universal wheels which should be straightforward to install for various python versions and machine architectures.

0.0.1

15 Feb 19:30
Compare
Choose a tag to compare

This is a pre-alpha release version for use by collaborators. Most functionality is in place, but development of minor additions will be ongoing. Please submit an issue if you have one and we will respond asap!