blackjax
blackjax copied to clipboard
Add "How to use custom gradients?" documentation
It's very simple: https://jax.readthedocs.io/en/latest/notebooks/Custom_derivative_rules_for_Python_code.html#python-debugging