Intel® Extension for OpenXLA* 0.1.0 Experimental Release
Pre-release
Pre-release
Major Features
Intel® Extension for OpenXLA* is Intel optimized Python package to extend official OpenXLA on Intel GPU. It is based on PJRT plugin mechanism, which can seamlessly run JAX models on Intel® Data Center GPU Max Series. The PJRT API simplified the integration, which allowed the Intel XPU plugin to be developed separately and quickly integrated into JAX. This release contains following major features:
- Kernel enabling and optimization
Common kernels are enabled with LLVM/SPIRV software stack. Convolution and GEMM are enabled with OneDNN. And Stable Diffusion is verified.
Known Issues
- Limited support for collective ops due to the limitation of oneCCL.