NVIDIA Corporation

Repositories

• NeMo

  NeMo: a toolkit for conversational AI

  nlp deep-learning neural-network speech-recognition nlp-machine-learning
  Python Apache-2.0 291 1,895 125 (1 issue needs help) 10 Updated Oct 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 28 155 68 (1 issue needs help) 6 Updated Oct 4, 2020

• PyProf

  A GPU performance profiling tool for PyTorch models
  Python Apache-2.0 13 214 4 8 Updated Oct 1, 2020

• hpc-container-maker

  HPC Container Maker

  docker hpc containers singularity
  Python Apache-2.0 50 231 5 4 Updated Oct 2, 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 113 877 59 3 Updated Oct 1, 2020

• apex

  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  Python BSD-3-Clause 610 4,568 387 18 Updated Sep 30, 2020

• DeepLearningExamples

  Deep Learning Examples
  Python 1,377 4,734 96 14 Updated Sep 30, 2020

• yum-packaging-nvidia-plugin

  NVIDIA yum and dnf plugins for RHEL driver packages

  nvidia-driver-packages precompiled-kernel-modules yum-packaging
  Python Apache-2.0 1 1 0 0 Updated Sep 29, 2020

• runx

  Deep Learning Experiment Management
  Python BSD-3-Clause 22 458 1 0 Updated Sep 29, 2020

• semantic-segmentation

  Nvidia Semantic Segmentation monorepo
  Python BSD-3-Clause 132 665 10 0 Updated Sep 29, 2020

• gbm-bench

  A benchmark to measure performance of popular Gradient boosting algorithms against popular ML datasets.
  Python 11 22 4 0 Updated Sep 29, 2020

• yum-packaging-precompiled-kmod

  NVIDIA precompiled kernel module packaging for RHEL

  nvidia-driver-packages precompiled-kernel-modules yum-packaging
  Python Apache-2.0 2 8 0 0 Updated Sep 27, 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 320 1,286 79 (2 issues need help) 6 Updated Sep 25, 2020

• jetson-gpio

  A Python library that enables the use of Jetson's GPIOs
  Python 97 281 2 0 Updated Sep 23, 2020

• nvtx-plugins

  Python bindings for NVTX
  Python Apache-2.0 9 45 6 4 Updated Sep 23, 2020

• framework-determinism

  Providing determinism in deep learning frameworks
  Python Apache-2.0 18 178 5 (1 issue needs help) 1 Updated Sep 22, 2020

• Megatron-LM

  Ongoing research training transformer language models at scale, including: BERT & GPT-2
  Python 263 1,515 27 5 Updated Sep 11, 2020

• ais-etl

  Provides for deploying custom ETL containers on AIStore, with subsequent user-defined extraction-transformation-loading in parallel, on the fly and/or offline, locally to user data.
  Python MIT 0 3 0 0 Updated Sep 9, 2020

• 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 172 533 4 3 Updated Sep 3, 2020

• waveglow

  A Flow-based Generative Network for Speech Synthesis
  Python BSD-3-Clause 389 1,614 40 3 Updated Sep 2, 2020

• ngc-container-replicator

  NGC Container Replicator

  deep-learning containers nvidia-docker nvidia-gpus
  Python BSD-3-Clause 8 11 2 (1 issue needs help) 1 Updated Aug 31, 2020

• sampleQAT

  Inference of quantization aware trained networks using TensorRT
  Python Apache-2.0 1 12 0 0 Updated Aug 19, 2020

• flownet2-pytorch

  Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
  Python 517 2,042 110 (1 issue needs help) 7 Updated Aug 5, 2020

• FastPhotoStyle

  Style transfer, deep learning, feature transform
  Python 1,101 10,384 43 8 Updated Jul 1, 2020

• sentiment-discovery

  Unsupervised Language Modeling at scale for robust sentiment classification
  Python 192 1,005 38 6 Updated Jun 28, 2020

• pix2pixHD

  Synthesizing and manipulating 2048x1024 images with conditional GANs

  deep-neural-networks computer-vision deep-learning computer-graphics pytorch generative-adversarial-network gan
  Python 1,001 4,804 138 9 Updated Jun 25, 2020

• PixelView

  A compact and extensible image viewer
  Python Apache-2.0 0 3 0 0 Updated Jun 22, 2020

• partialconv

  A New Padding Scheme: Partial Convolution based Padding
  Python 161 790 14 1 Updated Jun 3, 2020

• container-wiki

  Documentation for the NVIDIA cloud native technologies
  Python CC-BY-4.0 0 1 0 0 Updated Feb 18, 2020

• UnsupervisedLandmarkLearning

  Implementation for the unsupervised latent landmark learning work from NVIDIA Applied Deep Learning Research
  Python 4 25 0 0 Updated Feb 12, 2020