feat: modernize JAX/TPU distribution with Keras 3 on Kaggle v5e-8 (Fixes #275)#403

Paramveersingh-S wants to merge 4 commits into

google-gemma:maingoogle-gemma/cookbook:mainfrom

Paramveersingh-S:feat/tpu-v5e-parallelismParamveersingh-S/cookbook:feat/tpu-v5e-parallelismCopy head branch name to clipboard