s2fft
s2fft copied to clipboard
Differentiable and accelerated spherical transforms with JAX
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...
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...
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...
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...