Skip to content

@AI-Hypercomputer AI Hypercomputer

Change the repository type filter

All

    Repositories list

    21 repositories

    • maxdiffusion

      Public
      Python
      •
      Apache License 2.0
      •50•289•5•21•Updated Dec 18, 2025Dec 18, 2025

    • maxtext

      Public
      A simple, performant and scalable Jax LLM!
      gptmistralfine-tuningjaxsftlarge-language-modelsllmllama2mixtraldeepseek
      Python
      •
      Apache License 2.0
      •441•2k•72•215•Updated Dec 18, 2025Dec 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.
      gkegcloudtpu
      Python
      •
      Apache License 2.0
      •66•154•22•5•Updated Dec 18, 2025Dec 18, 2025

    • tpu-recipes

      Public
      Shell
      •
      Apache License 2.0
      •38•64•4•13•Updated Dec 18, 2025Dec 18, 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

    • 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
      •55•108•6•11•Updated Dec 18, 2025Dec 18, 2025

    • pathways-utils

      Public
      Package of Pathways-on-Cloud utilities
      Python
      •
      Apache License 2.0
      •9•21•0•1•Updated Dec 17, 2025Dec 17, 2025

    • accelerator-microbenchmarks

      Public
      Python
      •
      Apache License 2.0
      •19•7•3•4•Updated Dec 17, 2025Dec 17, 2025

    • RecML

      Public
      Python
      •
      Apache License 2.0
      •20•213•3•75•Updated Dec 16, 2025Dec 16, 2025

    • cloud-diagnostics-xprof

      Public
      Python
      •
      Apache License 2.0
      •4•12•0•0•Updated Dec 12, 2025Dec 12, 2025

    • cloud-accelerator-diagnostics

      Public
      Python
      •
      Apache License 2.0
      •14•25•5•0•Updated Dec 11, 2025Dec 11, 2025

    • google-cloud-mldiagnostics

      Public
      Python
      •
      Apache License 2.0
      •2•6•0•0•Updated Dec 8, 2025Dec 8, 2025

    • torchprime

      Public
      torchprime is a reference model implementation for PyTorch on TPU.
      Python
      •
      Other
      •12•42•89•14•Updated Oct 15, 2025Oct 15, 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
      •2•11•1•0•Updated Sep 29, 2025Sep 29, 2025

    • ml-goodput-measurement

      Public
      Python
      •
      Apache License 2.0
      •6•27•1•0•Updated Sep 11, 2025Sep 11, 2025

    • inference-benchmark

      Public
      Python
      •
      Apache License 2.0
      •15•17•7•4•Updated Jun 18, 2025Jun 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).
      gpuinferencepytorchtransformerllamagptgemmamodel-servingtpujax
      Python
      •
      Apache License 2.0
      •59•395•14•12•Updated Jun 10, 2025Jun 10, 2025

    • kithara

      Public
      Python
      •
      Apache License 2.0
      •8•16•0•2•Updated May 19, 2025May 19, 2025

    • ray-tpu

      Public
      Python
      •
      Apache License 2.0
      •7•15•3•0•Updated May 11, 2025May 11, 2025

    • aotc

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

    • cloud-tpu-monitoring-debugging

      Public
      HCL
      •
      Apache License 2.0
      •3•13•0•0•Updated Feb 18, 2025Feb 18, 2025