openxla/xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
解決する課題
XLAは、多様なハードウェア上で機械学習モデルを効率的に実行するという課題に対処します。モデルごとにハードウェア固有のコードを記述する必要性を排除し、GPU、CPU、および専用のMLアクセラレータ上での高性能な実行を可能にします。
仕組み
XLAはコンパイラとして機能し、PyTorch、TensorFlow、JAXなどの主要なフレームワークからのモデルを受け取り、ターゲットとなるハードウェアプラットフォームに合わせて最適化することで、高性能な実行を保証します。
対象者
このプロジェクトは、主にMLフレームワークの開発者、コンパイラを開発したいXLAコントリビューター、および新しいMLフロントエンドやハードウェアバックエンドのデバッグやサポート追加が必要なインテグレーターを対象としています。
ハイライト
- PyTorch、TensorFlow、JAXを含む主要なフレームワークをサポート
- CPU、GPU、およびMLアクセラレータ向けのクロスプラットフォーム最適化
- オープンソースの機械学習コンパイラ
関連
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト
- Dispatch