XLA (Accelerated Linear Algebra) adalah compiler open source untuk machine learning. Compiler XLA mengambil model dari framework populer seperti PyTorch, TensorFlow, dan JAX, serta mengoptimalkan model untuk eksekusi berperforma tinggi di berbagai platform hardware termasuk GPU, CPU, dan akselerator ML.
Sebagai bagian dari project OpenXLA, XLA dibuat secara kolaboratif oleh perusahaan hardware dan software ML terkemuka di industri, termasuk Alibaba, Amazon Web Services, AMD, Apple, Arm, Google, Intel, Meta, dan NVIDIA.
Manfaat utama
Build di mana saja: XLA sudah terintegrasi ke dalam framework ML terkemuka seperti TensorFlow, PyTorch, dan JAX.
Beroperasi di mana saja: Mendukung berbagai backend, termasuk GPU, CPU, dan accelerator ML, serta menyertakan infrastruktur plug-in untuk menambahkan dukungan untuk lebih banyak lagi.
Memaksimalkan dan menskalakan performa: Fitur ini mengoptimalkan performa model dengan siklus pengoptimalan yang telah diuji produksi dan partisi otomatis untuk paralelisme model.
Menghilangkan kompleksitas: Manfaatkan kecanggihan MLIR untuk menghadirkan kemampuan terbaik ke dalam satu toolchain compiler, sehingga Anda tidak perlu mengelola berbagai compiler khusus domain.
Siap untuk masa depan: Sebagai project open source, yang dibuat melalui kolaborasi vendor hardware dan software ML terkemuka, XLA dirancang untuk beroperasi di industri ML yang canggih.
Dokumentasi
Untuk mempelajari XLA lebih lanjut, lihat link di sebelah kiri. Jika Anda adalah developer XLA baru, sebaiknya mulai dengan arsitektur XLA, lalu baca Berkontribusi.