Skip to content

Add support for device kwarg in astype, and add matching utility func#21086

Open
Micky774 wants to merge 1 commit intojax-ml:mainfrom Micky774:astype-device