Skip to content

v1.1.0

Choose a tag to compare

View all tags
@CosmoMatt CosmoMatt released this 09 Apr 07:52
· 267 commits to main since this release
v1.1.0
76fa862
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

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
Assets 2
Loading