Tag

Jax

Every published article tagged with Jax.

AIAug 31, 2026

Porting Gemma 4 in Pure JAX Across TPU and GPU Architectures

Deploy gemma 4 jax tpu models. Fix execution abstraction leaks, compare performance disparities, optimize compiler behavior across hardware accelerators.

6 min read

Jax Gemma