Shirong wushirong

Pinned Loading

1. Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python 97.9k 27.1k

2. 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.

   Python 4.7k 382