Control Variates
================

.. currentmodule:: optax.monte_carlo

.. autosummary::
    control_delta_method
    control_variates_jacobians
    moving_avg_baseline

Control delta method
~~~~~~~~~~~~~~~~~~~~
.. autofunction:: control_delta_method

Control variates Jacobians
~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autofunction:: control_variates_jacobians

Moving average baseline
~~~~~~~~~~~~~~~~~~~~~~~
.. autofunction:: moving_avg_baseline