jax
6193e5e4 - revamp custom_jvp/vjp implementation to fix bugs

Commit

5 years ago

revamp custom_jvp/vjp implementation to fix bugs Co-authored-by: Dougal Maclaurin <dougalm@google.com>

References

Author

mattjj

Committer

mattjj

Parents