Attend the Women in ML Symposium on December 7 Register now

Module: tfp.experimental.distributions.marginal_fns.tfp_custom_gradient

Stay organized with collections Save and categorize content based on your preferences.

TF and JAX compatible custom gradients.

Functions

custom_gradient(...): Decorates a function and adds custom derivatives.

is_valid_gradient(...)

prevent_gradient(...)

JAX_MODE False