jax
43b6a8d3 - Reorder GSPMDSharding constructor methods to ensure fast path is taken.

Commit
209 days ago
Reorder GSPMDSharding constructor methods to ensure fast path is taken. PiperOrigin-RevId: 779196847
Author
Parents
Loading