Repositories list

• maxtext

  Public

  AI-Hypercomputer/maxtext’s past year of commit activity
  A simple, performant and scalable Jax LLM!

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

  Python

  •

  Apache License 2.0

  •475•2.2k•75•175•Updated Feb 26, 2026Feb 26, 2026

• maxdiffusion

  Public

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

  •

  Apache License 2.0

  •62•313•5•16•Updated Feb 26, 2026Feb 26, 2026

• 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 GP…

  gkegcloudtpu

  Python

  •

  Apache License 2.0

  •79•170•24•9•Updated Feb 26, 2026Feb 26, 2026

• gpu-recipes

  Public

  AI-Hypercomputer/gpu-recipes’s past year of commit activity
  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

  •61•117•6•13•Updated Feb 26, 2026Feb 26, 2026

• pathways-utils

  Public

  AI-Hypercomputer/pathways-utils’s past year of commit activity
  Package of Pathways-on-Cloud utilities

  Python

  •

  Apache License 2.0

  •11•25•0•10•Updated Feb 26, 2026Feb 26, 2026

• accelerator-microbenchmarks

  Public

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

  •

  Apache License 2.0

  •31•11•3•8•Updated Feb 25, 2026Feb 25, 2026

• tpu-recipes

  Public

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

  •

  Apache License 2.0

  •49•77•6•13•Updated Feb 23, 2026Feb 23, 2026

• google-cloud-mldiagnostics

  Public

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

  •

  Apache License 2.0

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

• RecML

  Public

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

  •

  Apache License 2.0

  •22•214•3•81•Updated Feb 18, 2026Feb 18, 2026

• torchprime

  Public

  AI-Hypercomputer/torchprime’s past year of commit activity
  torchprime is a reference model implementation for PyTorch on TPU.

  Python

  •

  Other

  •14•45•88•17•Updated Feb 18, 2026Feb 18, 2026

• kithara

  Public

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

  •

  Apache License 2.0

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

• accelerator-agents

  Public

  AI-Hypercomputer/accelerator-agents’s past year of commit activity
  Apache License 2.0

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

• ml-goodput-measurement

  Public

  AI-Hypercomputer/ml-goodput-measurement’s past year of commit activity
  Python

  •

  Apache License 2.0

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

• cloud-diagnostics-xprof

  Public

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

  •

  Apache License 2.0

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

• JetStream

  Public

  AI-Hypercomputer/JetStream’s past year of commit activity
  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

  •59•413•14•12•Updated Jan 5, 2026Jan 5, 2026

• cloud-accelerator-diagnostics

  Public

  AI-Hypercomputer/cloud-accelerator-diagnostics’s past year of commit activity
  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+ 8attentionllamagemmamodel-servingtpullmllm-inference+ 1

  Python

  •

  Apache License 2.0

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

• resiliency

  Public

  AI-Hypercomputer/resiliency’s past year of commit activity
  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

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

  •

  Apache License 2.0

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

• ray-tpu

  Public

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

  •

  Apache License 2.0

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

• aotc

  Public

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

  •

  Other

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

• cloud-tpu-monitoring-debugging

  Public

  AI-Hypercomputer/cloud-tpu-monitoring-debugging’s past year of commit activity
  HCL

  •

  Apache License 2.0

  •5•14•0•2•Updated Feb 18, 2025Feb 18, 2025