Gradient Utilities
==================

.. automodule:: molpipx.utils_gradients
   :members:
   :exclude-members: jit, vmap, jacrev, lax, value_and_grad
   :undoc-members:
   :show-inheritance:

   .. rubric:: Functions

   .. autosummary::
      :nosignatures:

      get_energy_and_forces
      get_forces
      get_forces_gp
      get_pip_grad