xla
[torchax] Support mark_sharding_with_gradients
#9122

Merged

[torchax] Support mark_sharding_with_gradients #9122

tengyifei merged 1 commit into master from yifeit/assume-pure-mark-sharding-grad

[torchax] Support mark_shardng_with_gradients

bff3edaa

tengyifei marked this pull request as ready for review 1 year ago

tengyifei requested a review from

qihqi 1 year ago

tengyifei changed the title ~~[torchax] Support mark_shardng_with_gradients~~ [torchax] Support mark_sharding_with_gradients 1 year ago

qihqi approved these changes on 2025-05-08

tengyifei enabled auto-merge (squash) 1 year ago

tengyifei merged 40873e68 into master 1 year ago

Reviewers

qihqi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone