Open source for data and AI

749 projects · 100 relicensed

Nvidia open-source projects

12 projects · 111,559 stars · nvidia.com

Website
Custom or mixed (see the LICENSE file)ML frameworks18.3k starsMegatron-LMOngoing research training transformer models at scale.Nvidia · +52 a weekCustom or mixed (see the LICENSE file)Model serving and local AI15k starsTensorRT-LLMTensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs.Nvidia · +53 a weekCustom or mixed (see the LICENSE file)ML frameworks12.3k starscosmosNVIDIA Cosmos is an open platform of world models, datasets, and tools that enables developers to build Physical AI for robots, autonomous vehicles, smart infrastructure, and more.Nvidia · +59 a weekCustom or mixed (see the LICENSE file)ML frameworks10.7k starscutlassCUDA Templates and Python DSLs for High-Performance Linear Algebra.Nvidia · +32 a weekApache 2.0Query and processing engines9,797 starscudfcuDF - GPU DataFrame Library.Nvidia · +5 a weekBSD 3-ClauseML frameworks9,001 starsapexA PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch.Nvidia · +0 a weekApache 2.0ML frameworks8,297 starsIsaac-GR00TNVIDIA Isaac GR00T N1.7 - A Foundation Model for Generalist Robots.Nvidia · +34 a weekApache 2.0Model serving and local AI6,459 starsFasterTransformerTransformer related optimization, including BERT, GPT.Nvidia · +1 a week · release/v5.3_tag, Jan 23, 2023Apache 2.0ML frameworks5,793 starsDALIA GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.Nvidia · +4 a weekApache 2.0ML frameworks5,319 starscumlNVIDIA cuML: GPU-Accelerated Machine Learning.Nvidia · +4 a weekBSD 3-ClauseImage, video and speech generation5,305 starstacotron2Tacotron 2 - PyTorch implementation with faster-than-realtime inference.Nvidia · +2 a weekCustom or mixed (see the LICENSE file)ML frameworks5,202 starsncclOptimized primitives for collective multi-GPU communication.Nvidia · +14 a week
ProjectStars
Megatron-LMOngoing research training transformer models at scale.Nvidia · ML frameworksCustom or mixed (see the LICENSE file)18.3k
TensorRT-LLMTensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs.Nvidia · Model serving and local AICustom or mixed (see the LICENSE file)15k
cosmosNVIDIA Cosmos is an open platform of world models, datasets, and tools that enables developers to build Physical AI for robots, autonomous vehicles, smart infrastructure, and more.Nvidia · ML frameworksCustom or mixed (see the LICENSE file)12.3k
cutlassCUDA Templates and Python DSLs for High-Performance Linear Algebra.Nvidia · ML frameworksCustom or mixed (see the LICENSE file)10.7k
cudfcuDF - GPU DataFrame Library.Nvidia · Query and processing enginesApache 2.09,797
apexA PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch.Nvidia · ML frameworksBSD 3-Clause9,001
Isaac-GR00TNVIDIA Isaac GR00T N1.7 - A Foundation Model for Generalist Robots.Nvidia · ML frameworksApache 2.08,297
FasterTransformerTransformer related optimization, including BERT, GPT.Nvidia · Model serving and local AIApache 2.06,459
DALIA GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.Nvidia · ML frameworksApache 2.05,793
cumlNVIDIA cuML: GPU-Accelerated Machine Learning.Nvidia · ML frameworksApache 2.05,319
tacotron2Tacotron 2 - PyTorch implementation with faster-than-realtime inference.Nvidia · Image, video and speech generationBSD 3-Clause5,305
ncclOptimized primitives for collective multi-GPU communication.Nvidia · ML frameworksCustom or mixed (see the LICENSE file)5,202
About and links