jax
12462e05 - Fix ValAccum to initialize Zero using .to_cotangent_aval()

Commit

18 days ago

Fix ValAccum to initialize Zero using .to_cotangent_aval() Co-authored-by: Matthew Johnson <mattjj@google.com> PiperOrigin-RevId: 863476521

References

Author

yashk2810

Committer

Google-ML-Automation

Parents