flax
d237a45f - Add support for `jax.sharding.PartitionSpec.UNCONSTRAINED` in logical specification
Commit
View On GitHub
Commit
1 year ago
Add support for `jax.sharding.PartitionSpec.UNCONSTRAINED` in logical specification PiperOrigin-RevId: 631645837
Author
marksandler2 marksandler2
Committer
a-googler a-googler
Parents
ea025eb1
FAQ Terms Privacy Refunds Impressum
Loading