Save the date! Google I/O returns May 18-20 Register now

tfp.substrates.jax.math.value_and_gradient

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