NVIDIA Corporation

Pinned

nvidia-docker Public

Build and run Docker containers leveraging NVIDIA GPUs

Makefile 14.6k 1.9k
DeepLearningExamples Public

Deep Learning Examples

Python 8.4k 2.3k
tensorflow Public

An Open Source Machine Learning Framework for Everyone

C++ 469 73
open-gpu-kernel-modules Public

NVIDIA Linux open GPU kernel module source

C 11.3k 736

NeMo Public
NeMo: a toolkit for conversational AI

Jupyter Notebook 4,238 Apache-2.0 998 28 43 Updated May 21, 2022
spark-rapids Public
Spark RAPIDS plugin - accelerate Apache Spark with GPUs

Scala 421 Apache-2.0 139 601 (24 issues need help) 21 Updated May 21, 2022
cuCollections Public

C++ 141 Apache-2.0 37 30 (6 issues need help) 11 Updated May 21, 2022
MinkowskiEngine Public
Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

Python 1,593 226 98 5 Updated May 21, 2022
DALI Public
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

C++ 3,859 502 173 (38 issues need help) 29 Updated May 21, 2022
NVFlare Public
NVIDIA Federated Learning Application Runtime Environment

Python 187 Apache-2.0 42 24 7 Updated May 21, 2022
Megatron-LM Public
Ongoing research training transformer models at scale

Python 2,955 583 116 8 Updated May 20, 2022
cub Public
Cooperative primitives for CUDA C++.

Cuda 1,182 BSD-3-Clause 374 63 (1 issue needs help) 18 Updated May 20, 2022
spark-rapids-jni Public
RAPIDS Accelerator JNI For Apache Spark

C++ 8 Apache-2.0 14 10 4 Updated May 20, 2022
aistore Public
AIStore: scalable storage for AI applications

Go 595 MIT 60 2 0 Updated May 20, 2022