概要
MLXはNumpyのような使い心地で設計されており、トランスフォーマー言語モデルのトレーニング、大規模なテキスト生成、テキストのファインチューニング、画像生成、音声認識などをAppleシリコン上で実行することができます。
MLXを使用することでApple siliconを最大限に活かすことができるとされている。
動作環境
MLXを使用するにはApple siliconを搭載したMacが必要になります。
MacOSは13.3以上 いわゆるVenturaです。Sonomaにしてあれば間違い無いです。公式文書ではSonomaでの使用を強く勧めています。
Pythonは3.8以上で動作するようです。
インストール
MLXのインストールにはターミナルで以下のコマンドを実行します。(Python3の場合)
Python
pip3 install mlx
活用方法
こちらのGitHubに機械学習のモデルがいくつかあります。
https://github.com/ml-explore/mlx-examples/tree/main