jax
8ccb907d - in custom_jvp/vjp stop_gradient on nondiff_argnums (#2804)

Commit

5 years ago

in custom_jvp/vjp stop_gradient on nondiff_argnums (#2804) fixes #2784

References

#2804 - for custom_jvp/vjp nondiff_argnums, call stop_gradient

Author

mattjj

mattjj

Parents

Loading