NVIDIA Corporation

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,819 496 166 (33 issues need help) 30 Updated Apr 29, 2022
apex Public
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Python 6,285 BSD-3-Clause 987 543 38 Updated Apr 29, 2022
cutlass Public
CUDA Templates for Linear Algebra Subroutines

C++ 1,758 322 23 5 Updated Apr 29, 2022
spark-rapids Public
Spark RAPIDS plugin - accelerate Apache Spark with GPUs

Scala 410 Apache-2.0 132 585 (24 issues need help) 14 Updated Apr 29, 2022
MatX Public
An efficient C++17 GPU numerical computing library with Python-like syntax

C++ 473 26 7 0 Updated Apr 29, 2022
NeMo Public
NeMo: a toolkit for conversational AI

Jupyter Notebook 4,141 Apache-2.0 967 30 36 Updated Apr 29, 2022
spark-rapids-jni Public
RAPIDS Accelerator JNI For Apache Spark

Cuda 7 Apache-2.0 8 8 1 Updated Apr 29, 2022
NVFlare Public
NVIDIA Federated Learning Application Runtime Environment

Python 169 Apache-2.0 40 23 9 Updated Apr 29, 2022
aistore Public
AIStore: scalable storage for AI applications

Go 587 MIT 60 3 0 Updated Apr 29, 2022
Torch-TensorRT Public
PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT

Jupyter Notebook 1,377 BSD-3-Clause 173 87 (2 issues need help) 12 Updated Apr 29, 2022

Pinned