Illustration: the project’s public image, pixelated by RepoMasters. Credits
RCEpic

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.

Owner
facebookincubator (organization)
Stars
Stars: 4,734
Forks
Forks: 387
Language
Python
License
Apache-2.0
Created
2022
Tier
RC (2,000 to 4,999 ★) What this tier means
View on GitHub

More repos by facebookincubator

All 48 repos by facebookincubator

Nearby in Python

All Python repos