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 upPinned repositories
Repositories
-
aistore
AIStore: scalable storage for AI applications
-
NeMo
NeMo: a toolkit for conversational AI
-
spark-rapids
Spark RAPIDS plugin - accelerate Apache Spark with GPUs
-
PyProf
A GPU performance profiling tool for PyTorch models
-
TRTorch
PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT
-
clara-dicom-adapter
DICOM Adapter is a component of the Clara Deploy SDK which facilitates integration with DICOM compliant systems, enables ingestion of imaging data, helps triggering of jobs with configurable rules and offers pushing the output of jobs to PACS systems.
-
Imageinary
Imageinary is a reproducible mechanism which is used to generate large image datasets at various resolutions. The tool supports multiple image types, including JPEGs, PNGs, BMPs, RecordIO, and TFRecord files
-
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.
-
DALI
A library containing both highly optimized building blocks and an execution engine for data pre-processing in deep learning applications
-
CUDALibrarySamples
CUDA Library Samples
-
Megatron-LM
Ongoing research training transformer language models at scale, including: BERT & GPT-2
-
nvidia-gcp-samples
NVIDIA GPU Accelerated Application Samples in Google Cloud Platform
-
deepops
Tools for building GPU clusters
-
go-gpuallocator
Go Abstraction for Allocating NVIDIA GPUs with Custom Policies
-
DALI_extra
Test data for DALI project
-
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
-
MinkowskiEngine
Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
-
flowtron
Flowtron is an auto-regressive flow-based generative network for text to speech synthesis with control over speech variation and style transfer
-
framework-determinism
Providing determinism in deep learning frameworks
-
pyxis
Container plugin for Slurm Workload Manager
-
clara-train-examples
Example notebooks demonstrating how to use Clara Train to build Medical Imaging Deep Learning models
-
TensorRT
TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.
-
mellotron
Mellotron: a multispeaker voice synthesis model based on Tacotron 2 GST that can make a voice emote and sing without emotive or singing training data
-
tacotron2
Tacotron 2 - PyTorch implementation with faster-than-realtime inference
-
HugeCTR
HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training
-
jitify
A single-header C++ library for simplifying the use of CUDA Runtime Compilation (NVRTC).