Device-agnostic

Eager-Mode

Custom Modules
Optimizers
Quantization

Graph-Mode

Custom Fusion Passes
oneDNN Fusion Passes

ATen Ops

Custom Ops

Graph Ops

Custom Fuses Ops
oneDNN Fused Ops

Kernels

GPU Custom & ATen Kernels
oneDNN GPU Kernels
oneMKL GPU Kernels
CPU Custom & ATen Kernels
oneDNN CPU Kernels
oneMKL CPU Kernels