:modulename:`jax_md_mod.custom_quantity.energy_wrapper`
=======================================================
.. currentmodule:: jax_md_mod.custom_quantity

.. autofunction:: energy_wrapper