jax
cdcfe9d8 - create SingleDeviceSharding when possible

Commit
33 days ago
create SingleDeviceSharding when possible Reverts changelist 901012175 PiperOrigin-RevId: 902827544
Parents
Loading