NVIDIA Corporation

Repositories

• spark-rapids

  Spark RAPIDS plugin - accelerate Apache Spark with GPUs

  big-data spark gpu rapids
  Scala Apache-2.0 82 292 394 (27 issues need help) 18 Updated May 7, 2021

• cutlass

  CUDA Templates for Linear Algebra Subroutines

  deep-learning cuda deep-learning-library
  C++ BSD-3-Clause 212 1,157 38 7 Updated May 7, 2021

• 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++ 401 3,264 107 24 Updated May 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 96 541 29 (1 issue needs help) 2 Updated May 7, 2021

• gpu-monitoring-tools

  Tools for monitoring NVIDIA GPUs on Linux
  C Apache-2.0 242 757 64 5 Updated May 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 55 363 110 14 Updated May 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 475 2,663 67 21 Updated May 7, 2021

• aistore

  AIStore: scalable storage for AI applications

  deep-learning etl high-performance hyper-converged object-storage commodity-hardware petascale
  Go MIT 39 390 1 0 Updated May 7, 2021

• thrust

  The C++ parallel algorithms library.

  cxx algorithms cpp gpu cpp14 cuda cpp11
  C++ 578 3,288 277 (2 issues need help) 8 Updated May 6, 2021

• cub

  Cooperative primitives for CUDA C++.

  cxx algorithms cpp gpu cpp14 cuda cpp11
  Cuda BSD-3-Clause 289 916 63 (6 issues need help) 13 Updated May 6, 2021

• proxyfs
  Go Apache-2.0 23 52 14 12 Updated May 6, 2021

• framework-determinism

  Providing determinism in deep learning frameworks

  deep-learning tensorflow pytorch ngc gpu-support determinism gpu-determinism
  Python Apache-2.0 24 260 8 (1 issue needs help) 2 Updated May 6, 2021

• deepops

  Tools for building GPU clusters
  Shell BSD-3-Clause 182 577 46 10 Updated May 6, 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 58 667 32 (5 issues need help) 14 Updated May 7, 2021

• fsi-samples

  A collection of open-source GPU accelerated Python tools and examples for quantitative analyst tasks and leverages RAPIDS AI project, Numba, cuDF, and Dask.
  Jupyter Notebook Apache-2.0 67 174 5 0 Updated May 6, 2021

• gpu-operator

  NVIDIA GPU Operator creates/configures/manages GPUs atop Kubernetes
  Go Apache-2.0 69 331 43 0 Updated May 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 210 661 6 1 Updated May 6, 2021

• DALI_deps

  3rd party dependencies for DALI project
  Shell Apache-2.0 4 3 0 1 Updated May 6, 2021

• nsight-training

  Training material for Nsight developer tools
  C 2 10 0 0 Updated May 6, 2021

• FasterTransformer

  Transformer related optimization, including BERT, GPT

  pytorch transformer gpt bert
  C++ Apache-2.0 27 243 3 0 Updated May 6, 2021

• cuCollections
  Cuda Apache-2.0 11 67 21 (1 issue needs help) 9 Updated May 5, 2021

• 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 167 1,182 35 3 Updated May 6, 2021

• libnvidia-container

  NVIDIA container runtime library
  C 104 318 30 0 Updated May 5, 2021

• DeepLearningExamples

  Deep Learning Examples
  Jupyter Notebook 1,725 5,844 90 22 Updated May 5, 2021

• spark-xgboost

  Forked from dmlc/xgboost

  Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Flink and DataFlow
  C++ Apache-2.0 7,983 9 0 1 Updated May 5, 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 863 3,658 179 21 Updated May 4, 2021

• Bobber

  Containerized testing of system components that impact AI workload performance
  Python MIT 1 8 9 9 Updated May 4, 2021

• k8s-device-plugin

  NVIDIA device plugin for Kubernetes

  kubernetes
  Go Apache-2.0 328 1,053 43 6 Updated May 5, 2021

• nvcomp

  A library for fast lossless compression/decompression on the GPU
  C++ BSD-3-Clause 20 136 11 1 Updated May 4, 2021

• cuda-samples

  Samples for CUDA Developers which demonstrates features in CUDA Toolkit

  cuda cuda-kernels cuda-driver-api cuda-opengl
  C 426 1,163 17 9 Updated May 4, 2021