MLXはApple Siliconのユニファイドメモリを活用し、CPUとGPU間のコピーを伴わない設計が特徴です。2025年後半には実験的なCUDA backendも追加され、Apple SiliconとNVIDIA GPUの両方で同一コードベースを動かせるようになりました。
機械学習フレームワークを比較したい相談
ユニファイドメモリを前提としたゼロコピー設計