Jacob Kelly

icon indicating a website link jacobjinkelly.github.io icon indicating an email [email protected]

icon company @deepmind icon location London, UK icon location Research Engineer @deepmind. Prev. @VectorInstitute, @deepgenomics. Undergrad @UofT in CS, Math, and Stats.

Results 1 issues of Jacob Kelly

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

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.