Owner

AI-Hypercomputer

14 indexed repositories · View on GitHub

• maxtext

  A simple, performant, and scalable Jax LLM!

  Python · 2420 stars

• JetStream

  JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel...

  Python · 457 stars

• maxdiffusion

  Python · 386 stars

• RecML

  Python · 226 stars

• xpk

  xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat...

  Python · 193 stars

• gpu-recipes

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

  Python · 140 stars

• tpu-recipes

  Shell · 126 stars

• accelerator-agents

  Python · 78 stars

• ml-goodput-measurement

  Python · 41 stars

• cloud-accelerator-diagnostics

  Python · 32 stars

• accelerator-microbenchmarks

  Python · 24 stars

• ray-tpu

  Python · 15 stars

• resiliency

  This project provides resilient training logic to improve the training efficiency of large-scale distributed workloads on Google Cloud.

  Python · 14 stars

• google-cloud-mldiagnostics

  Python · 14 stars