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.6k 320

  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 275 36

  3. maxdiffusion maxdiffusion Public

    Python 186 22

  4. 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 103 31

  5. jetstream-pytorch jetstream-pytorch Public

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

    Python 51 17

  6. gpu-recipes gpu-recipes Public

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

    Dockerfile 35 6

Repositories

Showing 10 of 16 repositories
  • maxtext Public

    A simple, performant and scalable Jax LLM!

    AI-Hypercomputer/maxtext’s past year of commit activity
    Python 1,623 Apache-2.0 320 33 (2 issues need help) 139 Updated Feb 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 103 Apache-2.0 31 15 27 Updated Feb 18, 2025
  • AI-Hypercomputer/cloud-tpu-monitoring-debugging’s past year of commit activity
    HCL 10 Apache-2.0 0 0 0 Updated Feb 18, 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 275 Apache-2.0 36 11 9 Updated Feb 18, 2025
  • torchprime Public

    TorchPrime is a reference model implementation for PyTorch on TPU/GPU.

    AI-Hypercomputer/torchprime’s past year of commit activity
    Python 9 1 34 6 Updated Feb 18, 2025
  • pathways-utils Public

    Package of Pathways-on-Cloud utilities

    AI-Hypercomputer/pathways-utils’s past year of commit activity
    Python 7 Apache-2.0 2 0 1 Updated Feb 15, 2025
  • tpu-recipes Public
    AI-Hypercomputer/tpu-recipes’s past year of commit activity
    Shell 11 Apache-2.0 7 3 5 Updated Feb 14, 2025
  • maxdiffusion Public
    AI-Hypercomputer/maxdiffusion’s past year of commit activity
    Python 186 Apache-2.0 22 4 (1 issue needs help) 10 Updated Feb 14, 2025
  • jetstream-pytorch Public

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

    AI-Hypercomputer/jetstream-pytorch’s past year of commit activity
    Python 51 Apache-2.0 17 9 5 Updated Feb 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
    Dockerfile 35 Apache-2.0 6 0 0 Updated Feb 7, 2025

People

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