NVIDIA Corporation

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)

Megatron-LM

Ongoing research training transformer models at scale

cutlass

CUDA Templates for Linear Algebra Subroutines

TransformerEngine

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.

gpu-operator

NVIDIA GPU Operator creates/configures/manages GPUs atop Kubernetes

stdexec

`std::execution`, the proposed C++ framework for asynchronous and parallel programming.

aistore

AIStore: scalable storage for AI applications

MatX

An efficient C++17 GPU numerical computing library with Python-like syntax

cccl

CUDA Core Compute Libraries

modulus

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

spark-rapids

Spark RAPIDS plugin - accelerate Apache Spark with GPUs

NeMo-Aligner

Scalable toolkit for efficient model alignment

cuCollections

cuda-quantum

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

NeMo-Framework-Launcher

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

NeMo-Curator

Scalable toolkit for data curation

Fuser

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

VisRTX

NVIDIA OptiX based implementation of ANARI

JAX-Toolbox

JAX-Toolbox

RTX-AI-Toolkit

The NVIDIA RTX™ AI Toolkit is a suite of tools and SDKs for Windows developers to customize, optimize, and deploy AI models across RTX PCs and cloud.

earth2studio

Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.

spark-rapids-tools

User tools for Spark RAPIDS

knavigator

knavigator is a development, testing, and optimization toolkit for AI/ML scheduling systems at scale on Kubernetes.

k8s-driver-manager

The NVIDIA Driver Manager is a Kubernetes component which assist in seamless upgrades of NVIDIA Driver on each node of the cluster.

spark-rapids-jni

RAPIDS Accelerator JNI For Apache Spark

numba-cuda

edk2

NVIDIA fork of tianocore/edk2

NV-Kernels

Ubuntu kernels which are optimized for NVIDIA server systems

edk2-platforms

NVIDIA fork of tianocore/edk2-platforms

doca-sosreport

A unified tool for collecting system logs and other debug information