Apple silicon向け機械学習用配列フレームワークMLXのSwift API。Swift開発者が機械学習の研究や実験を実行できます。
- スター
- 28,887
- フォーク
- 2,366
- 未解決の Issue
- 137
- 最終コミット
- 9 時間前
これは何?
MLXは、Appleの機械学習研究チームが公開した、Apple silicon向けの機械学習用配列フレームワークです。Python APIはNumPyに近い設計で、mlx.nnやmlx.optimizersなどの高水準パッケージは、モデル構築に使うPyTorch風のAPIを採用しています。C++、C、SwiftのAPIも利用できます。
MLXは遅延計算を採用しているため、配列は必要になったときにのみ実体化されます。また、計算グラフを動的に構築するので、引数の形状が変わっても時間のかかる再コンパイルは発生しません。自動微分、ベクトル化、グラフ最適化のための、組み合わせ可能な関数変換もサポートしています。
設計上の重要な特徴はユニファイドメモリです。配列は共有メモリ上に置かれ、デバイス間でデータをコピーせずにCPUまたはGPUで処理を実行できます。また、LLaMAによるテキスト生成、LoRAファインチューニング、Stable Diffusion、Whisperによる音声認識のサンプルも用意されています。
主な機能
- 01NumPyに近いPython APIと、PyTorchをモデルにしたnnおよびoptimizersモジュール
- 02Pythonに加えてC++、C、SwiftのAPIも利用可能
- 03遅延計算と動的に構築される計算グラフ
- 04自動微分とベクトル化のための組み合わせ可能な変換
- 05データ転送なしでCPUまたはGPU上の処理を可能にするユニファイドメモリモデル
ユースケース
- Apple silicon搭載Macで新しいモデルを研究・プロトタイピングする
- Apple製ハードウェアでLLMのテキスト生成やLoRAファインチューニングを実行する
- macOS上でStable DiffusionやWhisperによる音声認識をローカル実行する
- NumPy風のインターフェースを使ってC++またはSwiftで数値計算・機械学習コードを構築する
クイックスタート
- 1
macOSにpipでMLXをインストールする
pip install mlx - 2
ドキュメントのクイックスタートガイドを読む
使う前に知っておきたいこと
- Apple siliconを対象としています。READMEに記載されている主なインストール方法はmacOS向けです。
- Linux向けのCUDA版およびCPU専用パッケージにも言及されているため、プラットフォームのサポート状況はバックエンドによって異なります。
収録されている特集
トピック
よくある質問
- mlx とは?
- NumPy風のPython APIとC++、C、Swiftバインディングを備えた、Apple silicon向けの機械学習用配列フレームワーク。 MLXは、Appleの機械学習研究チームが公開した、Apple silicon向けの機械学習用配列フレームワークです。Python APIはNumPyに近い設計で、mlx.nnやmlx.optimizersなどの高水準パッケージは、モデル構築に使うPyTorch風のAPIを採用しています。C++、C、SwiftのAPIも利用できます。
- mlx は無料で使えますか?ライセンスは?
- mlx は MIT ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
- mlx は今もメンテナンスされていますか?
- 活発にメンテナンス中。mlx の最終コミットは 2026年10月22日 です。
- mlx は何に使えますか?
- Apple silicon搭載Macで新しいモデルを研究・プロトタイピングする Apple製ハードウェアでLLMのテキスト生成やLoRAファインチューニングを実行する macOS上でStable DiffusionやWhisperによる音声認識をローカル実行する NumPy風のインターフェースを使ってC++またはSwiftで数値計算・機械学習コードを構築する
- mlx のインストール方法は?
- macOSにpipでMLXをインストールする: pip install mlx ドキュメントのクイックスタートガイドを読む
- mlx はどこで入手できますか?
- ソースコードとインストール手順は GitHub にあります:https://github.com/ml-explore/mlx
概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。
関連プロジェクト
autogradを基盤とする動的ニューラルネットワークと、GPUアクセラレーション対応のテンソル計算を提供するPython深層学習フレームワーク。
ivy.transpileを使って、PyTorch、TensorFlow、JAX、NumPyなどのフレームワーク間で機械学習コードを変換するPythonライブラリ。
Pythonでコンパイラパイプラインをカスタマイズし、多様なハードウェアにモデルをデプロイできるオープンな機械学習コンパイルフレームワーク。
作りながら学ぶML講座。NumPyで一から作る深層学習フレームワークと、Pyodideで採点する350問以上のブラウザ演習。
PyTorch風のテンソルAPI、autograd、コンパイラを備え、多数のアクセラレータバックエンドに対応する最小構成のディープラーニングフレームワーク。