Skip to content
Change the repository type filter

All

    Repositories list

    • Primus-SaFE(Stability and Fault Endurance)
      Go
      Other
      15205Updated Mar 2, 2026Mar 2, 2026
    • Python
      Other
      1162113Updated Mar 2, 2026Mar 2, 2026
    • Primus

      Public
      Python
      Other
      2774617Updated Mar 2, 2026Mar 2, 2026
    • Toolkit for launching and observing MaxText training on Slurm-managed GPU clusters
      Shell
      MIT License
      11800Updated Mar 1, 2026Mar 1, 2026
    • GEAK

      Public
      It is an LLM-based AI agent, which can write correct and efficient gpu kernels automatically.
      Python
      Other
      136924Updated Feb 28, 2026Feb 28, 2026
    • AgentKernelArena provides an end-to-end siloed-benchmarking environment where different LLM-powered agents—such as Cursor Agent, Claude Code, Codex, SWE-agent, …
      Python
      Apache License 2.0
      38121Updated Feb 27, 2026Feb 27, 2026
    • TraceLens

      Public
      Automating analysis from trace files
      Python
      MIT License
      960779Updated Feb 27, 2026Feb 27, 2026
    • Magpie

      Public
      A lightweight, general-purpose framework for evaluating GPU kernel correctness and performance.
      Python
      MIT License
      33001Updated Feb 27, 2026Feb 27, 2026
    • DUET-VLM

      Public
      DUET-VLM: Dual stage Unified Efficient Token reduction for VLM Training and Inference
      Python
      Apache License 2.0
      01221Updated Feb 25, 2026Feb 25, 2026
    • nixl

      Public
      NVIDIA Inference Xfer Library (NIXL)
      C++
      Other
      248100Updated Feb 24, 2026Feb 24, 2026
    • dynamo

      Public
      A Datacenter Scale Distributed Inference Serving Framework
      Rust
      Other
      882000Updated Feb 24, 2026Feb 24, 2026
    • Nitro-E

      Public
      Python
      MIT License
      911021Updated Feb 24, 2026Feb 24, 2026
    • Repository for Showcasing DLRM v2 Functionality on AMD
      Python
      MIT License
      0000Updated Feb 16, 2026Feb 16, 2026
    • Python
      Apache License 2.0
      01400Updated Feb 10, 2026Feb 10, 2026
    • PARD

      Public
      PARD: Accelerating LLM Inference with Low-Cost PARallel Draft Model Adaptation (ICLR 26)
      Python
      MIT License
      11210Updated Feb 6, 2026Feb 6, 2026
    • For world model code developing and releasing.
      Python
      Other
      33510Updated Feb 6, 2026Feb 6, 2026
    • TraceLens-inference

      Public archive
      Automating analysis from trace files
      Python
      MIT License
      9000Updated Feb 5, 2026Feb 5, 2026
    • axlearn

      Public
      An Extensible Deep Learning Library
      Python
      Apache License 2.0
      404100Updated Jan 29, 2026Jan 29, 2026
    • Repo containing artifacts for Neurips 2025 tutorial- How to Build Agents to Generate Kernels for Faster LLMs (and Other Models!)
      Jupyter Notebook
      MIT License
      21400Updated Jan 23, 2026Jan 23, 2026
    • Python
      Other
      0500Updated Jan 22, 2026Jan 22, 2026
    • AMD 0.9B efficient text to video diffusion model
      Python
      Other
      64411Updated Jan 12, 2026Jan 12, 2026
    • This is a short course covering GPU optimization techniques for LLM inference
      Python
      MIT License
      0000Updated Jan 11, 2026Jan 11, 2026
    • Examples of training autodrive models in ROCm
      Python
      Other
      0200Updated Jan 9, 2026Jan 9, 2026
    • GEAK-eval

      Public
      Python
      61190Updated Dec 24, 2025Dec 24, 2025
    • Synthetic data generation pipeline, finetuning and evaluation scripts.
      Python
      Other
      1110Updated Dec 24, 2025Dec 24, 2025
    • Python
      Apache License 2.0
      1110Updated Dec 16, 2025Dec 16, 2025
    • monarch

      Public
      PyTorch Single Controller
      Rust
      BSD 3-Clause "New" or "Revised" License
      148006Updated Dec 2, 2025Dec 2, 2025
    • A PyTorch native platform for training generative AI models
      Python
      BSD 3-Clause "New" or "Revised" License
      7251501Updated Nov 18, 2025Nov 18, 2025
    • Python
      Other
      1700Updated Nov 14, 2025Nov 14, 2025
    • Instella

      Public
      Fully Open Language Models with Stellar Performance
      Python
      Other
      2931720Updated Nov 14, 2025Nov 14, 2025