Spatial Partitioning
===========================

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

Neighbor Lists
---------------
.. autofunction:: neighbor_list
.. autofunction:: neighbor_list_mask
.. autofunction:: to_jraph
.. autofunction:: to_dense

.. autoclass:: NeighborList
.. autoclass:: NeighborListFormat
.. autoclass:: NeighborListFns

Cell Lists
-----------
.. autofunction:: cell_list
.. autoclass:: CellList
