Media Summary: In this video we present an overview of the We've covered a lot of ground, from the fundamentals of Try Brilliant free for 30 days You'll also get 20% off an annual premium subscription

Leveraging The Jax Ai Stack - Detailed Analysis & Overview

In this video we present an overview of the We've covered a lot of ground, from the fundamentals of Try Brilliant free for 30 days You'll also get 20% off an annual premium subscription Spencer Huang is a Product Lead for Robotics at NVIDIA, focusing on open-source simulation frameworks, synthetic data ... Learn more: Introducing Build and Train an LLM with In this session, Robert Crowe, Product Manager at Google, spoke about Flax NNX, the new, streamlined API for building, ...

Pallas quickstart → Pallas design overview → Dive into the world of custom ...

Photo Gallery

Leveraging the JAX AI Stack
JAX AI Stack: Summary & Conclusion
JAX in 100 Seconds
Introduction to JAX and the JAX AI Stack
Spencer Huang Discusses NVIDIA's Physical AI Stack for Robotics
Build and Train an LLM with JAX
JAX AI Stack updates: Orbax, Chex, Optax, and more | JAX/OpenXLA DevLab Fall 2025
AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks
Machine Learning with JAX - From Zero to Hero | Tutorial #1
Why JAX and Flax NNX?
How to use custom JAX kernels with Pallas
Intro to JAX: Accelerating Machine Learning research
Sponsored
Sponsored
View Detailed Profile
Leveraging the JAX AI Stack

Leveraging the JAX AI Stack

In this video we present an overview of the

JAX AI Stack: Summary & Conclusion

JAX AI Stack: Summary & Conclusion

We've covered a lot of ground, from the fundamentals of

Sponsored
JAX in 100 Seconds

JAX in 100 Seconds

Try Brilliant free for 30 days https://brilliant.org/fireship You'll also get 20% off an annual premium subscription

Introduction to JAX and the JAX AI Stack

Introduction to JAX and the JAX AI Stack

In this talk we'll explore why

Spencer Huang Discusses NVIDIA's Physical AI Stack for Robotics

Spencer Huang Discusses NVIDIA's Physical AI Stack for Robotics

Spencer Huang is a Product Lead for Robotics at NVIDIA, focusing on open-source simulation frameworks, synthetic data ...

Sponsored
Build and Train an LLM with JAX

Build and Train an LLM with JAX

Learn more: https://bit.ly/4rce49q Introducing Build and Train an LLM with

JAX AI Stack updates: Orbax, Chex, Optax, and more | JAX/OpenXLA DevLab Fall 2025

JAX AI Stack updates: Orbax, Chex, Optax, and more | JAX/OpenXLA DevLab Fall 2025

Some updates on the

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

In this session, Robert Crowe, Product Manager at Google, spoke about Flax NNX, the new, streamlined API for building, ...

Machine Learning with JAX - From Zero to Hero | Tutorial #1

Machine Learning with JAX - From Zero to Hero | Tutorial #1

Become The

Why JAX and Flax NNX?

Why JAX and Flax NNX?

We'll explore the strengths of

How to use custom JAX kernels with Pallas

How to use custom JAX kernels with Pallas

Pallas quickstart → https://goo.gle/48bjFEN Pallas design overview → https://goo.gle/408hpw4 Dive into the world of custom ...

Intro to JAX: Accelerating Machine Learning research

Intro to JAX: Accelerating Machine Learning research

JAX

What is JAX?

What is JAX?

JAX