Grow your team on GitHub
GitHub is home to over 50 million developers working together. Join them to grow your own development teams, manage permissions, and collaborate on projects.
Sign up
Pinned repositories
Repositories
TRTorch
PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT
aistore
AIStore: scalable storage for AI applications
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.
tacotron2
Tacotron 2 - PyTorch implementation with faster-than-realtime inference
NeMo
NeMo: a toolkit for conversational AI
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.
MinkowskiEngine
Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
spark-rapids
Spark RAPIDS plugin - accelerate Apache Spark with GPUs
DALI
A library containing both highly optimized building blocks and an execution engine for data pre-processing in deep learning applications
triton-inference-server
The Triton Inference Server provides a cloud inferencing solution optimized for NVIDIA GPUs.
Megatron-LM
Ongoing research training transformer language models at scale, including: BERT & GPT-2
flowtron
Flowtron is an auto-regressive flow-based generative network for text to speech synthesis with control over speech variation and style transfer
nvcomp
A library for fast lossless compression/decompression on the GPU
deepops
Tools for building GPU clusters
DeepLearningExamples
Deep Learning Examples
nvidia-container-toolkit
Build and run containers leveraging NVIDIA GPUs
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
PyProf
A GPU performance profiling tool for PyTorch models
spark-xgboost-examples
XGBoost GPU accelerated on Spark example applications
data-science-stack
NVIDIA Data Science stack tools
sampleQAT
Inference of quantization aware trained networks using TensorRT
tensorflow
An Open Source Machine Learning Framework for Everyone
TensorRT
TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.
gpu-operator
NVIDIA GPU Operator creates/configures/manages GPUs atop Kubernetes
ipyparaview
iPython widget for server-side ParaView rendering in Jupyter.
enroot
A simple yet powerful tool to turn traditional container/OS images into unprivileged sandboxes.
nvtx-plugins
Python bindings for NVTX
