Pinned Loading

1. cuda-gpu-kernels cuda-gpu-kernels Public

   Custom GPU kernels written in CUDA

   Cuda 3

2. maskrcnn-benchmark maskrcnn-benchmark Public

   Forked from facebookresearch/maskrcnn-benchmark

   Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

   Python 1

3. apex apex Public

   Forked from NVIDIA/apex

   A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

   Python

4. ernie-2 ernie-2 Public

   CNN training for Ernie architecture using TensorFlow

   Python 1

5. pytorch/pytorch pytorch/pytorch Public

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python 86.7k 23.3k

6. NVIDIA/apex NVIDIA/apex Public

   A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

   Python 8.5k 1.4k