Skip to content

@AI-Hypercomputer AI Hypercomputer

Change the repository type filter

All

    Repositories list

    22 repositories

    • maxdiffusion

      Public
      Python
      •
      Apache License 2.0
      •63•315•5•15•Updated Mar 2, 2026Mar 2, 2026

    • maxtext

      Public
      A simple, performant and scalable Jax LLM!
      gptmistralfine-tuningjaxsftlarge-language-modelsllmllama2mixtraldeepseek
      Python
      •
      Apache License 2.0
      •477•2.2k•74•175•Updated Mar 2, 2026Mar 2, 2026

    • gpu-recipes

      Public
      Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.
      gpubenchmarksgoogle-cloud-platformservingdistributed-training
      Python
      •
      Apache License 2.0
      •61•118•6•13•Updated Feb 28, 2026Feb 28, 2026

    • accelerator-microbenchmarks

      Public
      Python
      •
      Apache License 2.0
      •31•11•3•8•Updated Feb 28, 2026Feb 28, 2026

    • pathways-utils

      Public
      Package of Pathways-on-Cloud utilities
      Python
      •
      Apache License 2.0
      •11•25•0•8•Updated Feb 27, 2026Feb 27, 2026

    • 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 GP…
      gkegcloudtpu
      Python
      •
      Apache License 2.0
      •79•170•24•7•Updated Feb 27, 2026Feb 27, 2026

    • RecML

      Public
      Python
      •
      Apache License 2.0
      •22•214•3•81•Updated Feb 27, 2026Feb 27, 2026

    • tpu-recipes

      Public
      Shell
      •
      Apache License 2.0
      •49•76•6•13•Updated Feb 23, 2026Feb 23, 2026

    • google-cloud-mldiagnostics

      Public
      Python
      •
      Apache License 2.0
      •2•8•0•0•Updated Feb 19, 2026Feb 19, 2026

    • torchprime

      Public
      torchprime is a reference model implementation for PyTorch on TPU.
      Python
      •
      Other
      •14•46•88•17•Updated Feb 18, 2026Feb 18, 2026

    • kithara

      Public
      Python
      •
      Apache License 2.0
      •9•16•0•2•Updated Feb 18, 2026Feb 18, 2026

    • accelerator-agents

      Public
      Apache License 2.0
      •0•1•0•0•Updated Feb 4, 2026Feb 4, 2026

    • ml-goodput-measurement

      Public
      Python
      •
      Apache License 2.0
      •6•31•1•0•Updated Feb 4, 2026Feb 4, 2026

    • cloud-diagnostics-xprof

      Public
      Python
      •
      Apache License 2.0
      •7•12•0•1•Updated Jan 23, 2026Jan 23, 2026

    • 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).
      gpuinferencepytorchtransformerllamagptgemmamodel-servingtpujax
      Python
      •
      Apache License 2.0
      •58•414•14•12•Updated Jan 5, 2026Jan 5, 2026

    • cloud-accelerator-diagnostics

      Public
      Python
      •
      Apache License 2.0
      •14•27•5•0•Updated Dec 23, 2025Dec 23, 2025

    • jetstream-pytorch

      Public archive
      PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
      inferencepytorchbatchingattentionllamagemmamodel-servingtpullmllm-inference
      Python
      •
      Apache License 2.0
      •21•79•10•3•Updated Dec 18, 2025Dec 18, 2025

    • resiliency

      Public
      This project provides resilient training logic to improve the training efficiency of large-scale distributed workloads on Google Cloud.
      Python
      •
      Apache License 2.0
      •3•12•1•1•Updated Sep 29, 2025Sep 29, 2025

    • inference-benchmark

      Public
      Python
      •
      Apache License 2.0
      •15•18•7•4•Updated Jun 18, 2025Jun 18, 2025

    • ray-tpu

      Public
      Python
      •
      Apache License 2.0
      •8•15•3•1•Updated May 11, 2025May 11, 2025

    • aotc

      Public
      Python
      •
      Other
      •4•0•1•1•Updated May 8, 2025May 8, 2025

    • cloud-tpu-monitoring-debugging

      Public
      HCL
      •
      Apache License 2.0
      •5•14•0•2•Updated Feb 18, 2025Feb 18, 2025