Blog#
How Lightricks trains video diffusion models at scale with JAX on TPU - November 11, 2025 by Yaki Bitterman & Yoav HaCohen, PhD
How JAX makes high-performance economics accessible - November 11, 2025 by John Stachurski & Srikanth Kilaru
AI Innovators: How JAX on TPU is helping Escalante advance AI-driven protein design - September 23, 2025 by Srikanth Kilaru
Beyond backpropagation: JAX’s symbolic power unlocks new frontiers in scientific computing - September 9, 2025 by Srikanth Kilaru, Zekun Shi, Min Lin
An efficient path to production AI: Kakao’s journey with JAX and Cloud TPUs - August 19, 2025 by Minho Ryu, Nayeon Kim, Srikanth Kilaru
A roboticist’s journey with JAX: Finding efficiency in optimal control and simulation - July 29, 2025 by Srikanth Kilaru, Max Muchen Sun
Stanford’s Marin foundation model: The first fully open model developed using JAX - July 16, 2025 by Srikanth Kilaru, David Hall