jax

46f796b3 - Dedupe shardings before passing them to _get_and_check_device_assignment

Dedupe shardings before passing them to _get_and_check_device_assignment In practice, the number of different shardings is usually much smaller then the number of inputs/output. PiperOrigin-RevId: 600558309