Gemma

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.