facebookincubator/AITemplate
View on GitHub ↗ ★ 4,731 Python
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.
11 mentions on Hacker News by 7 people: 3 stories, 8 comments · first 2022-10-03 · latest 2023-10-06
20082026
-
▲ 177
Render a neural network into CUDA/HIP codestory by fzliu · 2023-06-02 · 65 comments
-
▲ 3
PyTorch models to optimised C++ with 2x speedupstory by nl · 2022-10-04 · 0 comments
-
▲ 1
AITemplate for Stable Diffusionstory by hochmartinez · 2022-10-04 · 0 comments