pub fn best_device(force_cpu: bool) -> Result<Device>
Gets the best device, cpu, cuda if compiled with CUDA, or Metal