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