JAX MD Reference
===========================================

JAX MD is a collection of tools for working with collections of particles in
JAX. Different parts of the JAX MD tooling are designed to work independently
and so you should feel free to pick and choose which ones you use for your
particular problem.

.. toctree::
   :maxdepth: 2
   :caption: Reference

   jax_md.space
   jax_md.partition
   jax_md.energy
   jax_md.minimize
   jax_md.simulate
   jax_md.quantity
   jax_md.smap

Indices and tables
==================

* :ref:`genindex`
* :ref:`modindex`
* :ref:`search`
