.gitignore
CLAUDE.md
LICENSE
README.md
pyproject.toml
.claude/commands/release.md
.github/workflows/lint.yml
.github/workflows/release.yml
.github/workflows/tests.yml
jax_pme.egg-info/PKG-INFO
jax_pme.egg-info/SOURCES.txt
jax_pme.egg-info/dependency_links.txt
jax_pme.egg-info/requires.txt
jax_pme.egg-info/scm_file_list.json
jax_pme.egg-info/scm_version.json
jax_pme.egg-info/top_level.txt
jaxpme/__init__.py
jaxpme/_version.py
jaxpme/calculators.py
jaxpme/kspace.py
jaxpme/mesh.py
jaxpme/potentials.py
jaxpme/prefactors.py
jaxpme/solvers.py
jaxpme/utils.py
jaxpme/batched_flat/README.md
jaxpme/batched_flat/__init__.py
jaxpme/batched_flat/batching.py
jaxpme/batched_flat/calculators.py
jaxpme/batched_flat/potentials.py
jaxpme/batched_flat/preparation.py
jaxpme/batched_flat/solvers.py
jaxpme/batched_mixed/README.md
jaxpme/batched_mixed/__init__.py
jaxpme/batched_mixed/batching.py
jaxpme/batched_mixed/calculators.py
jaxpme/batched_mixed/kspace.py
jaxpme/batched_tiled/README.md
jaxpme/batched_tiled/__init__.py
jaxpme/batched_tiled/batching.py
jaxpme/batched_tiled/calculators.py
jaxpme/batched_tiled/kernel.py
jaxpme/batched_tiled/kspace.py
tests/conftest.py
tests/test_batched_flat.py
tests/test_batched_mixed_batching.py
tests/test_batched_mixed_calculator.py
tests/test_batched_tiled_batching.py
tests/test_batched_tiled_calculator.py
tests/test_batched_tiled_kernel_memory.py
tests/test_batched_tiled_mixed_pbc.py
tests/test_ewald.py
tests/test_inverse_power_law.py
tests/test_kspace.py
tests/test_slab_correction.py
tests/reference_structures/coulomb_test_frames.xyz
tests/reference_structures/mad15_2d_subset.xyz