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: Examples

   examples
   examples_units

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

   jax_md.space
   jax_md.partition
   jax_md.energy
   jax_md.minimize
   jax_md.simulate
   jax_md.nn
   jax_md.quantity
   jax_md.elasticity
   jax_md.rigid_body
   jax_md.smap
   jax_md.io
   jax_md.interpolate
   jax_md.dataclasses
   jax_md.reaxff

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

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