jax
572928df - fix custom_jvp_call_jaxpr transpose function (#3231)

Commit

5 years ago

fix custom_jvp_call_jaxpr transpose function (#3231) * make custom_jvp_call_jaxpr handle multilinear funs see #3226 * remove old comment

References

Author

mattjj

Parents