├── .all-contributorsrc ├── .clang-format ├── .coveragerc ├── .github ├── dependabot.yml ├── release.yml └── workflows │ ├── build.yml │ ├── docs.yml │ ├── linting.yml │ ├── stats.yml │ └── tests.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CITATION.cff ├── CMakeLists.txt ├── LICENCE.txt ├── MANIFEST.in ├── README.md ├── benchmarks ├── README.md ├── __init__.py ├── benchmarking.py ├── plotting.py ├── precompute_spherical.py ├── precompute_wigner.py ├── spherical.py └── wigner.py ├── codecov.yml ├── docs ├── Makefile ├── _static │ ├── MIT_Licence.png │ ├── arxiv-logomark-small.png │ ├── codecov.png │ ├── css │ │ ├── custom.css │ │ └── custom_tabs.css │ └── pypi.png ├── api │ ├── base_transforms │ │ ├── index.rst │ │ ├── spin_spherical_transform.rst │ │ └── wigner_transform.rst │ ├── index.rst │ ├── precompute_transforms │ │ ├── construct.rst │ │ ├── custom_ops.rst │ │ ├── fourier_wigner.rst │ │ ├── index.rst │ │ ├── spin_spherical.rst │ │ └── wigner.rst │ ├── recursions │ │ ├── index.rst │ │ ├── price_mcewen.rst │ │ ├── risbo.rst │ │ ├── risbo_jax.rst │ │ ├── trapani.rst │ │ ├── turok.rst │ │ └── turok_jax.rst │ ├── sampling │ │ ├── index.rst │ │ ├── spherical_samples.rst │ │ └── wigner_samples.rst │ ├── transforms │ │ ├── c_backend_spherical.rst │ │ ├── index.rst │ │ ├── on_the_fly_recursions.rst │ │ ├── spin_spherical_transform.rst │ │ └── wigner.rst │ └── utility │ │ ├── healpix_ffts.rst │ │ ├── index.rst │ │ ├── quadrature.rst │ │ ├── quadrature_jax.rst │ │ ├── quadrature_torch.rst │ │ ├── resampling.rst │ │ ├── resampling_jax.rst │ │ ├── resampling_torch.rst │ │ ├── rotation.rst │ │ ├── signal_generator.rst │ │ └── utils.rst ├── assets │ ├── authors │ │ ├── gopinathan.jpeg │ │ ├── graham.jpeg │ │ ├── mcewen.jpeg │ │ ├── minano.jpeg │ │ └── price.jpeg │ ├── figures │ │ ├── Wigner_recursion_github_docs.png │ │ ├── Wigner_recursion_legend_darkmode.png │ │ ├── sax_schematic_github_docs.png │ │ ├── sax_schematic_legend_darkmode.png │ │ ├── schematic.png │ │ ├── software_overview.png │ │ ├── spherical_sampling.png │ │ ├── spin_spherical_mw.pdf │ │ ├── spin_spherical_mw.png │ │ ├── wigner_mw.pdf │ │ └── wigner_mw.png │ ├── sax_logo.png │ └── sax_logo.svg ├── conf.py ├── index.rst └── tutorials │ ├── JAX_HEALPix │ └── JAX_HEALPix_frontend.nblink │ ├── JAX_SSHT │ └── JAX_SSHT_frontend.nblink │ ├── index.rst │ ├── rotation │ └── rotation.nblink │ ├── spherical_harmonic │ └── spherical_harmonic_transform.nblink │ ├── torch_frontend │ └── torch_frontend.nblink │ └── wigner │ └── wigner_transform.nblink ├── lib ├── include │ ├── hresult.h │ ├── kernel_helpers.h │ ├── kernel_nanobind_helpers.h │ ├── plan_cache.h │ ├── s2fft.h │ ├── s2fft_callbacks.h │ └── s2fft_kernels.h └── src │ ├── extensions.cc │ ├── plan_cache.cc │ ├── s2fft.cu │ ├── s2fft_callbacks.cu │ └── s2fft_kernels.cu ├── notebooks ├── JAX_CUDA_HEALPix.ipynb ├── JAX_HEALPix_frontend.ipynb ├── JAX_SSHT_frontend.ipynb ├── data │ └── Gaia_EDR3_flux.npy ├── plotting_functions.py ├── spherical_harmonic_transform.ipynb ├── spherical_rotation.ipynb ├── torch_frontend.ipynb └── wigner_transform.ipynb ├── pyproject.toml ├── s2fft ├── README.rst ├── __init__.py ├── base_transforms │ ├── __init__.py │ ├── spherical.py │ └── wigner.py ├── precompute_transforms │ ├── __init__.py │ ├── construct.py │ ├── custom_ops.py │ ├── fourier_wigner.py │ ├── spherical.py │ └── wigner.py ├── recursions │ ├── __init__.py │ ├── price_mcewen.py │ ├── risbo.py │ ├── risbo_jax.py │ ├── trapani.py │ ├── turok.py │ └── turok_jax.py ├── sampling │ ├── __init__.py │ ├── reindex.py │ ├── s2_samples.py │ └── so3_samples.py ├── transforms │ ├── __init__.py │ ├── c_backend_spherical.py │ ├── otf_recursions.py │ ├── spherical.py │ └── wigner.py └── utils │ ├── __init__.py │ ├── healpix_ffts.py │ ├── iterative_refinement.py │ ├── jax_primitive.py │ ├── quadrature.py │ ├── quadrature_jax.py │ ├── quadrature_torch.py │ ├── resampling.py │ ├── resampling_jax.py │ ├── resampling_torch.py │ ├── rotation.py │ ├── signal_generator.py │ └── torch_wrapper.py └── tests ├── __init__.py ├── conftest.py ├── test_fourier_wigner.py ├── test_healpix_ffts.py ├── test_lifting_transforms.py ├── test_quadrature.py ├── test_resampling.py ├── test_samples.py ├── test_signal_generator.py ├── test_spherical_base.py ├── test_spherical_custom_grads.py ├── test_spherical_precompute.py ├── test_spherical_transform.py ├── test_torch_wrapper.py ├── test_utils.py ├── test_wigner_base.py ├── test_wigner_custom_grads.py ├── test_wigner_precompute.py ├── test_wigner_recursions.py ├── test_wigner_samples.py └── test_wigner_transform.py /.all-contributorsrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.all-contributorsrc -------------------------------------------------------------------------------- /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.clang-format -------------------------------------------------------------------------------- /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.coveragerc -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.github/release.yml -------------------------------------------------------------------------------- /.github/workflows/build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.github/workflows/build.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.github/workflows/linting.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.github/workflows/linting.yml -------------------------------------------------------------------------------- /.github/workflows/stats.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.github/workflows/stats.yml -------------------------------------------------------------------------------- /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/CITATION.cff -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENCE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/LICENCE.txt -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/benchmarks/README.md -------------------------------------------------------------------------------- /benchmarks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/benchmarking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/benchmarks/benchmarking.py -------------------------------------------------------------------------------- /benchmarks/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/benchmarks/plotting.py -------------------------------------------------------------------------------- /benchmarks/precompute_spherical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/benchmarks/precompute_spherical.py -------------------------------------------------------------------------------- /benchmarks/precompute_wigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/benchmarks/precompute_wigner.py -------------------------------------------------------------------------------- /benchmarks/spherical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/benchmarks/spherical.py -------------------------------------------------------------------------------- /benchmarks/wigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/benchmarks/wigner.py -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/codecov.yml -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_static/MIT_Licence.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/_static/MIT_Licence.png -------------------------------------------------------------------------------- /docs/_static/arxiv-logomark-small.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/_static/arxiv-logomark-small.png -------------------------------------------------------------------------------- /docs/_static/codecov.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/_static/codecov.png -------------------------------------------------------------------------------- /docs/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/_static/css/custom.css -------------------------------------------------------------------------------- /docs/_static/css/custom_tabs.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/_static/css/custom_tabs.css -------------------------------------------------------------------------------- /docs/_static/pypi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/_static/pypi.png -------------------------------------------------------------------------------- /docs/api/base_transforms/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/base_transforms/index.rst -------------------------------------------------------------------------------- /docs/api/base_transforms/spin_spherical_transform.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/base_transforms/spin_spherical_transform.rst -------------------------------------------------------------------------------- /docs/api/base_transforms/wigner_transform.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/base_transforms/wigner_transform.rst -------------------------------------------------------------------------------- /docs/api/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/index.rst -------------------------------------------------------------------------------- /docs/api/precompute_transforms/construct.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/precompute_transforms/construct.rst -------------------------------------------------------------------------------- /docs/api/precompute_transforms/custom_ops.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/precompute_transforms/custom_ops.rst -------------------------------------------------------------------------------- /docs/api/precompute_transforms/fourier_wigner.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/precompute_transforms/fourier_wigner.rst -------------------------------------------------------------------------------- /docs/api/precompute_transforms/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/precompute_transforms/index.rst -------------------------------------------------------------------------------- /docs/api/precompute_transforms/spin_spherical.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/precompute_transforms/spin_spherical.rst -------------------------------------------------------------------------------- /docs/api/precompute_transforms/wigner.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/precompute_transforms/wigner.rst -------------------------------------------------------------------------------- /docs/api/recursions/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/recursions/index.rst -------------------------------------------------------------------------------- /docs/api/recursions/price_mcewen.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/recursions/price_mcewen.rst -------------------------------------------------------------------------------- /docs/api/recursions/risbo.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/recursions/risbo.rst -------------------------------------------------------------------------------- /docs/api/recursions/risbo_jax.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/recursions/risbo_jax.rst -------------------------------------------------------------------------------- /docs/api/recursions/trapani.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/recursions/trapani.rst -------------------------------------------------------------------------------- /docs/api/recursions/turok.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/recursions/turok.rst -------------------------------------------------------------------------------- /docs/api/recursions/turok_jax.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/recursions/turok_jax.rst -------------------------------------------------------------------------------- /docs/api/sampling/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/sampling/index.rst -------------------------------------------------------------------------------- /docs/api/sampling/spherical_samples.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/sampling/spherical_samples.rst -------------------------------------------------------------------------------- /docs/api/sampling/wigner_samples.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/sampling/wigner_samples.rst -------------------------------------------------------------------------------- /docs/api/transforms/c_backend_spherical.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/transforms/c_backend_spherical.rst -------------------------------------------------------------------------------- /docs/api/transforms/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/transforms/index.rst -------------------------------------------------------------------------------- /docs/api/transforms/on_the_fly_recursions.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/transforms/on_the_fly_recursions.rst -------------------------------------------------------------------------------- /docs/api/transforms/spin_spherical_transform.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/transforms/spin_spherical_transform.rst -------------------------------------------------------------------------------- /docs/api/transforms/wigner.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/transforms/wigner.rst -------------------------------------------------------------------------------- /docs/api/utility/healpix_ffts.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/healpix_ffts.rst -------------------------------------------------------------------------------- /docs/api/utility/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/index.rst -------------------------------------------------------------------------------- /docs/api/utility/quadrature.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/quadrature.rst -------------------------------------------------------------------------------- /docs/api/utility/quadrature_jax.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/quadrature_jax.rst -------------------------------------------------------------------------------- /docs/api/utility/quadrature_torch.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/quadrature_torch.rst -------------------------------------------------------------------------------- /docs/api/utility/resampling.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/resampling.rst -------------------------------------------------------------------------------- /docs/api/utility/resampling_jax.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/resampling_jax.rst -------------------------------------------------------------------------------- /docs/api/utility/resampling_torch.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/resampling_torch.rst -------------------------------------------------------------------------------- /docs/api/utility/rotation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/rotation.rst -------------------------------------------------------------------------------- /docs/api/utility/signal_generator.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/signal_generator.rst -------------------------------------------------------------------------------- /docs/api/utility/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/api/utility/utils.rst -------------------------------------------------------------------------------- /docs/assets/authors/gopinathan.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/authors/gopinathan.jpeg -------------------------------------------------------------------------------- /docs/assets/authors/graham.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/authors/graham.jpeg -------------------------------------------------------------------------------- /docs/assets/authors/mcewen.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/authors/mcewen.jpeg -------------------------------------------------------------------------------- /docs/assets/authors/minano.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/authors/minano.jpeg -------------------------------------------------------------------------------- /docs/assets/authors/price.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/authors/price.jpeg -------------------------------------------------------------------------------- /docs/assets/figures/Wigner_recursion_github_docs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/Wigner_recursion_github_docs.png -------------------------------------------------------------------------------- /docs/assets/figures/Wigner_recursion_legend_darkmode.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/Wigner_recursion_legend_darkmode.png -------------------------------------------------------------------------------- /docs/assets/figures/sax_schematic_github_docs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/sax_schematic_github_docs.png -------------------------------------------------------------------------------- /docs/assets/figures/sax_schematic_legend_darkmode.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/sax_schematic_legend_darkmode.png -------------------------------------------------------------------------------- /docs/assets/figures/schematic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/schematic.png -------------------------------------------------------------------------------- /docs/assets/figures/software_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/software_overview.png -------------------------------------------------------------------------------- /docs/assets/figures/spherical_sampling.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/spherical_sampling.png -------------------------------------------------------------------------------- /docs/assets/figures/spin_spherical_mw.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/spin_spherical_mw.pdf -------------------------------------------------------------------------------- /docs/assets/figures/spin_spherical_mw.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/spin_spherical_mw.png -------------------------------------------------------------------------------- /docs/assets/figures/wigner_mw.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/wigner_mw.pdf -------------------------------------------------------------------------------- /docs/assets/figures/wigner_mw.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/figures/wigner_mw.png -------------------------------------------------------------------------------- /docs/assets/sax_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/sax_logo.png -------------------------------------------------------------------------------- /docs/assets/sax_logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/assets/sax_logo.svg -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/tutorials/JAX_HEALPix/JAX_HEALPix_frontend.nblink: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/tutorials/JAX_HEALPix/JAX_HEALPix_frontend.nblink -------------------------------------------------------------------------------- /docs/tutorials/JAX_SSHT/JAX_SSHT_frontend.nblink: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/tutorials/JAX_SSHT/JAX_SSHT_frontend.nblink -------------------------------------------------------------------------------- /docs/tutorials/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/tutorials/index.rst -------------------------------------------------------------------------------- /docs/tutorials/rotation/rotation.nblink: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/tutorials/rotation/rotation.nblink -------------------------------------------------------------------------------- /docs/tutorials/spherical_harmonic/spherical_harmonic_transform.nblink: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/tutorials/spherical_harmonic/spherical_harmonic_transform.nblink -------------------------------------------------------------------------------- /docs/tutorials/torch_frontend/torch_frontend.nblink: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/tutorials/torch_frontend/torch_frontend.nblink -------------------------------------------------------------------------------- /docs/tutorials/wigner/wigner_transform.nblink: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/docs/tutorials/wigner/wigner_transform.nblink -------------------------------------------------------------------------------- /lib/include/hresult.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/include/hresult.h -------------------------------------------------------------------------------- /lib/include/kernel_helpers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/include/kernel_helpers.h -------------------------------------------------------------------------------- /lib/include/kernel_nanobind_helpers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/include/kernel_nanobind_helpers.h -------------------------------------------------------------------------------- /lib/include/plan_cache.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/include/plan_cache.h -------------------------------------------------------------------------------- /lib/include/s2fft.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/include/s2fft.h -------------------------------------------------------------------------------- /lib/include/s2fft_callbacks.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/include/s2fft_callbacks.h -------------------------------------------------------------------------------- /lib/include/s2fft_kernels.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/include/s2fft_kernels.h -------------------------------------------------------------------------------- /lib/src/extensions.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/src/extensions.cc -------------------------------------------------------------------------------- /lib/src/plan_cache.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/src/plan_cache.cc -------------------------------------------------------------------------------- /lib/src/s2fft.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/src/s2fft.cu -------------------------------------------------------------------------------- /lib/src/s2fft_callbacks.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/src/s2fft_callbacks.cu -------------------------------------------------------------------------------- /lib/src/s2fft_kernels.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/lib/src/s2fft_kernels.cu -------------------------------------------------------------------------------- /notebooks/JAX_CUDA_HEALPix.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/notebooks/JAX_CUDA_HEALPix.ipynb -------------------------------------------------------------------------------- /notebooks/JAX_HEALPix_frontend.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/notebooks/JAX_HEALPix_frontend.ipynb -------------------------------------------------------------------------------- /notebooks/JAX_SSHT_frontend.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/notebooks/JAX_SSHT_frontend.ipynb -------------------------------------------------------------------------------- /notebooks/data/Gaia_EDR3_flux.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/notebooks/data/Gaia_EDR3_flux.npy -------------------------------------------------------------------------------- /notebooks/plotting_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/notebooks/plotting_functions.py -------------------------------------------------------------------------------- /notebooks/spherical_harmonic_transform.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/notebooks/spherical_harmonic_transform.ipynb -------------------------------------------------------------------------------- /notebooks/spherical_rotation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/notebooks/spherical_rotation.ipynb -------------------------------------------------------------------------------- /notebooks/torch_frontend.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/notebooks/torch_frontend.ipynb -------------------------------------------------------------------------------- /notebooks/wigner_transform.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/notebooks/wigner_transform.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/pyproject.toml -------------------------------------------------------------------------------- /s2fft/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/README.rst -------------------------------------------------------------------------------- /s2fft/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/__init__.py -------------------------------------------------------------------------------- /s2fft/base_transforms/__init__.py: -------------------------------------------------------------------------------- 1 | from . import spherical, wigner 2 | -------------------------------------------------------------------------------- /s2fft/base_transforms/spherical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/base_transforms/spherical.py -------------------------------------------------------------------------------- /s2fft/base_transforms/wigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/base_transforms/wigner.py -------------------------------------------------------------------------------- /s2fft/precompute_transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/precompute_transforms/__init__.py -------------------------------------------------------------------------------- /s2fft/precompute_transforms/construct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/precompute_transforms/construct.py -------------------------------------------------------------------------------- /s2fft/precompute_transforms/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/precompute_transforms/custom_ops.py -------------------------------------------------------------------------------- /s2fft/precompute_transforms/fourier_wigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/precompute_transforms/fourier_wigner.py -------------------------------------------------------------------------------- /s2fft/precompute_transforms/spherical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/precompute_transforms/spherical.py -------------------------------------------------------------------------------- /s2fft/precompute_transforms/wigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/precompute_transforms/wigner.py -------------------------------------------------------------------------------- /s2fft/recursions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/recursions/__init__.py -------------------------------------------------------------------------------- /s2fft/recursions/price_mcewen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/recursions/price_mcewen.py -------------------------------------------------------------------------------- /s2fft/recursions/risbo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/recursions/risbo.py -------------------------------------------------------------------------------- /s2fft/recursions/risbo_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/recursions/risbo_jax.py -------------------------------------------------------------------------------- /s2fft/recursions/trapani.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/recursions/trapani.py -------------------------------------------------------------------------------- /s2fft/recursions/turok.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/recursions/turok.py -------------------------------------------------------------------------------- /s2fft/recursions/turok_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/recursions/turok_jax.py -------------------------------------------------------------------------------- /s2fft/sampling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/sampling/__init__.py -------------------------------------------------------------------------------- /s2fft/sampling/reindex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/sampling/reindex.py -------------------------------------------------------------------------------- /s2fft/sampling/s2_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/sampling/s2_samples.py -------------------------------------------------------------------------------- /s2fft/sampling/so3_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/sampling/so3_samples.py -------------------------------------------------------------------------------- /s2fft/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/transforms/__init__.py -------------------------------------------------------------------------------- /s2fft/transforms/c_backend_spherical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/transforms/c_backend_spherical.py -------------------------------------------------------------------------------- /s2fft/transforms/otf_recursions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/transforms/otf_recursions.py -------------------------------------------------------------------------------- /s2fft/transforms/spherical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/transforms/spherical.py -------------------------------------------------------------------------------- /s2fft/transforms/wigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/transforms/wigner.py -------------------------------------------------------------------------------- /s2fft/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/__init__.py -------------------------------------------------------------------------------- /s2fft/utils/healpix_ffts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/healpix_ffts.py -------------------------------------------------------------------------------- /s2fft/utils/iterative_refinement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/iterative_refinement.py -------------------------------------------------------------------------------- /s2fft/utils/jax_primitive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/jax_primitive.py -------------------------------------------------------------------------------- /s2fft/utils/quadrature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/quadrature.py -------------------------------------------------------------------------------- /s2fft/utils/quadrature_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/quadrature_jax.py -------------------------------------------------------------------------------- /s2fft/utils/quadrature_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/quadrature_torch.py -------------------------------------------------------------------------------- /s2fft/utils/resampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/resampling.py -------------------------------------------------------------------------------- /s2fft/utils/resampling_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/resampling_jax.py -------------------------------------------------------------------------------- /s2fft/utils/resampling_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/resampling_torch.py -------------------------------------------------------------------------------- /s2fft/utils/rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/rotation.py -------------------------------------------------------------------------------- /s2fft/utils/signal_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/signal_generator.py -------------------------------------------------------------------------------- /s2fft/utils/torch_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/s2fft/utils/torch_wrapper.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_fourier_wigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_fourier_wigner.py -------------------------------------------------------------------------------- /tests/test_healpix_ffts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_healpix_ffts.py -------------------------------------------------------------------------------- /tests/test_lifting_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_lifting_transforms.py -------------------------------------------------------------------------------- /tests/test_quadrature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_quadrature.py -------------------------------------------------------------------------------- /tests/test_resampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_resampling.py -------------------------------------------------------------------------------- /tests/test_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_samples.py -------------------------------------------------------------------------------- /tests/test_signal_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_signal_generator.py -------------------------------------------------------------------------------- /tests/test_spherical_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_spherical_base.py -------------------------------------------------------------------------------- /tests/test_spherical_custom_grads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_spherical_custom_grads.py -------------------------------------------------------------------------------- /tests/test_spherical_precompute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_spherical_precompute.py -------------------------------------------------------------------------------- /tests/test_spherical_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_spherical_transform.py -------------------------------------------------------------------------------- /tests/test_torch_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_torch_wrapper.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_utils.py -------------------------------------------------------------------------------- /tests/test_wigner_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_wigner_base.py -------------------------------------------------------------------------------- /tests/test_wigner_custom_grads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_wigner_custom_grads.py -------------------------------------------------------------------------------- /tests/test_wigner_precompute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_wigner_precompute.py -------------------------------------------------------------------------------- /tests/test_wigner_recursions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_wigner_recursions.py -------------------------------------------------------------------------------- /tests/test_wigner_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_wigner_samples.py -------------------------------------------------------------------------------- /tests/test_wigner_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/astro-informatics/s2fft/HEAD/tests/test_wigner_transform.py --------------------------------------------------------------------------------