Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge

tfp.substrates.jax.math.value_and_gradient

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