Jax

Software / App

Numerical computing library enabling gradients and multi-device training

Mentioned in 10 videos

What podcasters actually say about Jax.

10 mentions, no marketing. Save them all to a pod and ask any question.

Get Started Free

Videos Mentioning Jax

Build and Train an LLM with 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

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

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

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

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 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.

Anthropic's Digital God, Pope vs AI, Job Loss Narrative Flips, Open Source Crackdown Coming?

Anthropic's Digital God, Pope vs AI, Job Loss Narrative Flips, Open Source Crackdown Coming?

All-In Podcast

A high-performance numerical computation library often used in machine learning, whose speed is compared against new training complexes.

The Case for AI That Improves Itself | Deep Dives with a16z

The Case for AI That Improves Itself | Deep Dives with a16z

a16z Deep Dives

A machine learning library that Merindel's models aim to be proficient with.

Stanford AA203 Optimal and Learning-Based Control | Spring 2026 | Lecture 1: Course Overview

Stanford AA203 Optimal and Learning-Based Control | Spring 2026 | Lecture 1: Course Overview

Stanford Online

A tool that will be covered in the recitations to provide foundational knowledge for the class.

Stanford AA203 Optimal and Learning-Based Control | Spring 2026 | Lecture 6: Optimal Control

Stanford AA203 Optimal and Learning-Based Control | Spring 2026 | Lecture 6: Optimal Control

Stanford Online

A library used for automatic differentiation and linearizing dynamics in sequential convex programming.