Sole usage of Python 3.10's match-case in the project hinders quick-start for beginners; consider replacing with if-else for improved accessibility.
return torch.float64 if device is not mps or xpu, else return torch.float32