measure_valued_jacobians ======================== Auto-generated from jax_code_defs .. raw:: html

JAX (Core)

API: optax.monte_carlo.stochastic_gradient_estimators.measure_valued_jacobians
Strategy: Direct Mapping

Flax NNX

API: optax.monte_carlo.stochastic_gradient_estimators.measure_valued_jacobians
Strategy: Direct Mapping