s2fft icon indicating copy to clipboard operation
s2fft copied to clipboard

Differentiable and accelerated spherical transforms with JAX

Results 79 s2fft issues
Sort by recently updated
recently updated
newest added

Trying to finish up my PR https://github.com/astro-informatics/s2fft/pull/204 I ran to an issue ``` File "", line 1, in File "/home/wassim/Projects/CMB/s2fft/s2fft/__init__.py", line 2, in from .transforms import wigner File "/home/wassim/Projects/CMB/s2fft/s2fft/transforms/__init__.py", line...

blocker

Quick roadmap on the multi GPU healpix FFT For the single GPU one - [ ] Merge first PR https://github.com/astro-informatics/s2fft/pull/204 that uses CUFFT as backend for FFTs and does the...

enhancement

Creating a pure cuda healpix fft to avoid the long compile time when using pure JAX

We don't currently specify the code style convention we want the package code to adhere to. Ideally we should settle on something and document this to make it easy for...

enhancement
question
clean up
infrastructure

Some updates to #204 Still need to resolve: - [x] Resolving licensing for external code in `lib/include/kernel_helpers.h` [(link to comment)](https://github.com/astro-informatics/s2fft/pull/204#discussion_r1638357861) - [x] Adjusting license in header for `lib/include/kernel_nanobind_helpers.h` to list...