Pinned Loading

1. AITemplate-OSS AITemplate-OSS Public

   Forked from facebookincubator/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.

   Python 1

2. FBGEMM FBGEMM Public

   Forked from pytorch/FBGEMM

   FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

   C++

3. pytorch pytorch Public

   Forked from pytorch/pytorch

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   C++

4. triton triton Public

   Forked from triton-lang/triton

   Development repository for the Triton language and compiler

   C++

5. NVIDIA/cutlass NVIDIA/cutlass Public

   CUDA Templates for Linear Algebra Subroutines

   C++ 7k 1.1k

6. FasterTransformer FasterTransformer Public

   Forked from NVIDIA/FasterTransformer

   Transformer related optimization, including BERT, GPT

   C++