X

XLA

Software / AppMentioned in 1 video

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

Videos Mentioning XLA

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.