jax
fa66e731 - Increase sharding to avoid timeouts

Commit

1 year ago

Increase sharding to avoid timeouts PiperOrigin-RevId: 626008096

References

#20811 - Increase sharding to avoid timeouts

Author

apaszke

apaszke

Committer

a-googler

a-googler

Parents

Loading