XLA (コンパイラ)
表示
開発元 | OpenXLA |
---|---|
リポジトリ | xla - GitHub |
プログラミング 言語 | C++ |
対応OS | Linux, macOS, Windows |
種別 | コンパイラ |
ライセンス | Apache License 2.0 |
公式サイト |
openxla |
XLA(Accelerated Linear Algebra)は、OpenXLAによって開発された機械学習向けのオープンソースコンパイラである[1]。XLAは、より低レベルでの計算グラフの最適化により機械学習モデルの性能を向上させるように設計されており、大規模な計算や高性能な機械学習モデルに特に有用である。
XLAの主要な特徴は以下の通りである:[2]
- 計算グラフのコンパイル:計算グラフを効率的な機械コードにコンパイルする。
- 最適化技術:演算の融合、メモリ最適化、その他の技術を適用する。
- ハードウェアサポート:CPU、GPU、AIアクセラレータなど、様々なハードウェアに対してモデルを最適化する。
- モデル実行時間の改善:トレーニングと推論の両方において、機械学習モデルの実行時間の短縮を目指す。
- シームレスな統合:既存の機械学習コードに最小限の変更で使用可能である。
XLAは機械学習モデルの最適化における重要な進歩を表しており、開発者に計算効率と性能を向上させるためのツールを提供している。[3][4]
XLAはTensorFlow、PyTorch、JAXなどから利用可能である[5]。
XLAは、元々は、Googleが開発していたもので、2017年3月6日にTensorFlowの一部として発表された[6]。2022年よりOpenXLAが開発することとなった[7]。OpenXLAにはAlibaba, Amazon Web Services, AMD, Anyscale, Apple, Arm, Cerebras, Google, Graphcore, Hugging Face, Intel, Meta, NVIDIAが参加している[1]。
参照
[編集]- ^ a b “OpenXLA Project”. December 21, 2024閲覧。
- ^ Woodie, Alex (2023年3月9日). “OpenXLA Delivers Flexibility for ML Apps”. Datanami. 2023年12月10日閲覧。
- ^ “TensorFlow XLA: Accelerated Linear Algebra”. TensorFlow Official Documentation. 2023年12月10日閲覧。
- ^ Smith, John (2022-07-15). “Optimizing TensorFlow Models with XLA”. Journal of Machine Learning Research 23: 45–60.
- ^ “xla/README.md at main · openxla/xla”. December 22, 2024閲覧。
- ^ “XLA - TensorFlow, compiled - Google Developers Blog”. December 23, 2024閲覧。
- ^ “Google's open-source momentum: OpenXLA, new partnerships”. December 23, 2024閲覧。