Jax
3 mentions across 2 people
Visit ↗All mentions
“Tutorial on fine tuning Gemma on TPU v5 using Kinetic + Keras + JAX. Easiest stack to fully leverage TPUs at scale.”
Fine-tuning Gemma on TPU v5 with Kinetic, Keras, and JAX ↗“JAX is what a well-designed low-level machine learning framework looks like. Good design lets you deliver much greater performance with much lower effort.”
JAX: Exemplar of Efficient ML Framework Design ↗
“JAX is the open-source library behind Google's Gemini, Veo, and other advanced models. This short course teaches you to build and train a 20-million parameter language model from scratch using JAX and its ecosystem of tools.”
New Course Teaches LLM Development with JAX ↗