Stars
State-of-the-art framework for fast, large-scale training and inference of diffusion models
NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the effective training time by minimizing the downtime due to fa…
Scalable toolkit for efficient model reinforcement
[ICCV 2025] Repo for Objaverse++, Curated 3D Object Dataset with Quality Annotations
Local UI to run and train LLMs and diffusion models, including Qwen3.8, Kimi K3, MiniMax-H3, Gemma 4, DeepSeek-V4, FLUX and more.
NVIDIA 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.
BioNeMo Recipes: For building and adapting AI models in drug discovery at scale
A tool to configure, launch and manage your machine learning experiments.
🎨 ML Visuals contains figures and templates which you can reuse and customize to improve your scientific writing.
Segment Anything Model for large-scale, vectorized road network extraction from aerial imagery. CVPRW 2024
Scalable data pre processing and curation toolkit for LLMs
A collection of design patterns/idioms in Python
✨✨Latest Advances on Multimodal Large Language Models
Robust recipes to align language models with human and AI preferences
Freeing data processing from scripting madness by providing a set of platform-agnostic customizable pipeline processing blocks.
Minimalistic large language model 3D-parallelism training
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance…
Scalable toolkit for efficient model alignment
Ongoing research training transformer models at scale
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)
Provides end-to-end model development pipelines for LLMs and Multimodal models that can be launched on-prem or cloud-native.
State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.
Machine Learning Engineering Open Book
Resources from the EleutherAI Math Reading Group
neuralsim: 3D surface reconstruction and simulation based on 3D neural rendering.