NVIDIA Corporation

NVIDIA Corporation's repositories

NeMo

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

Language:PythonApache-2.011176 201 2182

Megatron-LM

Ongoing research training transformer models at scale

Language:PythonNOASSERTION9616 160 633

DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

Language:C++Apache-2.05026 91 1580

warp

A Python framework for high performance GPU simulation and graphics

Language:PythonNOASSERTION3963 54 206

k8s-device-plugin

NVIDIA device plugin for Kubernetes

Language:GoApache-2.02628 61 466

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.

Language:PythonApache-2.01708 36 282

aistore

AIStore: scalable storage for AI applications

Language:GoMIT1198 45 86

cccl

CUDA Core Compute Libraries

Language:C++NOASSERTION1045 32 1224

modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

Language:PythonApache-2.0851 37 260

spark-rapids

Spark RAPIDS plugin - accelerate Apache Spark with GPUs

Language:ScalaApache-2.0775 46 5095

NVFlare

NVIDIA Federated Learning Application Runtime Environment

Language:PythonApache-2.0582 20 294

NeMo-Aligner

Scalable toolkit for efficient model alignment

Language:PythonApache-2.0466 12 63

cuda-quantum

C++ and Python support for the CUDA Quantum programming model for heterogeneous quantum-classical workflows

Language:C++NOASSERTION448 21 611

NeMo-Framework-Launcher

Provides end-to-end model development pipelines for LLMs and Multimodal models that can be launched on-prem or cloud-native.

Language:PythonApache-2.0433 21 36

NeMo-Curator

Scalable toolkit for data curation

Language:Jupyter NotebookApache-2.0398 15 71

NeMo-text-processing

NeMo text processing for ASR and TTS

Language:PythonApache-2.0251 17 32

Fuser

A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")

Language:C++NOASSERTION239 18 593

JAX-Toolbox

Language:Jupyter NotebookApache-2.0215 25 175

modulus-sym

Framework providing pythonic APIs, algorithms and utilities to be used with Modulus core to physics inform model training as well as higher level abstraction for domain experts

Language:PythonApache-2.0148 14 68