Skip to content
Change the repository type filter

All

    Repositories list

    • Python
      Apache License 2.0
      7535494Updated Jun 1, 2026Jun 1, 2026
    • maxtext

      Public
      A simple, performant and scalable Jax LLM!
      Python
      Apache License 2.0
      5242.3k58149Updated Jun 1, 2026Jun 1, 2026
    • Python
      Apache License 2.0
      220112Updated May 30, 2026May 30, 2026
    • Package of Pathways-on-Cloud utilities
      Python
      Apache License 2.0
      123008Updated May 29, 2026May 29, 2026
    • Shell
      Apache License 2.0
      62104721Updated May 28, 2026May 28, 2026
    • Python
      Apache License 2.0
      351637Updated May 27, 2026May 27, 2026
    • Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.
      Python
      Apache License 2.0
      77133820Updated May 22, 2026May 22, 2026
    • RecML

      Public
      Python
      Apache License 2.0
      22224385Updated May 19, 2026May 19, 2026
    • Python
      Apache License 2.0
      63610Updated May 15, 2026May 15, 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
      88183252Updated May 14, 2026May 14, 2026
    • Python
      Apache License 2.0
      143160Updated May 5, 2026May 5, 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
      654421413Updated 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.