facebookincubator/AITemplate

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

语言：Python · GitHub 星标：4,731 · Fork：386

在 GitHub 上查看原仓库