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.
AI summary
Neural network optimizer
A framework that transforms deep neural networks into high-performance GPU-optimized C++ code for efficient inference serving.
- stars
- 4.6K
- forks
- 372
- watching
- 82
Similar projects
Found by comparing what the projects do, not just their names.
Neural network framework
A C++/CUDA framework for training and querying neural networks using GPUs
nvlabs/instant-ngp16.1K
Neural Graphics Toolkit
A software toolkit for training and rendering neural graphics primitives
Neural transformer
This is an implementation of neural style transfer in TensorFlow using the Adam optimizer.
sony/nnabla2.7K
Deep learning framework
A deep learning framework that provides a flexible and expressive Python API for building and training neural networks on various platforms.
GPU tensor library
A Python library providing tensors and dynamic neural networks with strong GPU acceleration
plasma-umass/scalene12.3K
Profiler
A high-performance Python profiler that analyzes CPU, GPU, and memory usage, providing detailed information and AI-powered optimization suggestions.
Attention algorithms
Implementations of efficient exact attention mechanisms for machine learning
Tensor processing library
An auto-differentiation library for sparse tensors used in computer vision and deep learning applications.
Transformer platform
A codebase for working with Open Pre-trained Transformers, enabling deployment and fine-tuning of transformer models on various platforms.
3D data processor
A deep learning library for 3D data processing and computer vision research using PyTorch
tflearn/tflearn9.6K
Deep Learning Library
A high-level API for deep learning that builds upon TensorFlow
Image Restoration Engine
A project demonstrating image restoration using neural networks without learning
luolc/adabound2.9K
Optimizer
An optimizer that combines the benefits of Adam and SGD algorithms
Neural reconstruction tool
An implementation of high-fidelity neural surface reconstruction from video frames using deep learning
Transformer component
A high-performance transformer-based NLP component optimized for GPU acceleration and integration into various frameworks.
