Added Apple Silicon GPU device

[missflash]

Added Apple Silicon GPU device

if torch.backends.mps.is_available():
    device = torch.device("mps")   # Apple Silicon GPU (Metal)
elif torch.cuda.is_available():
    device = torch.device("cuda")  # NVIDIA GPU
else:
    device = torch.device("cpu")   # CPU fallback

print(f"Using device: {device}")

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[rasbt]

Hi there, thanks for the PR. I agree that mps can be added here as it's not one of the main chapters (for the main chapters I don't recommend mps yet because of the current instability and numerical differences).

I tried to make a few updates to leave "cuda" as the default but when I pushed it, it created a new branch over at #820. I think that's because you did the PR from a main branch instead of a feature branch in your fork. Long story short, I am closing this in favor of #820 but don't worry, your contributions are counted in #820.