Repositories list

• maxtext

  Public
  A simple, performant and scalable Jax LLM!

  gptmistralfine-tuning+ 11

  gptmistralfine-tuningjaxsftlarge-language-modelsllmllama2mixtraldeepseek+ 4

  Python

  •

  Apache License 2.0

  •481•2.2k•75•183•Updated Mar 7, 2026Mar 7, 2026

• maxdiffusion

  Public
  Python

  •

  Apache License 2.0

  •63•316•6•17•Updated Mar 6, 2026Mar 6, 2026

• xpk

  Public
  xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerators such as TPUs and GP…

  gkegcloudtpu

  gkegcloudtpu

  Python

  •

  Apache License 2.0

  •79•171•24•12•Updated Mar 6, 2026Mar 6, 2026

• accelerator-microbenchmarks

  Public
  Python

  •

  Apache License 2.0

  •31•11•3•8•Updated Mar 6, 2026Mar 6, 2026

• gpu-recipes

  Public
  Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.

  gpubenchmarksgoogle-cloud-platform+ 2

  gpubenchmarksgoogle-cloud-platformservingdistributed-training

  Python

  •

  Apache License 2.0

  •61•118•6•14•Updated Mar 6, 2026Mar 6, 2026

• pathways-utils

  Public
  Package of Pathways-on-Cloud utilities

  Python

  •

  Apache License 2.0

  •11•25•1•7•Updated Mar 5, 2026Mar 5, 2026

• tpu-recipes

  Public
  Shell

  •

  Apache License 2.0

  •50•76•6•13•Updated Mar 4, 2026Mar 4, 2026

• torchprime

  Public archive
  torchprime is a reference model implementation for PyTorch on TPU.

  Python

  •

  Other

  •14•46•88•17•Updated Mar 3, 2026Mar 3, 2026

• cloud-tpu-monitoring-debugging

  Public
  HCL

  •

  Apache License 2.0

  •5•14•0•1•Updated Mar 2, 2026Mar 2, 2026

• RecML

  Public
  Python

  •

  Apache License 2.0

  •22•215•3•81•Updated Feb 27, 2026Feb 27, 2026

• google-cloud-mldiagnostics

  Public
  Python

  •

  Apache License 2.0

  •2•8•0•0•Updated Feb 19, 2026Feb 19, 2026

• kithara

  Public
  Python

  •

  Apache License 2.0

  •9•16•0•2•Updated Feb 18, 2026Feb 18, 2026

• accelerator-agents

  Public
  Apache License 2.0

  •0•1•0•0•Updated Feb 4, 2026Feb 4, 2026

• ml-goodput-measurement

  Public
  Python

  •

  Apache License 2.0

  •6•33•1•0•Updated Feb 4, 2026Feb 4, 2026

• cloud-diagnostics-xprof

  Public
  Python

  •

  Apache License 2.0

  •7•12•0•1•Updated Jan 23, 2026Jan 23, 2026

• JetStream

  Public
  JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

  gpuinferencepytorch+ 13

  gpuinferencepytorchtransformerllamagptgemmamodel-servingtpujax+ 6

  Python

  •

  Apache License 2.0

  •58•415•14•12•Updated Jan 5, 2026Jan 5, 2026

• cloud-accelerator-diagnostics

  Public
  Python

  •

  Apache License 2.0

  •14•27•5•0•Updated Dec 23, 2025Dec 23, 2025

• jetstream-pytorch

  Public archive
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"

  inferencepytorchbatching+ 8

  inferencepytorchbatchingattentionllamagemmamodel-servingtpullmllm-inference+ 1

  Python

  •

  Apache License 2.0

  •21•79•10•3•Updated Dec 18, 2025Dec 18, 2025

• resiliency

  Public
  This project provides resilient training logic to improve the training efficiency of large-scale distributed workloads on Google Cloud.

  Python

  •

  Apache License 2.0

  •3•12•1•1•Updated Sep 29, 2025Sep 29, 2025

• inference-benchmark

  Public
  Python

  •

  Apache License 2.0

  •15•18•7•4•Updated Jun 18, 2025Jun 18, 2025

• ray-tpu

  Public
  Python

  •

  Apache License 2.0

  •8•15•3•1•Updated May 11, 2025May 11, 2025

• aotc

  Public
  Python

  •

  Other

  •4•0•1•1•Updated May 8, 2025May 8, 2025