NVlabs's Repositories

Official PyTorch implementation of 3D Gaussian Mapping (3DGM)

⭐ 85 🌐 Public

Implementation of 6-DoF GraspNet with tensorflow and python. This repo has been tested with python 2.7 and tensorflow 1.12.

⭐ 275 🌐 Public

Official PyTorch implementation of A-ViT: Adaptive Tokens for Efficient Vision Transformer (CVPR 2022)

⭐ 165 🌐 Public

Adversarial Augmentation Against Adversarial Attacks

⭐ 32 🌐 Public

ACID: Action-Conditional Implicit Visual Dynamics for Deformable Object Manipulation

⭐ 74 🌐 Public

This repository contains a sample of the grasping dataset and tools to visualize grasps, generate random scenes, and render observations. The two sample files are in the HDF5 format.

⭐ 149 🌐 Public

AdaBatch: Adaptive Batch Sizes for Training Deep Neural Networks

⭐ 42 🌐 Public

adaptive-prediction

⭐ 74 🌐 Public

⭐ 381 🌐 Public

affordance_diffusion

Codes for "Affordance Diffusion: Synthesizing Hand-Object Interactions"

⭐ 129 🌐 Public

AFNO-transformer

Adaptive FNO transformer - official Pytorch implementation

⭐ 268 🌐 Public

A Vision-Language-Model for Detecting and Reasoning Over Failures in Robotic Manipulation

⭐ 54 🌐 Public

Official pytorch implementation of Active Learning for deep object detection via probabilistic modeling (ICCV 2021)

⭐ 176 🌐 Public

⭐ 18 🌐 Public

Alias-Free GAN project website and code

⭐ 1296 🌐 Public

⭐ 258 🌐 Public

⭐ 240 🌐 Public

⭐ 52 🌐 Public

⭐ 32 🌐 Public

⭐ 14 🌐 Public

⭐ 19 🌐 Public

⭐ 156 🌐 Public

BayesSimIG: Scalable Parameter Inference for Adaptive Domain Randomization with Isaac Gym

⭐ 44 🌐 Public

Bellman-Corrected Tree Search from https://arxiv.org/abs/2107.01715 (Neurips 2021)

⭐ 2 🌐 Public

⭐ 248 🌐 Public

⭐ 172 🌐 Public

Simulation scripts and assets for the BioTac.

⭐ 33 🌐 Public

⭐ 58 🌐 Public

Official PyTorch implementation of BLADE: Single-view Body Mesh Estimation through Accurate Depth Estimation (CVPR 2025). BLADE tackles close-range human mesh recovery where perspective distortion is strongest, and solves for camera pose and focal length in addition to SMPL(-X) parameters.

⭐ 31 🌐 Public

[CVPR 2022 (oral)] Bongard-HOI for benchmarking few-shot visual reasoning

⭐ 73 🌐 Public

Bongard-LOGO is a Python code repository with the purpose of generating synthetic Bongard problems on a large scale with little human intervention.

⭐ 53 🌐 Public

[CVPR 2023] BundleSDF: Neural 6-DoF Tracking and 3D Reconstruction of Unknown Objects

⭐ 1310 🌐 Public

CabiNet: Scaling Object Rearrangement in Clutter

⭐ 24 🌐 Public

⭐ 598 🌐 Public

CARI4D: Category Agnostic 4D Reconstruction of Human-Object Interaction

⭐ 6 🌐 Public

Closed-Loop Supervised Fine-Tuning of Tokenized Traffic Models. CVPR Oral 2025.

⭐ 166 🌐 Public

⭐ 23 🌐 Public

A generative foundation model for the kilometer-scale atmosphere

⭐ 88 🌐 Public

Single-Stage Keypoint-based Category-level Object Pose Estimation from an RGB Image (ICRA 2022)

⭐ 329 🌐 Public

⭐ 702 🌐 Public

CGBN: CUDA Accelerated Multiple Precision Arithmetic (Big Num) using Cooperative Groups

⭐ 231 🌐 Public

⭐ 95 🌐 Public

[ICLR'24] Efficient Video Diffusion Models via Content-Frame Motion-Latent Decomposition

⭐ 41 🌐 Public

[ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training

⭐ 253 🌐 Public

⭐ 15 🌐 Public

Research package for MPC VR teleop with Isaac Sim and cuRobo

⭐ 42 🌐 Public

⭐ 3 🌐 Public

Cross-embOdiment Mobility Policy via ResiduAl RL and Skill Synthesis

⭐ 71 🌐 Public

concept_learning

⭐ 6 🌐 Public

Programmable Neural Network Compression

⭐ 149 🌐 Public

conditional-restir-prototype

⭐ 114 🌐 Public

ConformalKeypoint

⭐ 40 🌐 Public

⭐ 425 🌐 Public

contact_graspnet

Efficient 6-DoF Grasp Generation in Cluttered Scenes

⭐ 438 🌐 Public

Contrastive-Correspondence

Codes for "Learning Contrastive Representation for Semantic Correspondence"

⭐ 10 🌐 Public

⭐ 129 🌐 Public

⭐ 69 🌐 Public

CoordGAN: Self-Supervised Dense Correspondences Emerge from GANs

⭐ 25 🌐 Public

⭐ 13 🌐 Public

⭐ 28 🌐 Public

⭐ 122 🌐 Public

THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.

⭐ 85 🌐 Public 📦 Archived

⭐ 16 🌐 Public

CuLE: A CUDA port of the Atari Learning Environment (ALE)

⭐ 241 🌐 Public

CUDA Accelerated Robot Library

⭐ 1284 🌐 Public

A FastAPI http server which calls out to the curobo motion planning library

⭐ 11 🌐 Public

⭐ 19 🌐 Public

cuTAMP: Differentiable GPU-Parallelized Task and Motion Planning

⭐ 78 🌐 Public

⭐ 103 🌐 Public

⭐ 33 🌐 Public

⭐ 538 🌐 Public

data-orchestration

⭐ 5 🌐 Public

[ICML 2025 Spotlight] Direct Discriminative Optimization: Supercharging Diffusion/Autoregressive with GAN-type Discrimination

⭐ 107 🌐 Public

Deep Anticheat project

⭐ 8 🌐 Public

PyTorch implementation of the DeepIM framework

⭐ 98 🌐 Public

Official PyTorch implementation of Dreaming to Distill: Data-free Knowledge Transfer via DeepInversion (CVPR 2020)

⭐ 517 🌐 Public

Deep_Object_Pose

Deep Object Pose Estimation (DOPE) – ROS inference (CoRL 2018)

⭐ 1141 🌐 Public

This package provides a framework to automatically perform grasp tests on an arbitrary object model of choice.

⭐ 142 🌐 Public

denoising-diffusion-gan

Tackling the Generative Learning Trilemma with Denoising Diffusion GANs https://arxiv.org/abs/2112.07804

⭐ 752 🌐 Public

describe-anything

[ICCV 2025] Implementation for Describe Anything: Detailed Localized Image and Video Captioning

⭐ 1433 🌐 Public

dex-ycb-toolkit

A Python package that provides evaluation and visualization tools for the DexYCB dataset

⭐ 190 🌐 Public

This code corresponds to simulation environments used as part of the DexMimicGen project.

⭐ 191 🌐 Public

⭐ 102 🌐 Public

:couple: Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral) :couple:

⭐ 1301 🌐 Public

Joint Disentangling and Adaptation for Cross-Domain Person Re-Identification. ECCV'20 (Oral)

⭐ 88 🌐 Public

Official Pytorch implementation for Distilling Image Classifiers in Object detection (NeurIPS2021)

⭐ 32 🌐 Public

Pose estimation refiner

⭐ 62 🌐 Public

This repository contains code to reproduce some of the results from the paper Sionna RT: Differentiable Ray Tracing for Radio Propagation Modeling using the Sionna™ link-level simulator.

⭐ 69 🌐 Public

diff-rt-calibration

⭐ 45 🌐 Public

Differentiable-Global-Router

⭐ 21 🌐 Public

[ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation

⭐ 506 🌐 Public

A new adversarial purification method that uses the forward and reverse processes of diffusion models to remove adversarial perturbations.

⭐ 331 🌐 Public

[ICLR 2022] Accelerated Policy Learning with Parallel Differentiable Simulation

⭐ 345 🌐 Public

⭐ 109 🌐 Public

DiffusionNFT: Online Diffusion Reinforcement with Forward Process

⭐ 501 🌐 Public

⭐ 69 🌐 Public

Official PyTorch implementation of Data-free Knowledge Distillation for Object Detection, WACV 2021.

⭐ 63 🌐 Public

The Official PyTorch Implementation of DiscoBox.

⭐ 96 🌐 Public

Differentiable Cutting Simulator

⭐ 129 🌐 Public

[NeurIPS 2024] DistillNeRF: Perceiving 3D Scenes from Single-Glance Images by Distilling Neural Fields and Foundation Model Features

⭐ 36 🌐 Public