Jax
Software / App
Numerical computing library enabling gradients and multi-device training
Mentioned in 5 videos
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.