Arm-Examples/ML-examples

Arm Machine Learning tutorials and examples

What it solves

このリポジトリは、開発者が Arm ベースのハードウェア、特にニューラル・プロセッシング・ユニット (NPU) と Cortex ベースのプラットフォーム上で AI モデルを最適化およびデプロイするための、実践的な機械学習の例とチュートリアルを集めたものです。

How it works

このプロジェクトは、Arm ハードウェア上での ML パイプラインのさまざまな側面を示す、一連の専門的な例を提供します：

  • Hardware Acceleration: Corstone-300 および Corstone-1000 プラットフォーム上の Ethos-U NPU 用の例。
  • Library Integration: KleidiAI ライブラリの使用方法のデモンストレーション。
  • Model Optimization: Ethos NPU 用にモデルの重みのサイズを削減するための、プルーニング (pruning) とクラスタリング (clustering) に関するチュートリアル。
  • Framework Integration: PyTorch/ExecuTorch を使用して音声認識用の Conformer ネットワークを訓練し、Ethos-U へエクスポートするガイド。
  • TensorFlow Lite Deployment: .tflite 格式のエクスポート用に RNN unrolling を示す Jupyter notebook。

Who it’s for

Arm Cortex および Ethos NPU ハードウェアをエッジ AI デプロイメントのためにターゲットとする、組み込みシステム開発者および ML エンジニア。

Highlights

  • Arm Ethos-U NPU および Corstone リファレンスプラットフォームのサポート。
  • プルーニング (pruning) とクラスタリング (clustering) のようなモデル圧縮技術の例。
  • PyTorch, ExecuTorch, および TensorFlow Lite との統合。
  • Conformer ネットワークを使用した音声認識のための特定のインプリメンテーション。