Leveraging the JAX AI Stack
9:44
Leveraging the JAX AI Stack
2.7K views5 months ago
YouTubeGoogle for Developers
Unlocking Low-Level Control: Customizing Keras Training Loops with JAX
6:33
Unlocking Low-Level Control: Customizing Keras Training Loop…
1.6K views2 weeks ago
YouTubeGoogle for Developers
Understanding JAX: JIT, XLA, and Pure Functions Explained
10:06
Understanding JAX: JIT, XLA, and Pure Functions Explained
19.4K views4 months ago
YouTubeGoogle for Developers
Build a Transformer with JAX
27:56
Build a Transformer with JAX
4K views1 year ago
YouTubeGoogle for Developers
Keras 3 Distributed Training: Scaling Models with JAX using DataParallel, and ModelParallel
6:51
Keras 3 Distributed Training: Scaling Models with JAX using Da…
2.6K views2 months ago
YouTubeGoogle for Developers
⚡ JAX Tutorial for Beginners | 100× Faster Than NumPy Explained
13:15
⚡ JAX Tutorial for Beginners | 100× Faster Than NumPy Explained
330 views3 months ago
YouTubeMLWorks
Debugging JAX & Flax NNX (Part 1)
6:42
Debugging JAX & Flax NNX (Part 1)
501 views5 months ago
YouTubeGoogle for Developers
40:46
The Ultimate JAX Beginner First Steps into High-Performance AI
3 views1 month ago
YouTubeCoding Hives
12:34
Debugging JAX & Flax NNX (Part 2)
364 views5 months ago
YouTubeGoogle for Developers
31:25
Large-scale JAX development with JAX-Toolbox
75 views7 months ago
YouTubePython Milano
38:00
AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Gui…
844 views5 months ago
YouTubeDeepLearningAI
9:39
Debugging JAX & Flax NNX (Part 3)
232 views5 months ago
YouTubeGoogle for Developers
7:13
JAX AI Stack: Summary & Conclusion
334 views5 months ago
YouTubeGoogle for Developers
46:17
Keras: Deep Learning Framework for JAX | JAX/OpenXLA DevLab F…
690 views5 months ago
YouTubeOpenXLA
10:02
Serving JAX Models with vLLM & SGLang
520 views5 months ago
YouTubeGoogle for Developers
1:55
Build and Train an LLM with JAX
6.4K views2 months ago
YouTubeDeepLearningAI
16:30
What is JAX?
10.6K views1 year ago
YouTubeGoogle for Developers
11:11
Why JAX and Flax NNX?
3.9K views5 months ago
YouTubeGoogle for Developers
1:01:53
Dan Foreman Mackey: JAX Training
82 views4 months ago
YouTubeFlatiron Institute
38:05
Exploratory Prompting - 6-Day Study Plans - Build and Train an L…
48 views2 months ago
YouTubeThe Flourishing
42:11
Introduction to JAX and the JAX AI Stack
416 views10 months ago
YouTubeGDG AI for Science
3:00
3 Minute Jax Guide - A Guide for League of Legends
22.1K views6 months ago
YouTube3 Minute League of Legends
10:32
Introducing Flax NNX (Part 1)
1.8K views5 months ago
YouTubeGoogle for Developers
6:03
Introducing Flax NNX (Part 3)
946 views5 months ago
YouTubeGoogle for Developers
3:00
3 Minute Jax Guide - A Guide for League of Legends
54.8K views10 months ago
YouTube3 Minute League of Legends
1:17:57
Machine Learning with JAX - From Zero to Hero | Tutorial #1
76.7K viewsOct 31, 2021
YouTubeAleksa Gordić - The AI Epiphany
1:05:24
Find in video from 02:02What is JAX?
ETH Zürich AISE: Introduction to JAX
2.5K viewsJul 24, 2024
YouTubeCAMLab, ETH Zürich
0:21
Get started with JAX using NNX 🦾🧠
2.7K viewsJan 28, 2025
YouTubeGoogle Cloud Tech
8:12
Find in video from 05:55Grand Illusion study
Demo: Gemma 2 architecture: JAX, Flax, and more
12.9K viewsOct 18, 2024
YouTubeGoogle for Developers
8:12
Find in video from 01:02Overview of JAX
Demo: JAX, Flax and Gemma
7.4K viewsApr 2, 2024
YouTubeGoogle for Developers
See more videos
Static thumbnail place holder