Module: tfp.experimental.distributions.marginal_fns.tfp_custom_gradient

TF and JAX compatible custom gradients.

Functions

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

is_valid_gradient(...)

prevent_gradient(...)

JAX_MODE False