jax icon indicating copy to clipboard operation
jax copied to clipboard

Published 20 hours ago • verified publisher icon google

Add jet rules for `scatter_mul` and `copy`, and fix typos in `scatter*` docstrings.

Open jacobjinkelly opened this issue 1 year ago • 0 comments

Add jet rules and tests for scatter_mul and copy, and fix typos in scatter* docstrings.

The jet rule for scatter_mul takes inspiration from its jvp rule.

jacobjinkelly avatar Sep 23 '22 09:09 jacobjinkelly