Module: tfp.experimental.substrates.jax.math.gradient

View source on GitHub

Functions for computing gradients.

Functions

value_and_gradient(...): Computes f(*xs) and its gradients wrt to *xs.