Stochastic Gradient Estimators and Control Variates# Warning This module has been deprecated and will be removed in optax 0.2.7 Control delta method# Control variates Jacobians# Moving average baseline# Measure valued Jacobians# Pathwise Jacobians# Score function Jacobians#
