Pinned Loading

1. pytorch/pytorch pytorch/pytorch Public

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python 84.5k 22.7k

2. FasterTransformer FasterTransformer Public

   Forked from NVIDIA/FasterTransformer

   Transformer related optimization, including BERT, GPT

   C++

3. pytorch/ao pytorch/ao Public

   PyTorch native quantization and sparsity for training and inference

   Python 1.6k 179

4. apache/tvm apache/tvm Public

   Open deep learning compiler stack for cpu, gpu and specialized accelerators

   Python 11.8k 3.5k

5. flash-attention flash-attention Public

   Forked from Dao-AILab/flash-attention

   Fast and memory-efficient exact attention

   Python

6. facebookincubator/gloo facebookincubator/gloo Public

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

   C++ 1.2k 303