jax
91777c33 - Fix select_and_gather_add_transpose rule.

Commit

5 years ago

Fix select_and_gather_add_transpose rule. Fixes a 'jax.ad_util.Zero' is not a valid JAX type error.

References

Author

LenaMartens

Committer

LenaMartens

Parents