Repositories list

• maxtext

  Public
  A simple, performant and scalable Jax LLM!

  gptmistralfine-tuning+ 11jaxsftlarge-language-modelsllmllama2mixtraldeepseek+ 4

  Python

  •

  Apache License 2.0

  •421•2k•70•239•Updated Nov 11, 2025Nov 11, 2025

• RecML

  Public

  AI-Hypercomputer/RecML’s past year of commit activity
  Python

  •

  Apache License 2.0

  •17•211•3•63•Updated Nov 11, 2025Nov 11, 2025

• maxdiffusion

  Public

  AI-Hypercomputer/maxdiffusion’s past year of commit activity
  Python

  •

  Apache License 2.0

  •43•274•5•17•Updated Nov 11, 2025Nov 11, 2025

• tpu-recipes

  Public

  AI-Hypercomputer/tpu-recipes’s past year of commit activity
  Shell

  •

  Apache License 2.0

  •25•53•4•11•Updated Nov 11, 2025Nov 11, 2025

• google-cloud-mldiagnostics

  Public

  AI-Hypercomputer/google-cloud-mldiagnostics’s past year of commit activity
  Python

  •

  Apache License 2.0

  •0•2•0•0•Updated Nov 10, 2025Nov 10, 2025

• xpk

  Public

  AI-Hypercomputer/xpk’s past year of commit activity
  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 GPUs on GKE.

  gkegcloudtpu

  Python

  •

  Apache License 2.0

  •57•151•26•34•Updated Nov 10, 2025Nov 10, 2025

• accelerator-microbenchmarks

  Public
  Python

  •

  Apache License 2.0

  •9•5•0•4•Updated Nov 10, 2025Nov 10, 2025

• cloud-diagnostics-xprof

  Public

  AI-Hypercomputer/cloud-diagnostics-xprof’s past year of commit activity
  Python

  •

  Apache License 2.0

  •2•12•0•0•Updated Nov 10, 2025Nov 10, 2025

• gpu-recipes

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

  gpubenchmarksgoogle-cloud-platform+ 2servingdistributed-training

  Python

  •

  Apache License 2.0

  •44•96•6•7•Updated Nov 5, 2025Nov 5, 2025

• cloud-accelerator-diagnostics

  Public
  Python

  •

  Apache License 2.0

  •12•24•5•0•Updated Oct 29, 2025Oct 29, 2025

• pathways-utils

  Public
  Package of Pathways-on-Cloud utilities

  Python

  •

  Apache License 2.0

  •4•20•0•3•Updated Oct 16, 2025Oct 16, 2025

• torchprime

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

  Python

  •

  Other

  •9•40•89•14•Updated Oct 15, 2025Oct 15, 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

  •0•9•1•0•Updated Sep 29, 2025Sep 29, 2025

• jetstream-pytorch

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

  inferencepytorchbatching+ 8attentionllamagemmamodel-servingtpullmllm-inference+ 1

  Python

  •

  Apache License 2.0

  •18•77•10•3•Updated Sep 15, 2025Sep 15, 2025

• ml-goodput-measurement

  Public
  Python

  •

  Apache License 2.0

  •3•25•1•0•Updated Sep 11, 2025Sep 11, 2025

• inference-benchmark

  Public

  AI-Hypercomputer/inference-benchmark’s past year of commit activity
  Python

  •

  Apache License 2.0

  •13•17•7•4•Updated Jun 18, 2025Jun 18, 2025

• 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+ 13transformerllamagptgemmamodel-servingtpujax+ 6

  Python

  •

  Apache License 2.0

  •52•388•14•12•Updated Jun 10, 2025Jun 10, 2025

• kithara

  Public
  Python

  •

  Apache License 2.0

  •6•16•0•2•Updated May 19, 2025May 19, 2025

• ray-tpu

  Public
  Python

  •

  Apache License 2.0

  •4•15•3•0•Updated May 11, 2025May 11, 2025

• aotc

  Public
  Python

  •

  Other

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

• cloud-tpu-monitoring-debugging

  Public
  HCL

  •

  Apache License 2.0

  •1•13•0•0•Updated Feb 18, 2025Feb 18, 2025