Merge pull request #31 from DifferentiableUniverseInitiative/docstrings
Docstrings
Showing
- .pre-commit-config.yaml 4 additions, 0 deletions.pre-commit-config.yaml
- examples/lpt_nbody_demo.py 1 addition, 1 deletionexamples/lpt_nbody_demo.py
- jaxdecomp/__init__.py 1 addition, 4 deletionsjaxdecomp/__init__.py
- jaxdecomp/_src/__init__.py 0 additions, 4 deletionsjaxdecomp/_src/__init__.py
- jaxdecomp/_src/cudecomp/fft.py 93 additions, 99 deletionsjaxdecomp/_src/cudecomp/fft.py
- jaxdecomp/_src/cudecomp/halo.py 185 additions, 190 deletionsjaxdecomp/_src/cudecomp/halo.py
- jaxdecomp/_src/cudecomp/transpose.py 154 additions, 109 deletionsjaxdecomp/_src/cudecomp/transpose.py
- jaxdecomp/_src/fft_utils.py 76 additions, 6 deletionsjaxdecomp/_src/fft_utils.py
- jaxdecomp/_src/jax/fft.py 290 additions, 114 deletionsjaxdecomp/_src/jax/fft.py
- jaxdecomp/_src/jax/fftfreq.py 41 additions, 15 deletionsjaxdecomp/_src/jax/fftfreq.py
- jaxdecomp/_src/jax/halo.py 193 additions, 117 deletionsjaxdecomp/_src/jax/halo.py
- jaxdecomp/_src/jax/transpose.py 20 additions, 20 deletionsjaxdecomp/_src/jax/transpose.py
- jaxdecomp/_src/padding.py 0 additions, 427 deletionsjaxdecomp/_src/padding.py
- jaxdecomp/_src/pencil_utils.py 99 additions, 30 deletionsjaxdecomp/_src/pencil_utils.py
- jaxdecomp/_src/spmd_ops.py 88 additions, 52 deletionsjaxdecomp/_src/spmd_ops.py
- jaxdecomp/fft.py 201 additions, 105 deletionsjaxdecomp/fft.py
- jaxdecomp/halo.py 61 additions, 2 deletionsjaxdecomp/halo.py
- jaxdecomp/transpose.py 188 additions, 6 deletionsjaxdecomp/transpose.py
- jaxdecomp/typing.py 12 additions, 0 deletionsjaxdecomp/typing.py
- tests/test_halo.py 0 additions, 2 deletionstests/test_halo.py
Loading
Please register or sign in to comment