jax
474cc95f - Allow `with_refs` to take Accums too.

Commit

204 days ago

Allow `with_refs` to take Accums too. PiperOrigin-RevId: 885825036

References

#36015 - Allow `with_refs` to take Accums too.

Author

yashk2810

yashk2810

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading