Guard autocast on unsupported devices (#14)

* Guard autocast on unsupported devices

Disable autocast on MPS (and other unsupported backends) to avoid runtime
errors while keeping existing behavior on CPU/CUDA.

* Avoid hardcoding device types in disabling autocast

---------

Co-authored-by: Jan Schlüter <github@jan-schlueter.de>