PyTorch 2.15、Mac上のAI処理を高速化
出典:Hugging Face
詳細を読む
Hugging FaceのPyTorchチームは2026年10月20日、Apple製GPU向けMPSを大幅に強化したPyTorch 2.15の変更点を公開しました。MPSGraph処理を専用Metalカーネルへ移し、Mac上での大規模言語モデル生成、動画処理、線形代数を高速化・省メモリー化しています。
M3 Ultra上でtorch.compileを使ったMoEモデルの生成速度は、PyTorch 2.14比で5〜6.5倍になりました。Qwen3.6-35B-A3Bは毎秒9.4トークンから61.3トークンへ伸び、語彙全体へのsoftmaxも約8倍高速になっています。
MPSは1値1バイトのFP8形式に対応し、FP16やBF16の半分の容量で重みを保持できます。Wan2.2のタイル分割VAEでは、パディングをMetalへ移したことでドライバーメモリーのピークが約270GiBから約8GiBへ減り、Qwen-Image-2.1の画像破損も同じ変更で解消しました。
Pythonから処理先を選べるMPSストリームも公開し、複素数のLU分解や三角行列求解など対応演算を増やしました。利用にはAppleシリコン、macOS 15以降、Python 3.11以降が必要で、次期2.16ではAttentionの全面Metal化も計画しています。