Pinned Loading

1. pytorch/pytorch pytorch/pytorch Public

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python 105k 30.7k

2. pytorch/ao pytorch/ao Public

   PyTorch native quantization for training and inference

   Python 3k 683

3. apache/tvm apache/tvm Public

   Open Machine Learning Compiler Framework

   Python 13.9k 4k

4. FasterTransformer FasterTransformer Public

   Forked from NVIDIA/FasterTransformer

   Transformer related optimization, including BERT, GPT

   C++

5. flash-attention flash-attention Public

   Forked from Dao-AILab/flash-attention

   Fast and memory-efficient exact attention

   Python

6. pytorch/gloo pytorch/gloo Public

   Collective communications library with various primitives for multi-machine training.

   C++ 1.5k 372