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 344

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

  3. maxdiffusion maxdiffusion Public

    Python 202 27

  4. RecML RecML Public

    Python 178 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 118 33

  6. gpu-recipes gpu-recipes Public

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

    Python 61 16

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,709 Apache-2.0 344 42 (2 issues need help) 184 Updated May 6, 2025
  • aotc Public
    AI-Hypercomputer/aotc’s past year of commit activity
    Python 0 0 0 0 Updated May 6, 2025
  • tpu-recipes Public
    AI-Hypercomputer/tpu-recipes’s past year of commit activity
    Shell 26 Apache-2.0 14 3 7 Updated May 6, 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 322 Apache-2.0 39 12 10 Updated May 6, 2025
  • RecML Public
    AI-Hypercomputer/RecML’s past year of commit activity
    Python 178 Apache-2.0 14 3 6 Updated May 6, 2025
  • AI-Hypercomputer/cloud-accelerator-diagnostics’s past year of commit activity
    Python 20 Apache-2.0 9 3 0 Updated May 6, 2025
  • torchprime Public

    torchprime is a reference model implementation for PyTorch on TPU.

    AI-Hypercomputer/torchprime’s past year of commit activity
    Python 17 3 69 (9 issues need help) 6 Updated May 5, 2025
  • AI-Hypercomputer/cloud-diagnostics-xprof’s past year of commit activity
    Python 4 Apache-2.0 1 1 0 Updated May 5, 2025
  • maxdiffusion Public
    AI-Hypercomputer/maxdiffusion’s past year of commit activity
    Python 202 Apache-2.0 27 4 (1 issue needs help) 9 Updated May 2, 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 118 Apache-2.0 33 17 28 Updated May 1, 2025

People

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

Most used topics

Loading…