NCCL

PyTorch - Tensors and Dynamic neural networks in Python with strong GPU accelerationpytorch/pytorch

A Python package for tensor computation with GPU acceleration and dynamic neural networks built on a tape-based autograd system.

TensorRTNVIDIA/TensorRT

NVIDIA's SDK for high-performance deep learning inference optimization and deployment on NVIDIA GPUs.

flashlightflashlight/flashlight

A fast, flexible C++ standalone library for machine learning with high-performance defaults and total internal modifiability.

DetectAndTrackfacebookresearch/DetectAndTrack

A CVPR 2018 algorithm for efficient multi-person pose estimation and tracking in videos, ranking first in the ICCV 2017 PoseTrack challenge.

JAX ToolboxNVIDIA/JAX-Toolbox

A collection of CI pipelines, Docker images, and optimized examples to simplify JAX development on NVIDIA GPUs.