MPS in PyTorch 2.15

TL;DR What this means for you Contents Getting started FP8 comes to MPS Faster language-model decoding Linear layers Mixture-of-experts layers Linear algebra gets a boost Complex-valued computations Small matrices and larger solves Convolutions, padding, and training Softmax and reductions A much smaller memory...