PR #9361 [WIP] Add autocast in torchax

[WIP] Add autocast in torchax #9361

qihqi wants to merge 3 commits into master from hanq_autocast

qihqi force pushed from b99bf643 to c57e2111 186 days ago

Implement autocast for torchax

0342e14d

qihqi force pushed from c57e2111 to 0342e14d 186 days ago

61683f1d

Cpp reg

1c1bc192

qihqi changed the title ~~Add autocast~~ [WIP] Add autocast 186 days ago

qihqi marked this pull request as draft 186 days ago

qihqi changed the title ~~[WIP] Add autocast~~ [WIP] Add autocast in torchax 186 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone