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 1.8k 360

  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 341 46

  3. maxdiffusion maxdiffusion Public

    Python 219 33

  4. RecML RecML Public

    Python 181 14

  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 122 39

  6. gpu-recipes gpu-recipes Public

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

    Python 70 25

Repositories

Showing 10 of 19 repositories
  • maxtext Public

    A simple, performant and scalable Jax LLM!

    AI-Hypercomputer/maxtext’s past year of commit activity
    Python 1,761 Apache-2.0 360 49 (2 issues need help) 197 Updated Jun 13, 2025
  • maxdiffusion Public
    AI-Hypercomputer/maxdiffusion’s past year of commit activity
    Python 219 Apache-2.0 33 4 (1 issue needs help) 11 Updated Jun 13, 2025
  • torchprime Public

    torchprime is a reference model implementation for PyTorch on TPU.

    AI-Hypercomputer/torchprime’s past year of commit activity
    Python 28 5 79 (8 issues need help) 9 Updated Jun 12, 2025
  • AI-Hypercomputer/inference-benchmark’s past year of commit activity
    Python 10 Apache-2.0 11 7 3 Updated Jun 12, 2025
  • 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 122 Apache-2.0 39 24 31 Updated Jun 12, 2025
  • pathways-utils Public

    Package of Pathways-on-Cloud utilities

    AI-Hypercomputer/pathways-utils’s past year of commit activity
    Python 15 Apache-2.0 2 0 2 Updated Jun 12, 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).

    AI-Hypercomputer/JetStream’s past year of commit activity
    Python 341 Apache-2.0 46 12 12 Updated Jun 10, 2025
  • RecML Public
    AI-Hypercomputer/RecML’s past year of commit activity
    Python 181 Apache-2.0 14 3 8 Updated Jun 10, 2025
  • 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 70 Apache-2.0 25 4 2 Updated Jun 10, 2025
  • tpu-recipes Public
    AI-Hypercomputer/tpu-recipes’s past year of commit activity
    Shell 33 Apache-2.0 17 3 7 Updated Jun 9, 2025

People

This organization has no public members. You must be a member to see who’s a part of this organization.