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 1 year ago

Implement autocast for torchax

0342e14d

qihqi force pushed from c57e2111 to 0342e14d 1 year ago

61683f1d

Cpp reg

1c1bc192

qihqi changed the title ~~Add autocast~~ [WIP] Add autocast 1 year ago

qihqi marked this pull request as draft 1 year ago

qihqi changed the title ~~[WIP] Add autocast~~ [WIP] Add autocast in torchax 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone