All

22 repositories

pathways-utils
Public
Package of Pathways-on-Cloud utilities
Python
•
Apache License 2.0
•11•25•1•7•Updated Mar 28, 2026Mar 28, 2026
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
•494•2.2k•79•172•Updated Mar 28, 2026Mar 28, 2026
maxdiffusion
Public
Python
•
Apache License 2.0
•66•319•7•20•Updated Mar 28, 2026Mar 28, 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
•68•123•7•20•Updated Mar 27, 2026Mar 27, 2026
tpu-recipes
Public
Shell
•
Apache License 2.0
•53•80•6•17•Updated Mar 27, 2026Mar 27, 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
•82•172•26•2•Updated Mar 27, 2026Mar 27, 2026
RecML
Public
Python
•
Apache License 2.0
•22•216•3•82•Updated Mar 26, 2026Mar 26, 2026
accelerator-microbenchmarks
Public
Python
•
Apache License 2.0
•34•12•3•7•Updated Mar 25, 2026Mar 25, 2026
accelerator-agents
Public
Python
•
Apache License 2.0
•0•4•0•0•Updated Mar 24, 2026Mar 24, 2026
google-cloud-mldiagnostics
Public
Python
•
Apache License 2.0
•2•8•0•0•Updated Mar 20, 2026Mar 20, 2026
cloud-accelerator-diagnostics
Public
Python
•
Apache License 2.0
•14•27•6•0•Updated Mar 17, 2026Mar 17, 2026
inference-benchmark
Public archive
Python
•
Apache License 2.0
•15•18•7•4•Updated Mar 11, 2026Mar 11, 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
kithara
Public archive
Python
•
Apache License 2.0
•9•16•0•2•Updated Feb 18, 2026Feb 18, 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
•8•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).
gpu inference pytorch
gpu inference pytorch transformer llama gpt gemma model-serving tpu jax
Python
•
Apache License 2.0
•62•418•14•13•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•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
ray-tpu
Public
Python
•
Apache License 2.0
•8•15•3•1•Updated May 11, 2025May 11, 2025
aotc
Public archive
Python
•
Other
•5•0•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

pathways-utils

maxtext

maxdiffusion

gpu-recipes

tpu-recipes

xpk

RecML

accelerator-microbenchmarks

accelerator-agents

google-cloud-mldiagnostics

cloud-accelerator-diagnostics

inference-benchmark

torchprime

cloud-tpu-monitoring-debugging

kithara

ml-goodput-measurement

cloud-diagnostics-xprof

JetStream

jetstream-pytorch

resiliency

ray-tpu

aotc

All

All

Repositories list

22 repositories