jax
e7f8503c - Merge pull request #2500 from google/custom-jvp-fix

Commit

5 years ago

Merge pull request #2500 from google/custom-jvp-fix revise custom_jvp / custom_vjp rule jaxpr staging

References

Author

mattjj

Parents