Gemma 4 in Pure JAX: What Changes Between Turing and Ada, and What Doesn't
DEV Community
Gemma 4 in Pure JAX: What Changes Between Turing and Ada, and What Doesn't
One hand-written Gemma 4 port, no PyTorch and no vLLM, on two NVIDIA GPUs a generation apart. Most of it ports untouched. Two things do not, and one of them was quietly eating 87% of decode.
0 comments
No comments yet.