Google I/O returns May 18-20! Reserve space and build your schedule Register now

tfp.substrates.jax.math.value_and_gradient

Computes f(*args) and its gradients wrt to *args.