:modulename:`learn.difftre`
=====================================

.. currentmodule:: chemtrain.learn.difftre

.. automodule:: chemtrain.learn.difftre


Loss Functions
---------------

.. autofunction:: init_default_loss_fn

.. autofunction:: init_rel_entropy_loss_fn

Gradient Computation Routines
------------------------------

.. autofunction:: init_difftre_gradient_and_propagation

.. autofunction:: init_rel_entropy_gradient_and_propagation

ESS Line-Search
----------------

.. autofunction:: init_step_size_adaption