xla
Use .backward() with in-place grad mutations for the GA API
#8768
Merged

Use .backward() with in-place grad mutations for the GA API #8768

rpsilva-aws
rpsilva-aws rpsilva-aws force pushed from c49a1fd6 to 5922c4d4 250 days ago
rpsilva-aws
rpsilva-aws Use .backward() with in-place grad mutations for the GA API
ecc92bd9
rpsilva-aws rpsilva-aws force pushed from 5922c4d4 to ecc92bd9 250 days ago
rpsilva-aws rpsilva-aws marked this pull request as ready for review 250 days ago
rpsilva-aws rpsilva-aws requested a review from tengyifei tengyifei 250 days ago
bhavya01
bhavya01 commented on 2025-02-28
rpsilva-aws rpsilva-aws requested a review from bhavya01 bhavya01 247 days ago
rpsilva-aws
rpsilva-aws
bhavya01
bhavya01 approved these changes on 2025-03-05
bhavya01 bhavya01 merged 4540d817 into master 244 days ago
rpsilva-aws rpsilva-aws deleted the rpsilva_grad_acc_chkpt branch 244 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone