Updated code with latest jax-tpu-embedding wheel to run on V7#147
Merged
ajkv-google merged 6 commits intomainfrom Jan 21, 2026
Merged
Updated code with latest jax-tpu-embedding wheel to run on V7#147ajkv-google merged 6 commits intomainfrom
ajkv-google merged 6 commits intomainfrom
Commits
Commits on Jan 21, 2026
- committed
- committed
- committed
- authored
- committed
- committed
