flax
3747c877 - Remove pointless comprehension in jax_utils pmap helper.

Commit

3 years ago

Remove pointless comprehension in jax_utils pmap helper. PiperOrigin-RevId: 483564165

Author

levskaya

Committer

a-googler

Parents