Skip to content

LonghornSilicon/awesome-ai-accelerators

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 

Repository files navigation

Awesome AI Accelerators Awesome

A curated list of papers, tools, and resources for AI accelerators — spanning the entire stack from silicon to serving systems.

PRs Welcome License: CC0-1.0

AI accelerators are the engines behind modern machine learning — the GPUs, TPUs, NPUs, and custom ASICs that make training and inference practical at scale. Building and programming them is a deeply interdisciplinary craft that spans deep learning, computer architecture, compilers, GPU programming, RTL design, and physical implementation.

This list is a curated, opinionated map of that landscape. It favors canonical papers and widely adopted open-source projects over exhaustive coverage. Whether you are an undergraduate taking your first architecture course, a PhD student researching dataflow architectures, or an engineer optimizing kernels at a frontier lab, the goal is to give you the best resource for each topic — and a clear path through the rest.


Contents


AI Foundations

Understanding the workloads is the prerequisite for accelerating them. This section covers the models that drive accelerator design.

AI Fundamentals

Deep Learning

Transformers & LLMs

Diffusion Models

Vision Models

Reinforcement Learning


AI Inference

Inference — not training — dominates the lifetime compute of a deployed model. This section covers the algorithms and systems that make LLM serving fast and cheap.

Quantization

  • TurboQuant — Online vector quantization with near-optimal distortion, delivering large KV cache memory savings and attention speedups.
  • LLM.int8() — 8-bit matrix multiplication for transformers with outlier handling.
  • GPTQ — Accurate post-training quantization to 3–4 bits.
  • AWQ: Activation-aware Weight Quantization — Protecting salient weights for low-bit LLM inference.
  • SmoothQuant — Migrating quantization difficulty from activations to weights.
  • QLoRA — 4-bit NormalFloat quantization enabling fine-tuning on a single GPU.
  • FP8 Formats for Deep Learning — The NVIDIA/Arm/Intel proposal behind Hopper's FP8 support.
  • A Survey of Quantization Methods — Comprehensive survey of efficient neural network inference.

KV Cache

Speculative Decoding

Prefix Caching

Continuous Batching

Mixture of Experts

Sparsity & Pruning

Scheduling

  • FastServe — Preemptive scheduling to minimize job completion time.
  • Llumnix — Dynamic request rescheduling across serving instances.
  • Splitwise — Phase-splitting prefill and decode across heterogeneous hardware.

Memory Optimization

Serving Systems


GPU Programming

The lingua franca of accelerator programming. This section covers the major parallel programming models.

CUDA

Triton

ROCm

Metal


AI Kernels

The hand-tuned building blocks where most accelerator FLOPs are spent.

GEMM

FlashAttention

  • FlashAttention — IO-aware exact attention that avoids materializing the score matrix.
  • FlashAttention-2 — Better parallelism and work partitioning for near-peak throughput.
  • FlashAttention-3 — Exploiting Hopper asynchrony and FP8 for the H100.
  • flash-attention — The reference implementation from Tri Dao's lab.
  • FlashInfer — A kernel library for LLM serving with attention variants.

PagedAttention

  • PagedAttention (vLLM) — Non-contiguous KV cache blocks to eliminate memory fragmentation.
  • vLLM Kernels — Production PagedAttention and quantized kernel implementations.

Normalization & Softmax

RoPE & Positional Encodings

  • RoFormer: Rotary Position Embedding — RoPE, the positional encoding used across modern LLMs.
  • YaRN — Efficient context-window extension for RoPE models.
  • ALiBi — Attention with linear biases for length extrapolation.

Fused Operators

Kernel Optimization Guides


Runtime Systems

Frameworks that take a trained model and run it efficiently on real hardware.

  • vLLM — High-throughput LLM serving engine built around PagedAttention.
  • TensorRT-LLM — NVIDIA's optimized LLM inference library with in-flight batching.
  • SGLang — Fast serving with RadixAttention and a co-designed frontend language.
  • MLC-LLM — Universal LLM deployment across GPUs, phones, and browsers via TVM.
  • llama.cpp — Portable C/C++ inference with aggressive quantization (GGUF).
  • Ollama — Simple local model running built on llama.cpp.
  • TensorRT — NVIDIA's high-performance deep learning inference SDK.
  • ONNX Runtime — Cross-platform inference and training accelerator.
  • Apache TVM — End-to-end deep learning compiler stack (see Compilers).
  • tinygrad — A minimalist framework with a lazy, fused execution model.
  • ExecuTorch — PyTorch's on-device inference runtime for mobile and edge.
  • OpenXLA / XLA — The domain-specific compiler behind JAX and TensorFlow.
  • DeepSpeed — Microsoft's training and inference optimization library.
  • LMDeploy — Toolkit for compressing and serving LLMs.
  • Text Generation Inference (TGI) — Hugging Face's production serving toolkit.

Compilers

The bridge between ML frameworks and silicon.


Computer Architecture

The foundation every accelerator is built on.

Foundations

Memory Systems & Cache

Interconnect & NoC

Dataflow & Systolic Arrays

SIMD, SIMT & Tensor Cores

Processing-in-Memory


AI Accelerator Architecture

The chips themselves, organized by class.

GPUs

TPUs & Systolic ASICs

NPUs & Edge AI

FPGA & Reconfigurable

RISC-V for AI


Simulators & Modeling

Tools for exploring the design space before committing to silicon.

  • gem5 — The de facto community architecture simulator.
  • Accel-Sim — A framework for simulating and validating GPU architectures.
  • GPGPU-Sim — Cycle-level simulation of NVIDIA GPUs running CUDA/OpenCL.
  • SCALE-Sim — ARM's systolic-array CNN accelerator simulator.
  • Timeloop / Accelergy — Mapping and energy/area modeling for DNN accelerators.
  • Sparseloop — Analytical modeling for sparse tensor accelerators.
  • ASTRA-sim — Distributed ML training platform and network co-design simulator.
  • LLMCompass — Hardware evaluation for LLM inference.
  • Roofline: An Insightful Visual Performance Model — Williams, Waterman, and Patterson's foundational model.

Chip Design

Following the people and publications that explain how chips are actually designed, built, and brought to market — the industry context that connects everything above.

Blogs & Newsletters

  • SemiAnalysis — Dylan Patel's deep dives on the semiconductor industry, AI hardware, and the supply chain.
  • The EDA Primer: From RTL to Silicon — SemiAnalysis's end-to-end walkthrough of the chip design flow; the single best starting point for the big picture.

YouTube Channels

  • Asianometry — Accessible, well-researched videos on semiconductor history, manufacturing, and the industry.
  • CalebWritesCode — Hands-on GPU programming and low-level systems tutorials.
  • Dwarkesh Patel — Long-form interviews with leading figures in AI and hardware.
  • 3Blue1Brown — Grant Sanderson's visual explanations of the math and neural networks behind the workloads.

Interactive & Visual

  • brrrviz — An interactive visualization of GPU architecture and how modern GPUs go brrr.

RTL Design

Register-transfer-level design in Verilog and SystemVerilog — the entry point to any accelerator implementation. Start with the textbook, follow the lectures, then build fluency on the practice sites.


Design Verification

Confirming the RTL does what it should — typically the majority of the effort on a real chip. Learn the methodology from the textbook, then practice UVM on the training sites.

  • TextbookSystemVerilog for Verification, Spear & Tumbush. The definitive introduction to the SystemVerilog features and testbench techniques used in industry.
  • Lectures & ResourcesVerification Academy. Siemens EDA's free hub of courses, the canonical UVM Cookbook, and methodology guides — the go-to reference for DV.
  • TrainingChipVerify. Free, approachable SystemVerilog and UVM tutorials with worked examples, ideal for getting members productive quickly.
  • Hands-oncocotb. Write testbenches in Python instead of SystemVerilog; the most accessible on-ramp to writing real verification.

Physical Design

Turning a synthesized netlist into a manufacturable, timing-clean layout (synthesis → floorplan → placement → CTS → routing → timing/signoff). Ground yourself in the textbooks, then run a full flow hands-on.


University Courses

The best openly available courses across the stack.

About

A curated list of AI accelerator papers, resources, tools, and open-source projects.

Resources

Stars

133 stars

Watchers

1 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors