Energy Minimization Routines
==============================

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

Minimization Routines
----------------------

.. autofunction:: gradient_descent
.. autofunction:: fire_descent

Data Types
-----------

.. autoclass:: FireDescentState
