TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

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.