NVIDIA Corporation

Repositories

• DALI

  A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

  python machine-learning deep-learning neural-network mxnet gpu image-processing
  C++ 419 3,355 110 18 Updated Jul 8, 2021

• TRTorch

  PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT

  machine-learning deep-learning cuda pytorch nvidia jetson tensorrt
  2 packages Jupyter Notebook BSD-3-Clause 70 744 58 (4 issues need help) 20 Updated Jul 7, 2021

• NeMo

  NeMo: a toolkit for conversational AI

  nlp text-to-speech deep-learning neural-network machine-translation speech-synthesis speech-recognition
  Jupyter Notebook Apache-2.0 553 2,875 67 31 Updated Jul 7, 2021

• TensorRT

  TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.

  deep-learning nvidia tensorrt
  C++ Apache-2.0 945 3,900 145 25 Updated Jul 7, 2021

• spark-rapids

  Spark RAPIDS plugin - accelerate Apache Spark with GPUs

  big-data spark gpu rapids
  Scala Apache-2.0 96 318 430 (25 issues need help) 19 Updated Jul 7, 2021

• MagnumIO

  Magnum IO community repo
  C++ Apache-2.0 2 7 0 3 Updated Jul 7, 2021

• libcudacxx

  The C++ Standard Library for your entire system.

  cxx cpp gpu llvm cpp14 cuda cpp11
  C++ 111 1,754 79 21 Updated Jul 7, 2021

• aistore

  AIStore: scalable storage for AI applications

  deep-learning etl high-performance hyper-converged object-storage commodity-hardware petascale
  Go MIT 42 424 1 0 Updated Jul 7, 2021

• thrust

  The C++ parallel algorithms library.

  cxx algorithms cpp gpu cpp14 cuda cpp11
  C++ 591 3,354 290 (2 issues need help) 8 Updated Jul 7, 2021

• cub

  Cooperative primitives for CUDA C++.

  cxx algorithms cpp gpu cpp14 cuda cpp11
  Cuda BSD-3-Clause 302 946 69 (3 issues need help) 16 Updated Jul 7, 2021

• Bobber

  Containerized testing of system components that impact AI workload performance
  Python MIT 1 8 8 5 Updated Jul 7, 2021

• vagrant-swift-all-in-one

  Vagrant Swift All In One
  Ruby Apache-2.0 56 75 4 10 Updated Jul 7, 2021

• NVTabular

  NVTabular is a feature engineering and preprocessing library for tabular data designed to quickly and easily manipulate terabyte scale datasets used to train deep learning based recommender systems.
  Python Apache-2.0 66 426 136 (1 issue needs help) 17 Updated Jul 7, 2021

• cuCollections
  Cuda Apache-2.0 14 81 23 (1 issue needs help) 10 Updated Jul 7, 2021

• deepops

  Tools for building GPU clusters
  Shell BSD-3-Clause 188 604 32 10 Updated Jul 7, 2021

• NVIDIAOpticalFlowSDK

  Optical Flow SDK exposes the latest hardware capability of Turing GPUs dedicated to computing the relative motion of pixels between images. The hardware uses sophisticated algorithms to yield highly accurate flow vectors, with robust frame-to-frame intensity variations and tracks the true object motion faster and more accurately.
  C 4 36 0 1 Updated Jul 7, 2021

• FasterTransformer

  Transformer related optimization, including BERT, GPT

  pytorch transformer gpt bert
  C++ Apache-2.0 51 364 14 2 Updated Jul 7, 2021

• ansible-role-enroot
  BSD-3-Clause 11 6 0 5 Updated Jul 7, 2021

• VideoProcessingFramework

  Set of Python bindings to C++ libraries which provides full HW acceleration for video decoding, encoding and GPU-accelerated color space and pixel format conversions
  C++ Apache-2.0 101 572 26 0 Updated Jul 7, 2021

• CUDALibrarySamples

  CUDA Library Samples
  C++ 49 196 6 1 Updated Jul 7, 2021

• apex

  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  Python BSD-3-Clause 775 5,492 468 26 Updated Jul 7, 2021

• cutlass

  CUDA Templates for Linear Algebra Subroutines

  deep-learning cuda deep-learning-library
  C++ BSD-3-Clause 229 1,221 47 7 Updated Jul 7, 2021

• framework-determinism

  Providing determinism in deep learning frameworks

  deep-learning tensorflow pytorch ngc gpu-support determinism gpu-determinism
  Python Apache-2.0 25 282 8 (1 issue needs help) 2 Updated Jul 7, 2021

• gdrcopy

  A fast GPU memory copy library based on NVIDIA GPUDirect RDMA technology

  linux libraries nvidia gpu-memory kernel-mode-driver gpudirect-rdma
  C++ MIT 90 437 26 (4 issues need help) 3 Updated Jul 7, 2021

• nccl-tests

  NCCL Tests
  Cuda 73 187 10 1 Updated Jul 7, 2021

• open-gpu-doc

  Documentation of NVIDIA chip/hardware interfaces
  C MIT 65 938 0 1 Updated Jul 6, 2021

• retinanet-examples

  Fast and accurate object detection with end-to-end GPU optimization

  python deep-learning neural-network pytorch object-detection tensorrt retinanet
  Python BSD-3-Clause 219 684 10 1 Updated Jul 7, 2021

• swift

  Forked from openstack/swift

  OpenStack Storage (Swift). Mirror of code maintained at opendev.org.
  Python Apache-2.0 1,039 3 0 0 Updated Jul 6, 2021

• Megatron-LM

  Ongoing research training transformer language models at scale, including: BERT & GPT-2
  Python 405 2,199 66 9 Updated Jul 6, 2021

• container-config

  Set up your container runtime with NVIDIA GPUs support
  Go Apache-2.0 3 6 0 0 Updated Jul 6, 2021