xla
Add autocast feature as `torchax.amp.autocast`.
#9364

Merged

Add autocast feature as `torchax.amp.autocast`. #9364

qihqi merged 2 commits into master from hanq_autocast_stopgap

Add autocast feature as `torchax.amp.autocast`.

fdf0387a

qihqi requested a review from

tengyifei 139 days ago

yapf

321fc82f

qihqi requested a review from

lsy323 139 days ago

qihqi enabled auto-merge (squash) 139 days ago

bhavya01 approved these changes on 2025-06-17

qihqi merged 89cc066a into master 138 days ago

Reviewers

bhavya01

tengyifei

lsy323

Assignees

No one assigned

Labels

None yet

Milestone

No milestone