Skip to content
Change the repository type filter

All

    Repositories list

    • maxdiffusion

      Public
      Python
      Apache License 2.0
      7034588Updated May 5, 2026May 5, 2026
    • maxtext

      Public
      A simple, performant and scalable Jax LLM!
      Python
      Apache License 2.0
      5102.3k58139Updated May 5, 2026May 5, 2026
    • accelerator-agents

      Public
      Python
      Apache License 2.0
      216112Updated May 4, 2026May 4, 2026
    • tpu-recipes

      Public
      Shell
      Apache License 2.0
      5995718Updated May 4, 2026May 4, 2026
    • Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.
      Python
      Apache License 2.0
      77131825Updated May 2, 2026May 2, 2026
    • pathways-utils

      Public
      Package of Pathways-on-Cloud utilities
      Python
      Apache License 2.0
      122705Updated Apr 30, 2026Apr 30, 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…
      Python
      Apache License 2.0
      88180253Updated Apr 30, 2026Apr 30, 2026
    • accelerator-microbenchmarks

      Public
      Python
      Apache License 2.0
      341637Updated Apr 29, 2026Apr 29, 2026
    • Python
      Apache License 2.0
      63510Updated Apr 27, 2026Apr 27, 2026
    • cloud-accelerator-diagnostics

      Public
      Python
      Apache License 2.0
      143060Updated Apr 21, 2026Apr 21, 2026
    • RecML

      Public
      Python
      Apache License 2.0
      22223384Updated Apr 14, 2026Apr 14, 2026
    • HCL
      Apache License 2.0
      51501Updated Apr 8, 2026Apr 8, 2026
    • Python
      Apache License 2.0
      21000Updated Apr 7, 2026Apr 7, 2026
    • inference-benchmark

      Public archive
      Python
      Apache License 2.0
      152074Updated Mar 11, 2026Mar 11, 2026
    • torchprime

      Public archive
      torchprime is a reference model implementation for PyTorch on TPU.
      Python
      Other
      14478817Updated Mar 3, 2026Mar 3, 2026
    • kithara

      Public archive
      Python
      Apache License 2.0
      101802Updated Feb 18, 2026Feb 18, 2026
    • Python
      Apache License 2.0
      81301Updated 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).
      Python
      Apache License 2.0
      634321413Updated Jan 5, 2026Jan 5, 2026
    • jetstream-pytorch

      Public archive
      PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
      Python
      Apache License 2.0
      2182103Updated Dec 18, 2025Dec 18, 2025
    • This project provides resilient training logic to improve the training efficiency of large-scale distributed workloads on Google Cloud.
      Python
      Apache License 2.0
      31411Updated Sep 29, 2025Sep 29, 2025
    • ray-tpu

      Public
      Python
      Apache License 2.0
      81531Updated May 11, 2025May 11, 2025
    • aotc

      Public archive
      Python
      Other
      5111Updated May 8, 2025May 8, 2025
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.