pub fn default_device() -> Device
The default device: the GPU when Metal is available, else the CPU (as in MLX).