Skip to content
@AI-Hypercomputer

AI Hypercomputer

Reference implementations, benchmarks, recipes, and all things Google Cloud AI Hypercomputer

Popular repositories Loading

  1. maxtext maxtext Public

    A simple, performant and scalable Jax LLM!

    Python 2.1k 447

  2. JetStream 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).

    Python 398 58

  3. maxdiffusion maxdiffusion Public

    Python 296 53

  4. RecML RecML Public

    Python 214 20

  5. xpk 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 GPUs on GKE.

    Python 158 66

  6. gpu-recipes gpu-recipes Public

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

    Python 109 55

Repositories

Showing 10 of 21 repositories
  • maxdiffusion Public
    AI-Hypercomputer/maxdiffusion’s past year of commit activity
    Python 296 Apache-2.0 53 5 (1 issue needs help) 19 Updated Jan 10, 2026
  • maxtext Public

    A simple, performant and scalable Jax LLM!

    AI-Hypercomputer/maxtext’s past year of commit activity
    Python 2,080 Apache-2.0 447 76 (2 issues need help) 180 Updated Jan 10, 2026
  • gpu-recipes Public

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

    AI-Hypercomputer/gpu-recipes’s past year of commit activity
    Python 109 Apache-2.0 55 6 11 Updated Jan 10, 2026
  • AI-Hypercomputer/accelerator-microbenchmarks’s past year of commit activity
    Python 9 Apache-2.0 21 3 5 Updated Jan 10, 2026
  • AI-Hypercomputer/ml-goodput-measurement’s past year of commit activity
    Python 27 Apache-2.0 6 1 0 Updated Jan 9, 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 GPUs on GKE.

    AI-Hypercomputer/xpk’s past year of commit activity
    Python 158 Apache-2.0 66 22 2 Updated Jan 9, 2026
  • tpu-recipes Public
    AI-Hypercomputer/tpu-recipes’s past year of commit activity
    Shell 68 Apache-2.0 43 5 12 Updated Jan 9, 2026
  • RecML Public
    AI-Hypercomputer/RecML’s past year of commit activity
    Python 214 Apache-2.0 20 3 76 Updated Jan 9, 2026
  • pathways-utils Public

    Package of Pathways-on-Cloud utilities

    AI-Hypercomputer/pathways-utils’s past year of commit activity
    Python 22 Apache-2.0 10 0 9 Updated Jan 8, 2026
  • torchprime Public

    torchprime is a reference model implementation for PyTorch on TPU.

    AI-Hypercomputer/torchprime’s past year of commit activity
    Python 44 12 89 (8 issues need help) 15 Updated Jan 6, 2026