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.7k 341

  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 320 40

  3. maxdiffusion maxdiffusion Public

    Python 201 26

  4. RecML RecML Public

    Python 171 13

  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 115 32

  6. jetstream-pytorch jetstream-pytorch Public

    PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"

    Python 60 17

Repositories

Showing 10 of 19 repositories
  • 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 115 Apache-2.0 32 17 27 Updated Apr 25, 2025
  • maxtext Public

    A simple, performant and scalable Jax LLM!

    AI-Hypercomputer/maxtext’s past year of commit activity
    Python 1,701 Apache-2.0 341 42 (2 issues need help) 184 Updated Apr 25, 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 320 Apache-2.0 40 12 12 Updated Apr 25, 2025
  • maxdiffusion Public
    AI-Hypercomputer/maxdiffusion’s past year of commit activity
    Python 201 Apache-2.0 26 4 (1 issue needs help) 9 Updated Apr 24, 2025
  • pathways-utils Public

    Package of Pathways-on-Cloud utilities

    AI-Hypercomputer/pathways-utils’s past year of commit activity
    Python 14 Apache-2.0 2 0 4 Updated Apr 24, 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 57 Apache-2.0 16 4 0 Updated Apr 24, 2025
  • torchprime Public

    torchprime is a reference model implementation for PyTorch on TPU.

    AI-Hypercomputer/torchprime’s past year of commit activity
    Python 15 3 71 (10 issues need help) 9 Updated Apr 24, 2025
  • RecML Public
    AI-Hypercomputer/RecML’s past year of commit activity
    Python 171 Apache-2.0 13 3 8 Updated Apr 24, 2025
  • accelerator-microbenchmarks Public
    AI-Hypercomputer/accelerator-microbenchmarks’s past year of commit activity
    Python 1 Apache-2.0 2 0 0 Updated Apr 23, 2025
  • tpu-recipes Public
    AI-Hypercomputer/tpu-recipes’s past year of commit activity
    Shell 23 Apache-2.0 13 3 6 Updated Apr 23, 2025
View all repositories

People

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