All

22 repositories

maxtext
Public
A simple, performant, and scalable Jax LLM!
gpt mistral fine-tuning
gpt mistral fine-tuning jax sft large-language-models llm llama2 mixtral deepseek
Python
•
Apache License 2.0
•565•2.4k•70•220•Updated Jul 18, 2026Jul 18, 2026
pathways-utils
Public
Package of Pathways-on-Cloud utilities
Python
•
Apache License 2.0
•12•31•0•23•Updated Jul 18, 2026Jul 18, 2026
maxdiffusion
Public
Python
•
Apache License 2.0
•84•374•11•9•Updated Jul 18, 2026Jul 18, 2026
ml-goodput-measurement
Public
Python
•
Apache License 2.0
•7•40•1•1•Updated Jul 17, 2026Jul 17, 2026
accelerator-agents
Public
Python
•
Apache License 2.0
•7•46•1•21•Updated Jul 16, 2026Jul 16, 2026
tpu-recipes
Public
Shell
•
Apache License 2.0
•70•115•7•24•Updated Jul 15, 2026Jul 15, 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…
gke gcloud tpu
gke gcloud tpu
Python
•
Apache License 2.0
•91•192•25•0•Updated Jul 13, 2026Jul 13, 2026
cloud-accelerator-diagnostics
Public
Python
•
Apache License 2.0
•15•32•6•0•Updated Jul 9, 2026Jul 9, 2026
accelerator-microbenchmarks
Public
Python
•
Apache License 2.0
•37•22•3•8•Updated Jul 7, 2026Jul 7, 2026
RecML
Public
Python
•
Apache License 2.0
•22•228•3•60•Updated Jul 2, 2026Jul 2, 2026
gpu-recipes
Public
Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.
gpu benchmarks google-cloud-platform
gpu benchmarks google-cloud-platform serving distributed-training
Python
•
Apache License 2.0
•81•138•8•24•Updated Jul 1, 2026Jul 1, 2026
google-cloud-mldiagnostics
Public
Python
•
Apache License 2.0
•3•12•2•1•Updated Jun 10, 2026Jun 10, 2026
cloud-tpu-monitoring-debugging
Public
HCL
•
Apache License 2.0
•5•17•0•1•Updated Apr 8, 2026Apr 8, 2026
inference-benchmark
Public archive
Python
•
Apache License 2.0
•15•22•7•4•Updated Mar 11, 2026Mar 11, 2026
torchprime
Public archive
torchprime is a reference model implementation for PyTorch on TPU.
Python
•
Other
•15•48•88•17•Updated Mar 3, 2026Mar 3, 2026
kithara
Public archive
Python
•
Apache License 2.0
•10•19•0•2•Updated Feb 18, 2026Feb 18, 2026
cloud-diagnostics-xprof
Public
Python
•
Apache License 2.0
•8•17•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).
gpu inference pytorch
gpu inference pytorch transformer llama gpt gemma model-serving tpu jax
Python
•
Apache License 2.0
•67•451•14•11•Updated Jan 5, 2026Jan 5, 2026
jetstream-pytorch
Public archive
PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
inference pytorch batching
inference pytorch batching attention llama gemma model-serving tpu llm llm-inference
Python
•
Apache License 2.0
•21•86•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•14•1•1•Updated Sep 29, 2025Sep 29, 2025
ray-tpu
Public
Python
•
Apache License 2.0
•8•15•3•1•Updated May 11, 2025May 11, 2025
aotc
Public archive
Python
•
Other
•5•1•1•1•Updated May 8, 2025May 8, 2025

ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

AI Hypercomputer

All

All

22 repositories

maxtext

pathways-utils

maxdiffusion

ml-goodput-measurement

accelerator-agents

tpu-recipes

xpk

cloud-accelerator-diagnostics

accelerator-microbenchmarks

RecML

gpu-recipes

google-cloud-mldiagnostics

cloud-tpu-monitoring-debugging

inference-benchmark

torchprime

kithara

cloud-diagnostics-xprof

JetStream

jetstream-pytorch

resiliency

ray-tpu

aotc

All

All

Repositories list

22 repositories