Differentiable Rendering Toolkit

Development repository for the Triton language and compiler

Fast and memory-efficient exact attention

[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x compared to FlashAttention, without losing end-to-end metrics across language, image, and video models.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Lemonade helps users discover and run local AI apps by serving optimized LLMs right from their own GPUs and NPUs. Join our discord: https://discord.gg/5xXzkMu8Zk

Port of OpenAI's Whisper model in C/C++

A GPU port of DOOM

AMD HIP port of Distributed MoE in a Single Kernel [NeurIPS '25]

Claude code workspace for developing ROCm

Ahead of Time (AOT) Triton Math Library

TurboDiffusion: 100–200× Acceleration for Video Diffusion Models

[ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

Port of Facebook's LLaMA model in C/C++

The best ChatGPT that $100 can buy.

Easily create a beautiful website using Academic and Hugo

Collective communications library with various primitives for multi-machine training.

Making Flux go brrr on GPUs.

Data manipulation and transformation for audio signal processing, powered by PyTorch

The HIP Environment and ROCm Kit - A lightweight open source build system for HIP and ROCm

🔥 A minimal training framework for scaling FLA models

🚀 Efficient implementations of state-of-the-art linear attention models in Torch and Triton

Phantom: Subject-Consistent Video Generation via Cross-Modal Alignment