Explore projects

an easy-to-use knn-mt toolkit

Optimized primitives for collective multi-GPU communication

Detectron2 is FAIR's next-generation platform for object detection, segmentation and other visual recognition tasks.

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

Sample codes for CUDA programming book

physics meets neural networks

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

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

Convolutional Neural Networks