jax
82a79c59 - Fix handling of SymbolicZero output when batching custom_jvp.

Commit

1 year ago

Fix handling of SymbolicZero output when batching custom_jvp. Co-authored-by: Matthew Johnson <mattjj@google.com>

References

Author

dfm

Committer

dfm

Parents