NO PUSH

examples/lpt_nbody_demo.py

examples/scatter.py

pyproject.toml

scripts/autotune.py

scripts/test_fft3d.py

src/jaxdecomp/__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

src/jaxdecomp/_src/jax/fft.py

src/jaxdecomp/_src/jax/fftfreq.py

src/jaxdecomp/_src/jax/halo.py

src/jaxdecomp/_src/jax/transpose.py

src/jaxdecomp/_src/pencil_utils.py

src/jaxdecomp/_src/spmd_ops.py

src/jaxdecomp/fft.py

src/jaxdecomp/transpose.py

src/jaxdecomp/typing.py

tests/conftest.py