Jax
Software / App
Numerical computing library enabling gradients and multi-device training
Mentioned in 6 videos
Common Themes
Videos Mentioning Jax
Build and Train an LLM with JAX
DeepLearningAI
Numerical computing library enabling gradients and multi-device training
A Comprehensive Overview of Large Language Models - Latent Space Paper Club
Latent Space
A high-performance numerical computation library, often used for machine learning.
The End of Finetuning — with Jeremy Howard of Fast.ai
Latent Space
A backup plan for Google's AI future after TensorFlow 2's perceived failure, initially intended as a research project but now a key framework.
AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks
DeepLearningAI
Information Theory for Language Models: Jack Morris
Latent Space
A framework that some people switch from to learn CUDA.
Stanford CS336 Language Modeling from Scratch | Spring 2026 | Lecture 5: GPUs, TPUs
Stanford Online
A high-performance numerical computation library that has a compiler for optimizing code, mentioned in the context of TPUs and their architecture.