ml-explore/mlx-swift

Swift API for MLX

何を解決するか

MLX Swiftは、MLX配列フレームワーク用のネイティブなSwift APIを提供し、Apple silicon上でSwift言語内で機械学習の研究や実験を直接行えるようにします。

動作方法

MLXフレームワークのSwift拡張として機能し、ユーザーがSwiftを使ってモデルの構築と学習が可能になります。Xcode、SwiftPM、CMakeを含む複数のビルドシステムをサポートしており、macOS/iOSではMetal、LinuxではCUDAのバックエンドを提供します。

対象ユーザー

macOS、iOS、Linux上でApple siliconの性能を活用して機械学習タスクを実行したいSwift開発者および機械学習研究者。

特徴

  • MLX配列フレームワーク用のネイティブなSwift API。
  • 機能拡張パッケージ mlx-swift-lm を通じてLLMおよびVLMの実装をサポート。
  • macOS、iOS、Linuxのクロスプラットフォーム対応。
  • MNISTにLeNetを訓練する例、LLM/VLMを使ったチャットアプリの実行、Stable Diffusionによる画像生成の例を含む。

関連

  • プロジェクト
    ml-explore/mlx-swift-lm

    MLX Swiftを使用して、大規模言語モデル（LLM）および視覚言語モデル（VLM）を用いたアプリケーションを構築するためのSwiftパッケージ。モデルのロード、ファインチューニング、およびガイド付き生成機能を備えています。

  • プロジェクト
    ml-explore/mlx-swift-examples

    MLX Swift 用の例プログラムとライブラリのコレクション。iOS および macOS 上で LLM、VLM、画像生成アプリを構築する方法を示しています。

  • プロジェクト
    ml-explore/mlx

    Apple機械学習研究チームが開発した配列フレームワークで、Apple silicon上で機械学習に最適化されており、統合メモリモデルと、NumPy/PyTorchに似た使い慣れたAPIを備えています。

  • プロジェクト
    SharpAI/SwiftLM

    SSDエキスパートストリーミングとKVキャッシュ圧縮機能を備え、OpenAI互換APIを介してMLXモデルを提供する、Apple Silicon向けのネイティブSwift推論サーバー。

  • プロジェクト
    Blaizzy/mlx-audio-swift

    Apple Silicon上でのオーディオ処理のためのモジュール式Swift SDK。MLXフレームワークを使用して、デバイス上でのテキスト読み上げ、音声文字起こし、話者分離を実現します。