flax
Add support for `jax.sharding.PartitionSpec.UNCONSTRAINED` in logical specification
#3902
Merged

Add support for `jax.sharding.PartitionSpec.UNCONSTRAINED` in logical specification #3902

ghost merged 1 commit into main from test_630547382
copybara-service
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
marksandler2 Add support for `jax.sharding.PartitionSpec.UNCONSTRAINED` in logical…
d237a45f
copybara-service copybara-service force pushed to d237a45f 1 year ago
copybara-service copybara-service closed this 1 year ago
copybara-service copybara-service deleted the test_630547382 branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone