OpenXLA adalah ekosistem terbuka dari komponen infrastruktur machine learning (ML) yang berperforma tinggi, portabel, dan dapat diperluas yang menyederhanakan pengembangan ML dengan mendefragmentasi alat antara framework frontend dan backend hardware. Dibuat oleh para pemimpin industri dalam pemodelan, software, dan hardware AI.
Bagaimana komunitas menggunakan OpenXLA? Halaman ini menggabungkan link ke repositori dan project menggunakan OpenXLA untuk memberikan inspirasi dan pointer kode.
Memiliki project yang menggunakan OpenXLA? Kirim permintaan pull kepada kami dan tambahkan ke halaman ini.
Framework
- JAX adalah framework ML dengan
API mirip NumPy untuk menulis model ML berperforma tinggi
- PyTorch/XLA menyediakan jembatan dari PyTorch
ke OpenXLA dan StableHLO
- TensorFlow adalah framework ML
lama dengan ekosistem yang besar
Plugin PJRT
- libTPU memungkinkan model dijalankan di Cloud TPU Google
Kompilasi Edge
- Google AI Edge menggunakan StableHLO sebagai format input
untuk di-deploy ke perangkat seluler menggunakan LiteRT
- AI Edge Torch mengekspor
model PyTorch untuk deployment seluler melalui StableHLO
- AI Edge Torch mengekspor
model PyTorch untuk deployment seluler melalui StableHLO
Alat dan Visualisasi
- Penjelajah Model menawarkan
visualisasi grafik hierarkis dengan dukungan untuk model StableHLO