flax
21231af9 - Replace `XLACompatibleSharding` with `jax.sharding.Sharding` since the former is deprecated and will be removed in the future.
Commit
View On GitHub
Commit
1 year ago
Replace `XLACompatibleSharding` with `jax.sharding.Sharding` since the former is deprecated and will be removed in the future. PiperOrigin-RevId: 640572753
Author
yashk2810 yashk2810
Committer
a-googler a-googler
Parents
617a6cd7
FAQ Terms Privacy Refunds Impressum
Loading