Skip to content
/ xla Public
forked from openxla/xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

License

Notifications You must be signed in to change notification settings

zml/xla

 
 

Repository files navigation

XLA

XLA (Accelerated Linear Algebra) is an open-source machine learning (ML) compiler for GPUs, CPUs, and ML accelerators.

The XLA compiler takes models from popular ML frameworks such as PyTorch, TensorFlow, and JAX, and optimizes them for high-performance execution across different hardware platforms including GPUs, CPUs, and ML accelerators.

Get started

If you want to use XLA to compile your ML project, refer to the corresponding documentation for your ML framework:

If you're not contributing code to the XLA compiler, you don't need to clone and build this repo. Everything here is intended for XLA contributors who want to develop the compiler and XLA integrators who want to debug or add support for ML frontends and hardware backends.

Contribute

If you'd like to contribute to XLA, review How to Contribute and then see the developer guide.

Contacts

  • For questions, contact Thea Lamkin - thealamkin at google.com.

Resources

Code of Conduct

While under TensorFlow governance, all community spaces for SIG OpenXLA are subject to the TensorFlow Code of Conduct.

About

A machine learning compiler for GPUs, CPUs, and ML accelerators

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 88.9%
  • MLIR 5.8%
  • Starlark 3.8%
  • Python 0.8%
  • C 0.3%
  • Smarty 0.2%
  • Other 0.2%