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 317 38

  3. maxdiffusion maxdiffusion Public

    Python 201 25

  4. RecML RecML Public

    Python 170 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 59 17

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,690 Apache-2.0 341 40 (2 issues need help) 177 Updated Apr 21, 2025
  • cloud-diagnostics-xprof Public
    AI-Hypercomputer/cloud-diagnostics-xprof’s past year of commit activity
    Python 2 Apache-2.0 0 1 0 Updated Apr 21, 2025
  • RecML Public
    AI-Hypercomputer/RecML’s past year of commit activity
    Python 170 Apache-2.0 13 3 5 Updated Apr 21, 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 317 Apache-2.0 38 12 11 Updated Apr 21, 2025
  • maxdiffusion Public
    AI-Hypercomputer/maxdiffusion’s past year of commit activity
    Python 201 Apache-2.0 25 4 (1 issue needs help) 8 Updated Apr 18, 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 56 Apache-2.0 15 4 0 Updated Apr 18, 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 115 Apache-2.0 32 18 29 Updated Apr 18, 2025
  • tpu-recipes Public
    AI-Hypercomputer/tpu-recipes’s past year of commit activity
    Shell 21 Apache-2.0 13 3 6 Updated Apr 17, 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 Apr 17, 2025
  • kithara Public
    AI-Hypercomputer/kithara’s past year of commit activity
    Python 13 Apache-2.0 6 0 1 Updated Apr 16, 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.

Top languages

Loading…

Most used topics

Loading…