API Reference
Returns the gradient of the model log-probability w.r.t. the node value.
This method is unlikely to be efficient. Consider using JAX explicitly to JIT-compile more complex functions. Alternatively, you can use Goose.
Any