flax
00a028a4 - Fix Orbax aggregate criteria and bump back to latest Orbax version

Commit

2 years ago

Fix Orbax aggregate criteria and bump back to latest Orbax version PiperOrigin-RevId: 540691732

Author

IvyZX

Committer

a-googler

Parents