Summary
Add support for running the PyTorch backend of phyai on Apple Silicon GPUs through PyTorch's MPS backend (torch.device("mps")).
Motivation
PyTorch supports GPU acceleration on macOS through the Metal Performance Shaders (MPS) backend. Supporting MPS in phyai would allow users to run development, inference, and lightweight training workflows locally on Apple Silicon Macs without requiring CUDA hardware.
This should provide a native PyTorch-based Apple Silicon path, separate from MLX support.
References
Summary
Add support for running the PyTorch backend of
phyaion Apple Silicon GPUs through PyTorch's MPS backend (torch.device("mps")).Motivation
PyTorch supports GPU acceleration on macOS through the Metal Performance Shaders (MPS) backend. Supporting MPS in
phyaiwould allow users to run development, inference, and lightweight training workflows locally on Apple Silicon Macs without requiring CUDA hardware.This should provide a native PyTorch-based Apple Silicon path, separate from MLX support.
References