JAX

3 statements across 3 episodes · 1 bullish · 1 bearish · 3 people on the record · first statement Oct 20, 2023 by Jeremy Howard · said 25 times in 11 episodes since 2023 · across every show →

Mentions by year

brought up most by Kevin Wang (4), Dylan Patel (4), Shawn Wang (3), Mark Huang (3), Jeremy Howard (3), Yi Tay (2), Nathan Lambert (2)

tap a year for its mentions
00531052023202420252026episodesmentions
0352023202420252026episodes it came up in
001.52.5352023202420252026episodesmentions per episode
2026 2 mentions in 1 episode
2025 5 mentions in 2 episodes 3 per episode
2024 9 mentions in 5 episodes 2 per episode
2023 9 mentions in 3 episodes 3 per episode

every mention, scene by scene, with the transcript →

Everything said about JAX, oldest first

Oct 20, 2023 positive
Assertion Not checkable as stated
Howard: JAX was a grassroots Google reaction against TensorFlow 2
“But I mean, in the meantime, I will say, you know, Google now does have a backup plan. You know, they have JAX, which was never a strategy. It was just a bunch of people who also recognized TensorFlow two as shit, and they just decided to build something else.”
Jeremy Howard Oct 20, 2023 ▶ 1:01:58 The End of Finetuning — with Jeremy Howard of Fast.ai
May 31, 2024 negative
Assertion Not checkable as stated
Huang: Berkeley's JAX Ring Attention does not work well on GPUs
“The Jaxx implementation just does not work on, on GPUs very well. Like, any naive setup that you do, like, it just won't run out of the box very easily”
Mark Huang May 31, 2024 ▶ 26:02 How to train a Million Context LLM — with Mark Huang of Gradient.ai
Jul 5, 2024 neutral
Insight
Yi Tay: AI researchers should stay framework-agnostic instead of mastering one stack
“I don't think there's any specific thing. In fact, I will try to be as, like, agnostic, like, I don't, like, I don't really say, like, okay, you need to learn JAX, you need to learn this, by the time you finish learning, there's a new framework out, anyway. So…”
Yi Tay Jul 5, 2024 ▶ 40:34 The 10,000x Yolo Researcher Metagame — with Yi Tay of Reka
Made with StarZero

Turn any episode into a week of clips.

This entire site, over 200 episodes transcribed, diarized, checked and made playable, runs on the StarZero media pipeline. Drop in your own episode and the podcast clipper finds the moments worth sharing, cuts them, captions them, and reframes them for every feed.