NVIDIA Corporation

Repositories

DeepLearningExamples

Deep Learning Examples

Python 1,227 4,353 68 11 Updated Aug 1, 2020

NeMo

NeMo: a toolkit for conversational AI

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

Python Apache-2.0 224 1,497 95 (1 issue needs help) 13 Updated Aug 1, 2020

apex

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

Python BSD-3-Clause 572 4,331 360 15 Updated Aug 1, 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 91 68 5 Updated Aug 1, 2020

jitify

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

C++ BSD-3-Clause 29 252 11 6 Updated Aug 1, 2020

TRTorch

PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT

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

deepops

Tools for building GPU clusters

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

spark-rapids

Spark RAPIDS plugin - accelerate Apache Spark with GPUs

Scala Apache-2.0 31 131 186 (3 issues need help) 7 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 245 0 1 Updated Aug 1, 2020

triton-inference-server

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

inference

C++ BSD-3-Clause 275 1,305 36 7 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 219 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,706 143 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 776 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 569 2,576 225 (6 issues need help) 21 Updated Jul 31, 2020

framework-determinism

Providing determinism in deep learning frameworks

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

PyProf

A GPU performance profiling tool for PyTorch models

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

DALI_extra

Test data for DALI project

Apache-2.0 16 12 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,506 11,461 88 0 Updated Jul 29, 2020

egx-platform

Bring AI to the EDGE

Shell Apache-2.0 5 16 0 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,246 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 160 44 (1 issue needs help) 3 Updated Jul 28, 2020

nccl

Optimized primitives for collective multi-GPU communication

C++ 349 1,273 40 13 Updated Jul 28, 2020

NVTX

The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resources in your applications.

C Apache-2.0 2 17 1 2 Updated Jul 27, 2020

rtx_compute_samples

RTX compute samples

C++ BSD-3-Clause 1 29 0 0 Updated Jul 27, 2020