/AITemplate_OSS

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.

Primary LanguagePythonApache License 2.0Apache-2.0

This repository is not active