jax
9630a2b0 - Use `to_cotangent_aval()` in `SymbolicZero` check in _flatten_bwd

Commit

59 days ago

Use `to_cotangent_aval()` in `SymbolicZero` check in _flatten_bwd PiperOrigin-RevId: 845603376

Author

yashk2810

Committer

Google-ML-Automation

Parents