GitHub repository: zyxie/xla
Description: A machine learning compiler for GPUs, CPUs, and ML accelerators