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.
- ATK
- 5028
- DEF
- 3546
Common
facebookincubator/AITemplate
Common · Base · GenesisAITemplate 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.
Attack5,028
Defense3,546
Drag the card to turn it. Use the flip button to see the back.
Open card page