Skip to content

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

Merged
qihqi merged 2 commits into
masterfrom
hanq_autocast_stopgap
Jun 17, 2025
Merged

Add autocast feature as torchax.amp.autocast.#9364
qihqi merged 2 commits into
masterfrom
hanq_autocast_stopgap

Commits

Commits on Jun 16, 2025