xla 6
- How JAX Shards a Computation Across a Mesh
- What JAX Traces, and What It Refuses
- Programming the TPU: What Its Open-Source Compiler Already Tells You
- XLA Up Close: The Performance Bargain and Its Rigid Price
- A Tour of XLA: The Compiler Beneath JAX, TensorFlow, and PyTorch
- A tour of MLIR: The Dialect Stack Everyone Depends On