PyTorch 2.15 officially supports MPS: FP8, MoE acceleration, and a comprehensive upgrade to the Metal kernel.
PyTorch 2.15 brings a major MPS backend upgrade to Apple Silicon Macs, natively supporting FP8 tensor operations on macOS for the first time, boosting MoE model decoding speed by 5–6.5 times, reducing peak video VAE memory usage from 270 GiB to 8 GiB, and migrating numerous operators from MPSGraph to dedicated Metal kernels.