Google xla

Google XLA (Accelerated Linear Algebra) is a domain-specific open-source compiler framework managed by Google on GitHub that drastically optimizes machine learning mathematical operations across PyTorch, JAX, and TensorFlow models for execution on GPUs/TPUs.