Skip to content

@AMD-AGI AMD-AGI

Change the repository type filter

All

    Repositories list

    61 repositories

    • GEAK

      Public
      It is an LLM-based AI agent, which can write correct and efficient gpu kernels automatically.
      Python
      •
      MIT License
      •13•81•8•19•Updated Mar 28, 2026Mar 28, 2026

    • Primus

      Public
      Python
      •
      Other
      •28•82•6•25•Updated Mar 28, 2026Mar 28, 2026

    • Primus-DLRM

      Public
      DLRM implementation for Primus
      Python
      •
      MIT License
      •0•0•0•0•Updated Mar 28, 2026Mar 28, 2026

    • Primus-SaFE

      Public
      Primus-SaFE(Stability and Fault Endurance)
      Go
      •
      Other
      •1•54•0•9•Updated Mar 28, 2026Mar 28, 2026

    • TraceLens

      Public
      Automating analysis from trace files
      Python
      •
      MIT License
      •9•66•94•13•Updated Mar 28, 2026Mar 28, 2026

    • Magpie

      Public
      A lightweight, general-purpose framework for evaluating GPU kernel correctness and performance.
      Python
      •
      MIT License
      •5•49•1•1•Updated Mar 27, 2026Mar 27, 2026

    • AgentKernelArena

      Public
      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
      •3•13•12•1•Updated Mar 27, 2026Mar 27, 2026

    • Primus-Turbo

      Public
      Python
      •
      Other
      •13•64•10•5•Updated Mar 27, 2026Mar 27, 2026

    • Apex

      Public
      Agents, and RL environment, for optimizing GPU kernels on AMD ROCm using LLM agents. Benchmarks LLM serving workloads end-to-end, profiles bottleneck kernels, o…
      Python
      •
      MIT License
      •7•50•0•1•Updated Mar 26, 2026Mar 26, 2026

    • FLy

      Public
      Python
      •
      MIT License
      •0•0•1•0•Updated Mar 25, 2026Mar 25, 2026

    • maxtext-slurm

      Public
      Toolkit for launching and observing MaxText training on Slurm-managed GPU clusters
      Shell
      •
      MIT License
      •2•27•0•1•Updated Mar 22, 2026Mar 22, 2026

    • mlcommons-inference

      Public
      Reference implementations of MLPerf® inference benchmarks
      Python
      •
      Apache License 2.0
      •616•0•0•0•Updated Mar 16, 2026Mar 16, 2026

    • PARD

      Public
      PARD: Accelerating LLM Inference with Low-Cost PARallel Draft Model Adaptation (ICLR 26)
      Python
      •
      MIT License
      •1•19•1•0•Updated Mar 13, 2026Mar 13, 2026

    • DUET-VLM

      Public
      DUET-VLM: Dual stage Unified Efficient Token reduction for VLM Training and Inference
      Python
      •
      Apache License 2.0
      •1•21•1•1•Updated Mar 5, 2026Mar 5, 2026

    • nixl

      Public
      NVIDIA Inference Xfer Library (NIXL)
      C++
      •
      Other
      •278•1•0•0•Updated Feb 24, 2026Feb 24, 2026

    • dynamo

      Public
      A Datacenter Scale Distributed Inference Serving Framework
      Rust
      •
      Other
      •963•0•0•0•Updated Feb 24, 2026Feb 24, 2026

    • Nitro-E

      Public
      Python
      •
      MIT License
      •10•113•2•1•Updated Feb 24, 2026Feb 24, 2026

    • DLRMBenchmark

      Public
      Repository for Showcasing DLRM v2 Functionality on AMD
      amdrecommender-systemdlrm
      amdrecommender-systemdlrm
      Python
      •
      MIT License
      •0•0•0•0•Updated Feb 16, 2026Feb 16, 2026

    • HummingbirdXT

      Public
      Python
      •
      Apache License 2.0
      •0•14•0•0•Updated Feb 10, 2026Feb 10, 2026

    • Micro-World

      Public
      For world model code developing and releasing.
      Python
      •
      Other
      •4•45•0•0•Updated Feb 6, 2026Feb 6, 2026

    • TraceLens-inference

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

    • axlearn

      Public
      An Extensible Deep Learning Library
      Python
      •
      Apache License 2.0
      •402•1•0•0•Updated Jan 29, 2026Jan 29, 2026

    • Neurips2025-GPU-kernels-Tutorial

      Public
      Repo containing artifacts for Neurips 2025 tutorial- How to Build Agents to Generate Kernels for Faster LLMs (and Other Models!)
      Jupyter Notebook
      •
      MIT License
      •2•14•0•0•Updated Jan 23, 2026Jan 23, 2026

    • ReasonLite

      Public
      Python
      •
      Other
      •0•7•1•0•Updated Jan 22, 2026Jan 22, 2026

    • Hummingbird

      Public
      AMD 0.9B efficient text to video diffusion model
      Python
      •
      Other
      •6•44•1•1•Updated Jan 12, 2026Jan 12, 2026

    • GPU-Optimization-for-LLM-Inference

      Public
      This is a short course covering GPU optimization techniques for LLM inference
      Python
      •
      MIT License
      •0•0•0•0•Updated Jan 11, 2026Jan 11, 2026

    • awesome-rocm-autodrive

      Public
      Examples of training autodrive models in ROCm
      Python
      •
      Other
      •0•3•0•0•Updated Jan 9, 2026Jan 9, 2026

    • GEAK-eval

      Public
      Python
      •6•11•9•0•Updated Dec 24, 2025Dec 24, 2025

    • sand-pipeline

      Public
      Synthetic data generation pipeline, finetuning and evaluation scripts.
      Python
      •
      Other
      •1•1•1•0•Updated Dec 24, 2025Dec 24, 2025

    • Athena-PRM

      Public
      Python
      •
      Apache License 2.0
      •1•1•1•0•Updated Dec 16, 2025Dec 16, 2025
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.