XLA

Software / App

The compiler layer below Jax that optimizes Jax code into machine code for accelerators like GPUs and TPUs, handling automatic distribution and communication primitives.

Mentioned in 3 videos

Videos Mentioning XLA

Ep 18: Petaflops to the People — with George Hotz of tinycorp

Ep 18: Petaflops to the People — with George Hotz of tinycorp

Latent Space

XLA is mentioned as a complex instruction system, contrasted with tiny grad's restricted instruction system.

Chris Lattner: Compilers, LLVM, Swift, TPU, and ML Accelerators | Lex Fridman Podcast #21

Chris Lattner: Compilers, LLVM, Swift, TPU, and ML Accelerators | Lex Fridman Podcast #21

Lex Fridman

Google's compiler system for TensorFlow, one of the hardware-specific compilers that MLIR aims to integrate.

AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks

AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks

DeepLearningAI

The compiler layer below Jax that optimizes Jax code into machine code for accelerators like GPUs and TPUs, handling automatic distribution and communication primitives.