apple/coreai-models

Model export recipes, Python primitives, and Swift runtime utilities for on-device AI

解決する課題

このプロジェクトは、PyTorch で AI モデルをトレーニングすることと、Apple シリコンデバイス上でそれらを効率的にデプロイすることの間のギャップを埋めるために必要なツールを提供します。Core AI フレームワークを使用して、macOS および iOS 上でモデルを変換、最適化、実行するプロセスを簡素化します。

仕組み

以下の3つの部分からなるエコシステムを提供します:

  1. Export Recipes: 一般的なオープンソースモデル(Hugging Face のものなど)を .aimodel 形式に変換するためのスクリプト。
  2. Python Primitives: 開発者が Apple のハードウェアと互換性のあるカスタムモデルを PyTorch で作成するためのビルディングブロック。
  3. Swift Runtime: 開発者がエクスポートされたモデルを Swift ベースのアプリケーションに直接統合できるようにするパッケージ。

さらに、モデルの作成、圧縮、デプロイのワークフローに関する専門的なガイダンスを提供する、コーディングエージェント(Claude Code や Codex など)向けのプラグイン「Skills」も含まれています。

対象ユーザー

PyTorch モデルを使用して、Apple プラットフォーム向けの高性能なオンデバイス AI アプリケーションを構築したい開発者や機械学習エンジニア向けです。

ハイライト

  • Model Catalog: さまざまなオープンソースモデル向けのすぐに使えるエクスポートレシピが含まれています。
  • Agentic Skills: モデル作成や圧縮タスクを自動化するのに役立つ、コーディングエージェント用のプラグインを提供します。
  • Native Integration: macOS および iOS 上でのシームレスなランタイム実行のための Swift パッケージが含まれています。
  • Optimization Tools: モデルがデバイス上で効率的に動作するように、重みの圧縮、量子化、パレット化を検討するためのユーティリティを提供します。

関連

  • プロジェクト
  • プロジェクト
  • Dispatch
  • プロジェクト
  • プロジェクト