ASCL.net

Astrophysics Source Code Library

Making codes discoverable since 1999

ASCL Code Record

[ascl:2404.027] s2fft: Differentiable and accelerated spherical transforms

S2FFT computes Fourier transforms on the sphere and rotation group using JAX (ascl:2111.002) or PyTorch. It leverages autodiff to provide differentiable transforms, which are also deployable on hardware accelerators (e.g., GPUs and TPUs). More specifically, S2FFT provides support for spin spherical harmonic and Wigner transforms (for both real and complex signals), with support for adjoint transformations where needed, and comes with different optimisations (precompute or not) that one may select depending on available resources and desired angular resolution L.

Code site:
https://github.com/astro-informatics/s2fft
Described in:
https://ui.adsabs.harvard.edu/abs/2023arXiv231114670P
Bibcode:
2024ascl.soft04027P
Preferred citation method:

Please see citation information here: https://github.com/astro-informatics/s2fft?tab=readme-ov-file#attribution-books


Views: 762

ascl:2404.027
Add this shield to your page
Copy the above HTML to add this shield to your code's website.