All Tools
O
Dev ToolsFreeOpen Source
OPENXLA
Open-source ML compiler for optimizing models across GPUs, CPUs, and accelerators
Apache-2.0
ABOUT
ML models need to run efficiently across diverse hardware (GPUs, CPUs, TPUs, custom accelerators), but manually optimizing kernels for each target is impractical and non-portable. OpenXLA solves this by providing a compiler framework that automatically optimizes and compiles ML graphs for any target hardware — generating efficient kernels, fusing operations, and applying hardware-specific optimizations without manual tuning.
INTEGRATION GUIDE
1. AOT-compile trained models for deployment on edge devices, mobile, and embedded systems with optimized kernels
2. JIT-compile and optimize PyTorch and TensorFlow model graphs for GPU training with automatic kernel fusion
3. Target custom AI accelerators and emerging hardware architectures without writing device-specific kernels
4. Reduce model inference latency through operation fusion, memory planning, and quantization-aware compilation
5. Enable cross-device portability where the same model binary runs on CPUs, GPUs, and specialized accelerators
TAGS
ml-compilergpuoptimizationjitkernel-generationhardware-acceleration