
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
The Lens
XLA is the machine learning compiler that takes models from JAX, PyTorch, or TensorFlow and optimizes them for GPUs, CPUs, TPUs, and other accelerators. If you use any of those frameworks you're already running XLA, you just don't see it. The OpenXLA repo is where the compiler itself lives, and it's open source under Apache 2.0.
Self-hosting in the normal sense doesn't apply. XLA is infrastructure that ML frameworks call into. End developers interact with it through `jax.jit`, `torch.compile` (with the OpenXLA backend), or TensorFlow's XLA flags. Building from source is a serious commitment that involves Bazel, platform-specific dependencies, and several days of patience. Most teams should consume the pre-built versions their framework ships.
Use this directly if you're doing ML infrastructure work: writing a new framework, adding hardware support, or debugging compilation issues. Solo developers and small teams: you don't interact with XLA directly, you benefit from it. Large teams doing custom silicon or model serving optimization: this is the layer you're working in.
The catch: governance still effectively sits with Google despite the OpenXLA branding. The Code of Conduct points back to TensorFlow governance, and Google contributes the majority of the changes. Apache 2.0 protects against the worst outcomes, but "open governance" reads aspirationally here.
Get tools like this every Wednesday
One featured tool, three on the radar. No fluff.
Free vs Self-Hosted vs Paid
fully free**Free:** Apache 2.0 source. Used at no cost in every major ML framework.
**Self-hosted:** XLA runs wherever your ML workload runs. The compilation overhead is real but amortized over training and inference runs. No separate service to deploy.
**Paid:** None for XLA itself. Google sells TPUs and Vertex AI on the cloud side, but those are products that use XLA, not XLA itself.
Free Apache 2.0 ML compiler. You're already using it through your framework.
License: Apache License 2.0
Use freely. Patent grant included.
Commercial use: ✓ Yes
About
- Owner
- OpenXLA (Organization)
- Stars
- 4,259
- Forks
- 806
Explore Further
More tools in the directory
openclaw
Your own personal AI assistant. Any OS. Any Platform. The lobster way. 🦞
371.0k ★claw-code
The repo is finally unlocked. enjoy the party! The fastest repo in history to surpass 100K stars ⭐. Join Discord: https://discord.gg/5TUQKqFWd Built in Rust using oh-my-codex.
191.1k ★n8n
Fair-code workflow automation with native AI capabilities
187.5k ★