toto

.clangd

.github/workflows/github-deploy.yml

.github/workflows/tests.yml

.pre-commit-config.yaml

CMakeLists.txt

README.md

examples/lpt_nbody_demo.py

examples/scatter.py

examples/visualizer.ipynb

joss-paper/assets/fft.svg

joss-paper/assets/halo-exchange.svg

pyproject.toml

scripts/autotune.py

scripts/test_fft3d.py

src/jaxdecomp/__init__.py

src/jaxdecomp/_src/__init__.py

src/jaxdecomp/_src/cudecomp/fft.py

src/jaxdecomp/_src/cudecomp/halo.py

src/jaxdecomp/_src/cudecomp/transpose.py

src/jaxdecomp/_src/fft_utils.py