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
Alat dan Visualisasi
- Penjelajah Model menawarkan visualisasi grafik hierarkis dengan dukungan untuk model StableHLO