Georg Ostrovski
Georg Ostrovski
This currently results in ```TypeError: No canonicalize_dtype handler for type: ``` as opposed to regular `device_put` which doesn't do anything when in jitted code but doesn't fail either. The message...