00:51
2026-10-10
dev.to
large-language-models
Gemma 4 E2B in Pure JAX on a Colab TPU: Google's 4-Bit Export Against an Exact Repack
A developer published an Apache 2.0 Colab notebook that serves Gemma 4 E2B on a single TPU v5e chip using a pure-JAX engine and compares Google's 4-bit QAT export against an exact repack of the traineβ¦