:modulename:`jax_md_mod.custom_energy.dihedral_energy`
======================================================
.. currentmodule:: jax_md_mod.custom_energy

.. autofunction:: dihedral_energy