LICENSE
README.md
pyproject.toml
src/vbjax_dynamics/__init__.py
src/vbjax_dynamics/loops.py
src/vbjax_dynamics/utils.py
src/vbjax_dynamics.egg-info/PKG-INFO
src/vbjax_dynamics.egg-info/SOURCES.txt
src/vbjax_dynamics.egg-info/dependency_links.txt
src/vbjax_dynamics.egg-info/requires.txt
src/vbjax_dynamics.egg-info/top_level.txt
tests/test_accuracy.py
tests/test_jit.py
tests/test_vmap_parallel.py