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.

github.com/inisis/AITemplate
by @inisis

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.