LICENSE
NOTICE
README.md
pyproject.toml
python/lammps_jax/__init__.py
python/lammps_jax/comm.py
python/lammps_jax/eam.py
python/lammps_jax/eam_pallas.py
python/lammps_jax/export.py
python/lammps_jax/mace.py
python/lammps_jax.egg-info/PKG-INFO
python/lammps_jax.egg-info/SOURCES.txt
python/lammps_jax.egg-info/dependency_links.txt
python/lammps_jax.egg-info/requires.txt
python/lammps_jax.egg-info/top_level.txt
python/lammps_jax/dist/__init__.py
python/lammps_jax/dist/parallel/__init__.py
python/lammps_jax/dist/parallel/feature_exchange.py
python/lammps_jax/dist/parallel/force_decomposition.py
python/lammps_jax/dist/parallel/ghost_exchange.py
python/lammps_jax/dist/parallel/local_neighbor_list.py
python/lammps_jax/dist/parallel/replicate_data.py
python/lammps_jax/dist/scripts/__init__.py
python/lammps_jax/dist/scripts/lj_bench.py
python/lammps_jax/dist/scripts/nequix_bench.py
tests/test_dist_parallel.py
tests/test_distributed.py
tests/test_eam.py
tests/test_eam_pallas.py
tests/test_export.py
tests/test_lammps.py
tests/test_mace.py