NVIDIA Corporation

Repositories

NeMo

NeMo: a toolkit for conversational AI

nlp deep-learning neural-network speech-recognition nlp-machine-learning

Python Apache-2.0 226 1,500 95 (1 issue needs help) 14 Updated Aug 3, 2020

spark-rapids

Spark RAPIDS plugin - accelerate Apache Spark with GPUs

Scala Apache-2.0 32 131 186 (3 issues need help) 7 Updated Aug 3, 2020

DeepLearningExamples

Deep Learning Examples

Python 1,232 4,371 68 12 Updated Aug 3, 2020

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 20 92 68 5 Updated Aug 2, 2020

apex

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

Python BSD-3-Clause 573 4,337 360 17 Updated Aug 3, 2020

triton-inference-server

The Triton Inference Server provides a cloud inferencing solution optimized for NVIDIA GPUs.

inference

C++ BSD-3-Clause 276 1,306 35 6 Updated Aug 2, 2020

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 55 379 11 (1 issue needs help) 0 Updated Aug 2, 2020

gpu-monitoring-tools

Tools for monitoring NVIDIA GPUs on Linux

C Apache-2.0 171 526 12 2 Updated Aug 2, 2020

jitify

A single-header C++ library for simplifying the use of CUDA Runtime Compilation (NVRTC).

C++ BSD-3-Clause 29 253 11 4 Updated Aug 2, 2020

TRTorch

PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT

C++ BSD-3-Clause 22 259 32 (11 issues need help) 3 Updated Aug 1, 2020

deepops

Tools for building GPU clusters

Jsonnet BSD-3-Clause 130 396 53 (1 issue needs help) 14 Updated Aug 1, 2020

aistore

AIStore: scalable storage for AI applications

deep-learning high-performance map-reduce erasure-coding object-storage commodity-hardware scale-out

Go MIT 30 246 0 1 Updated Aug 1, 2020

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 75 322 15 (4 issues need help) 6 Updated Jul 31, 2020

pyxis

Container plugin for Slurm Workload Manager

C Apache-2.0 7 41 2 0 Updated Jul 31, 2020

hpc-container-maker

HPC Container Maker

docker hpc containers singularity

Python Apache-2.0 46 220 3 1 Updated Jul 31, 2020

DALI

A library containing both highly optimized building blocks and an execution engine for data pre-processing in deep learning applications

python machine-learning deep-learning neural-network gpu image-processing gpu-tensorflow

C++ 331 2,709 144 17 Updated Jul 31, 2020

MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

computer-vision deep-learning neural-network cuda high-dimensional-data pytorch convolutional-neural-networks

Python 99 778 53 3 Updated Jul 31, 2020

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 570 2,580 225 (6 issues need help) 21 Updated Jul 31, 2020

framework-determinism

Providing determinism in deep learning frameworks

Python Apache-2.0 14 155 4 (1 issue needs help) 1 Updated Jul 31, 2020

PyProf

A GPU performance profiling tool for PyTorch models

Python Apache-2.0 11 177 4 0 Updated Jul 30, 2020

DALI_extra

Test data for DALI project

Apache-2.0 16 13 0 0 Updated Jul 30, 2020

enroot

A simple yet powerful tool to turn traditional container/OS images into unprivileged sandboxes.

Shell Apache-2.0 19 145 1 0 Updated Jul 30, 2020

data-science-stack

NVIDIA Data Science stack tools

Shell Apache-2.0 15 114 14 1 Updated Jul 30, 2020

ais-tar2tf

TensorFlow-native AIStore dataset (AisDataset) and associated data loaders

Python MIT 0 2 0 0 Updated Jul 29, 2020

nvidia-docker

Build and run Docker containers leveraging NVIDIA GPUs

docker gpu cuda nvidia-docker

Makefile Apache-2.0 1,507 11,482 89 0 Updated Jul 29, 2020

egx-platform

Bring AI to the EDGE

Shell Apache-2.0 5 16 1 0 Updated Jul 29, 2020

OpenSeq2Seq

Toolkit for efficient experimentation with Speech Recognition, Text2Speech and NLP

text-to-speech deep-learning tensorflow multi-node speech-synthesis speech-recognition seq2seq

Python Apache-2.0 313 1,249 74 (2 issues need help) 5 Updated Jul 28, 2020

spark-xgboost-examples

XGBoost GPU accelerated on Spark example applications

Jupyter Notebook Apache-2.0 8 11 3 1 Updated Jul 28, 2020

AMGX

Distributed multigrid linear solver library on GPU

Cuda BSD-3-Clause 65 161 44 (1 issue needs help) 3 Updated Jul 28, 2020

nccl

Optimized primitives for collective multi-GPU communication

C++ 350 1,276 43 13 Updated Jul 27, 2020