├── .clang-format ├── .devcontainer ├── Dockerfile └── devcontainer.json ├── .github ├── dependabot.yml └── workflows │ ├── gpu-build.yml │ ├── tests.yml │ └── wheels.yml ├── .gitignore ├── .gitmodules ├── .pre-commit-config.yaml ├── CMakeLists.txt ├── LICENSE ├── README.md ├── ci ├── Dockerfile └── Jenkinsfile ├── lib ├── cufinufft_wrapper.cc ├── cufinufft_wrapper.h ├── jax_finufft_cpu.cc ├── jax_finufft_cpu.h ├── jax_finufft_gpu.cc ├── kernel_helpers.h ├── kernels.cc.cu ├── kernels.h └── nanobind_kernel_helpers.h ├── pyproject.toml ├── src └── jax_finufft │ ├── __init__.py │ ├── lowering.py │ ├── ops.py │ ├── options.py │ └── shapes.py └── tests ├── freethreaded_test.py ├── ops_test.py ├── options_test.py └── shapes_test.py /.clang-format: -------------------------------------------------------------------------------- 1 | BasedOnStyle: Google 2 | ColumnLimit: 99 3 | -------------------------------------------------------------------------------- /.devcontainer/Dockerfile: -------------------------------------------------------------------------------- 1 | ../ci/Dockerfile -------------------------------------------------------------------------------- /.devcontainer/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/.devcontainer/devcontainer.json -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/gpu-build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/.github/workflows/gpu-build.yml -------------------------------------------------------------------------------- /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.github/workflows/wheels.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/.github/workflows/wheels.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/.gitmodules -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/README.md -------------------------------------------------------------------------------- /ci/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/ci/Dockerfile -------------------------------------------------------------------------------- /ci/Jenkinsfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/ci/Jenkinsfile -------------------------------------------------------------------------------- /lib/cufinufft_wrapper.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/lib/cufinufft_wrapper.cc -------------------------------------------------------------------------------- /lib/cufinufft_wrapper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/lib/cufinufft_wrapper.h -------------------------------------------------------------------------------- /lib/jax_finufft_cpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/lib/jax_finufft_cpu.cc -------------------------------------------------------------------------------- /lib/jax_finufft_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/lib/jax_finufft_cpu.h -------------------------------------------------------------------------------- /lib/jax_finufft_gpu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/lib/jax_finufft_gpu.cc -------------------------------------------------------------------------------- /lib/kernel_helpers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/lib/kernel_helpers.h -------------------------------------------------------------------------------- /lib/kernels.cc.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/lib/kernels.cc.cu -------------------------------------------------------------------------------- /lib/kernels.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/lib/kernels.h -------------------------------------------------------------------------------- /lib/nanobind_kernel_helpers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/lib/nanobind_kernel_helpers.h -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/jax_finufft/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/src/jax_finufft/__init__.py -------------------------------------------------------------------------------- /src/jax_finufft/lowering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/src/jax_finufft/lowering.py -------------------------------------------------------------------------------- /src/jax_finufft/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/src/jax_finufft/ops.py -------------------------------------------------------------------------------- /src/jax_finufft/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/src/jax_finufft/options.py -------------------------------------------------------------------------------- /src/jax_finufft/shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/src/jax_finufft/shapes.py -------------------------------------------------------------------------------- /tests/freethreaded_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/tests/freethreaded_test.py -------------------------------------------------------------------------------- /tests/ops_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/tests/ops_test.py -------------------------------------------------------------------------------- /tests/options_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/tests/options_test.py -------------------------------------------------------------------------------- /tests/shapes_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/flatironinstitute/jax-finufft/HEAD/tests/shapes_test.py --------------------------------------------------------------------------------