
Rank #1,698 4,730 starsPython
AITemplate
by facebookincubator (organization)
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. (description from the project’s GitHub page)
Repository facts
- Stars
- 4,730
- Forks
- 387
- Open issues and pull requests
- 163
- Primary language
- Python
- License
- Apache License 2.0 (Apache-2.0)
- Owner
- facebookincubator · Organization
- Created
- Jul 15, 2022
- Last push
- Oct 7, 2026