flax
594b4529 - Remove process_index() check in flax/examples, to match Orbax API.

Commit

3 years ago

Remove process_index() check in flax/examples, to match Orbax API. PiperOrigin-RevId: 516880068

Author

IvyZX

Committer

a-googler

Parents