Defining Spaces
===========================

.. default-role:: code
.. automodule:: jax_md.space

Spaces
--------
.. autofunction:: free
.. autofunction:: periodic
.. autofunction:: periodic_general

Higher Order Functions
------------------------
.. autofunction:: metric
.. autofunction:: map_product
.. autofunction:: map_bond
.. autofunction:: map_neighbor
.. autofunction:: canonicalize_displacement_or_metric

Helper Functions
------------------

.. autofunction:: transform
.. autofunction:: square_distance
.. autofunction:: distance
.. autofunction:: pairwise_displacement
.. autofunction:: periodic_shift
