jax
cd54dd97 - Implement the invertible decorator in terms of custom_vjp (#3957)

Commit
5 years ago
Implement the invertible decorator in terms of custom_vjp (#3957) This simplifies the implementation significantly, as we can piggyback off of all the logic for custom derivatives. For example, the previous implementation didn't support differentiating with respect to a subset of function parameters, but the new one does.
Author
Parents
Loading