Core concept
HLO (High-Level Ops) intermediate. Backends: GPU (via CUDA), TPU, CPU.
Advertisement
How it works
Compile time slow. Some ops unsupported. Debugging generated code hard.
Advertisement
Trade-offs + gotchas
JAX-heavy workflows. TPU (JAX is preferred stack). PyTorch/XLA on TPU.