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 352

  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 333 42

  3. maxdiffusion maxdiffusion Public

    Python 213 30

  4. RecML RecML Public

    Python 179 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 121 36

  6. gpu-recipes gpu-recipes Public

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

    Python 65 20

Repositories

Showing 10 of 19 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…