AITemplate Homepage, Documentation and Downloads – Meta Open Source Python Framework – News Fast Delivery
AITemplate (AIT) is a Python framework that translates deep neural networks into CUDA (NVIDIA GPU) / HIP (AMD GPU) C++ code for fast inference services. Highlights of AITemplate include: High performance: Approaching roofline fp16 TensorCore (NVIDIA GPU)/MatrixCore (AMD GPU) performance on major models, including ResNet, MaskRCNN, BERT, VisionTransformer, Stable Diffusion, etc. Unified, open, flexible: Seamless […]