Explore @NVlabs Open Source projects

NVlabs

Stars
128,796
Global Org. Rank 74 (Top 0.03 %)
Registered over 11 years ago
Most used languages

Python
71.8 %

Jupyter Notebook
7.8 %

C++
6.7 %

HTML
4.7 %

Cuda 2.4 %

C
1.6 %

JavaScript
1.6 %

Others 3.6 %

instant-ngp

Instant neural graphics primitives: lightning fast NeRF and more

Cuda

15,102

stylegan

StyleGAN - Official TensorFlow Implementation

Python

13,882

stylegan2

StyleGAN2 - Official TensorFlow Implementation

Python

10,740

SPADE

Semantic Image Synthesis with SPADE

Python

7,518

stylegan3

Official PyTorch implementation of StyleGAN3

Python

6,108

neuralangelo

Official implementation of "Neuralangelo: High-Fidelity Neural Surface Reconstruction" (CVPR 2023)

Python

4,125

imaginaire

NVIDIA's Deep Imagination Team's PyTorch Library

Python

3,941

stylegan2-ada-pytorch

StyleGAN2-ADA - Official PyTorch implementation

Python

3,866

ffhq-dataset

Flickr-Faces-HQ Dataset (FFHQ)

Python

3,483

tiny-cuda-nn

Lightning fast C++/CUDA neural network framework

eg3d

MUNIT

Multimodal Unsupervised Image-to-Image Translation

Python

2,564

SegFormer

Official PyTorch implementation of SegFormer

Python

2,252

nvdiffrec

Official code for the CVPR 2022 (oral) paper "Extracting Triangular 3D Models, Materials, and Lighting From Images".

Python

2,019

few-shot-vid2vid

Pytorch implementation for few-shot photorealistic video-to-video translation.

Python

1,780

stylegan2-ada

StyleGAN2 with adaptive discriminator augmentation (ADA) - Official TensorFlow implementation

Python

1,778

FUNIT

Translate images to unseen domains in the test time with few example images.

Python

1,545

PWC-Net

PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume, CVPR 2018 (Oral)

Python

1,512

noise2noise

Noise2Noise: Learning Image Restoration without Clean Data - Official TensorFlow implementation of the ICML 2018 paper

Python

1,356

alias-free-gan

Alias-Free GAN project website and code

1,320

prismer

The implementation of "Prismer: A Vision-Language Model with Multi-Task Experts".

Python

1,287

DG-Net

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

Python

1,268

nvdiffrast

Nvdiffrast - Modular Primitives for High-Performance Differentiable Rendering

C++

1,137

edm

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)

Python

1,014

Deep_Object_Pose

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

Python

955

VoxFormer

Official PyTorch implementation of VoxFormer [CVPR 2023 Highlight]

Python

937

NVAE

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)

Python

889

BundleSDF

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

Python

842

ODISE

Official PyTorch implementation of ODISE: Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Models [CVPR 2023 Highlight]

Python

779

GroupViT

Official PyTorch implementation of GroupViT: Semantic Segmentation Emerges from Text Supervision, CVPR 2022.

Python

679

FasterViT

[ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention

Python

664

GA3C

Hybrid CPU/GPU implementation of the A3C algorithm for deep reinforcement learning.

Python

641

denoising-diffusion-gan

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

Python

634

genvs

610

sionna

Sionna: An Open-Source Library for Next-Generation Physical Layer Research

Jupyter Notebook

580

curobo

CUDA Accelerated Robot Library

Python

545

FB-BEV

Official PyTorch implementation of FB-BEV & FB-OCC - Forward-backward view transformation for vision-centric autonomous driving perception

Dancing2Music

planercnn

PlaneRCNN detects and reconstructs piece-wise planar surfaces from a single RGB image

Python

502

pacnet

Pixel-Adaptive Convolutional Neural Networks (CVPR '19)

CALM

DeepInversion

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

Python

474

EmerNeRF

PyTorch Implementation of EmerNeRF: Emergent Spatial-Temporal Scene Decomposition via Self-Supervision

Python

456

FAN

Official PyTorch implementation of Fully Attentional Networks

Python

454

FourCastNet

Initial public release of code, data, and model weights for FourCastNet

Python

421

GCVit

[ICML 2023] Official PyTorch implementation of Global Context Vision Transformers

Python

414

intrinsic3d

Intrinsic3D - High-Quality 3D Reconstruction by Joint Appearance and Geometry Optimization with Spatially-Varying Lighting (ICCV 2017)

C++

411

nvdiffmodeling

Differentiable rasterization applied to 3D model simplification tasks

Python

404

flip

A tool for visualizing and communicating the errors in rendered images.

C++

375

wetectron

Weakly-supervised object detection.

Python

355

FoundationPose

FoundationPose: Unified 6D Pose Estimation and Tracking of Novel Objects

JavaScript

349

nvdiffrecmc

Official code for the NeurIPS 2022 paper "Shape, Light, and Material Decomposition from Images using Monte Carlo Rendering and Denoising".

341

geomapnet

Geometry-Aware Learning of Maps for Camera Localization (CVPR2018)

Python

338

GLAMR

[CVPR 2022 Oral] Official PyTorch Implementation of "GLAMR: Global Occlusion-Aware Human Mesh Recovery with Dynamic Cameras”.

Python

329

LSGM

The Official PyTorch Implementation of "LSGM: Score-based Generative Modeling in Latent Space" (NeurIPS 2021)

Python

326

ssn_superpixels

Superpixel Sampling Networks (ECCV2018)

Python

323

DiffiT

Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation

315

FreeSOLO

FreeSOLO for unsupervised instance segmentation, CVPR 2022

Python

307

long-video-gan

Official PyTorch implementation of LongVideoGAN

Python

297

neuralrgbd

Neural RGB→D Sensing: Per-pixel depth and its uncertainty estimation from a monocular RGB video

Python

294

selfsupervised-denoising

High-Quality Self-Supervised Deep Image Denoising - Official TensorFlow implementation of the NeurIPS 2019 paper

Python

293

Taylor_pruning

Pruning Neural Networks with Taylor criterion in Pytorch

Python

279

timeloop

Timeloop performs modeling, mapping and code-generation for tensor algebra workloads on various accelerator architectures.

metfaces-dataset

few_shot_gaze

Pytorch implementation and demo of FAZE: Few-Shot Adaptive Gaze Estimation (ICCV 2019, oral)

Python

272

splatnet

SPLATNet: Sparse Lattice Networks for Point Cloud Processing (CVPR2018)

MinVIS

edm2

Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)

Python

261

contact_graspnet

Efficient 6-DoF Grasp Generation in Cluttered Scenes

Python

260

CenterPose

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

Python

251

trajdata

A unified interface to many trajectory forecasting datasets.

Python

245

STEP

STEP: Spatio-Temporal Progressive Learning for Video Action Detection. CVPR'19 (Oral)

Python

244

matchlib

SystemC/C++ library of commonly-used hardware functions and components for HLS.

C++

235

sim-web-visualizer

Web Based Visualizer for Simulation Environments

Python

231

SCOPS

SCOPS: Self-Supervised Co-Part Segmentation (CVPR'19)

Python

221

UMR

Self-supervised Single-view 3D Reconstruction

Python

221

DiffRL

[ICLR 2022] Accelerated Policy Learning with Parallel Differentiable Simulation

Python

220

cule

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

C++

216

SSV

Pytorch implementation of SSV: Self-Supervised Viewpoint Learning from Image Collections (CVPR 2020)

Python

214

DiffPure

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

Python

210

latentfusion

LatentFusion: End-to-End Differentiable Reconstruction and Rendering for Unseen Object Pose Estimation

I2SB

nvbio

NVBIO is a library of reusable components designed to accelerate bioinformatics applications using CUDA.

C++

193

6dof-graspnet

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

Python

186

NVBit

183

AFNO-transformer

Adaptive FNO transformer - official Pytorch implementation

Python

174

UnseenObjectClustering

Learning RGB-D Feature Embeddings for Unseen Object Instance Segmentation

Python

166

AL-MDN

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

Python

159

fermat

Fermat is a high performance research oriented physically based rendering system, trying to produce beautiful pictures following the mathematician’s principle of least time

C++

158

PoseCNN-PyTorch

PyTorch implementation of the PoseCNN framework

156

mask-auto-labeler

Python

153

mimicgen_environments

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

Bi3D

RVT

Official Code for RVT: Robotic View Transformer for 3D Object Manipulation

Python

147

condensa

Programmable Neural Network Compression

Python

146

traffic-behavior-simulation

Python

145

learningrigidity

Learning Rigidity in Dynamic Scenes with a Moving Camera for 3D Motion Field Estimation (ECCV 2018)

Python

144

ocrodeg

document image degradation

Jupyter Notebook

142

ocropus3

Repository collecting all the submodules for the new PyTorch-based OCR System.

Shell

141

100

CGBN

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

Cuda

139

NVlabs

Top repositories