NVIDIA Research Projects

All

325 repositories

RADIO
Public
Official repository for "AM-RADIO: Reduce All Domains Into One"
Python
•
Other
•32•780•20•2•Updated Nov 5, 2024Nov 5, 2024
Sana
Public
SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformer
10•523•10•0•Updated Nov 5, 2024Nov 5, 2024
timeloop
Public
Timeloop performs modeling, mapping and code-generation for tensor algebra workloads on various accelerator architectures.
C++
•
BSD 3-Clause "New" or "Revised" License
•102•335•54•1•Updated Nov 5, 2024Nov 5, 2024
consistory
Public
Python
•
Other
•0•48•1•0•Updated Nov 4, 2024Nov 4, 2024
COAT
Public
0•21•0•0•Updated Nov 4, 2024Nov 4, 2024
ProtoMotions
Public
Python
•
Other
•22•321•3•0•Updated Nov 3, 2024Nov 3, 2024
flip
Public
A tool for visualizing and communicating the errors in rendered images.
C++
•41•518•2•1•Updated Nov 3, 2024Nov 3, 2024
gbrl_sb3
Public
GBRL-based Actor-Critic algorithms implemented in stable-baselines3
Python
•
Other
•2•25•1•0•Updated Nov 3, 2024Nov 3, 2024
object_centric_diffusion
Public
SPOT: SE(3) Pose Trajectory Diffusion for Object-Centric Manipulation
JavaScript
•1•6•0•0•Updated Nov 1, 2024Nov 1, 2024
matchlib
Public
SystemC/C++ library of commonly-used hardware functions and components for HLS.
C++
•
Other
•43•260•1•1•Updated Nov 1, 2024Nov 1, 2024
GL0AM
Public
GL0AM GPU Accelerated Gate Level Logic Simulator
Python
•
Other
•0•2•0•0•Updated Oct 31, 2024Oct 31, 2024
DiffiT
Public
[ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation
14•452•4•0•Updated Oct 31, 2024Oct 31, 2024
VILA
Public
VILA - a multi-image visual language model with training, inference and evaluation recipe, deployable from cloud to edge (Jetson Orin and laptops)
Python
•
Apache License 2.0
•157•2k•54•2•Updated Oct 31, 2024Oct 31, 2024
MaskLLM
Public
[NeurIPS 24 Spotlight] MaskLLM: Learnable Semi-structured Sparsity for Large Language Models
Python
•
Other
•11•107•2•0•Updated Oct 30, 2024Oct 30, 2024
mimicgen
Public
This code corresponds to simulation environments used as part of the MimicGen project.
Python
•
Other
•50•312•1•2•Updated Oct 30, 2024Oct 30, 2024
FoundationPose
Public
[CVPR 2024 Highlight] FoundationPose: Unified 6D Pose Estimation and Tracking of Novel Objects
Python
•
Other
•196•1.5k•56•5•Updated Oct 24, 2024Oct 24, 2024
M2T2
Public
M2T2: Multi-Task Masked Transformer for Object-centric Pick and Plac
Python
•
Other
•2•33•3•1•Updated Oct 23, 2024Oct 23, 2024
ConvSSM
Public
Python
•
Other
•3•59•0•0•Updated Oct 22, 2024Oct 22, 2024
nvTorchCam
Public
Python
•
Apache License 2.0
•4•110•0•0•Updated Oct 17, 2024Oct 17, 2024
LITA
Public
Python
•
Other
•10•144•5•0•Updated Oct 14, 2024Oct 14, 2024
CircuitOps
Public
Python
•
Apache License 2.0
•14•70•3•0•Updated Oct 11, 2024Oct 11, 2024
DiSECt
Public
Differentiable Cutting Simulator
Python
•
Other
•12•101•0•0•Updated Oct 10, 2024Oct 10, 2024
earth2grid
Public
Python
•
Apache License 2.0
•2•15•5•0•Updated Oct 7, 2024Oct 7, 2024
gbrl
Public
Gradient Boosting Reinforcement Learning (GBRL)
C++
•
Other
•4•85•0•0•Updated Oct 6, 2024Oct 6, 2024
MambaVision
Public
Official PyTorch Implementation of MambaVision: A Hybrid Mamba-Transformer Vision Backbone
deep-learning transformers image-classification mamba visual-recognition self-attention hybrid-models huggingface-transformers vision-transformer foundation-models
Python
•
Other
•43•795•12•2•Updated Oct 5, 2024Oct 5, 2024
Minitron
Public
A family of compressed models obtained via pruning and knowledge distillation
16•276•2•0•Updated Oct 2, 2024Oct 2, 2024
ParallelInversion
Public
Parallel Inversion of Neural Radiance Fields for Robust Pose Estimation (ICRA 2023)
deep-learning pytorch rgb object-pose-estimation
Cuda
•
Other
•4•28•3•0•Updated Oct 2, 2024Oct 2, 2024
sionna
Public
Sionna: An Open-Source Library for Next-Generation Physical Layer Research
open-source machine-learning deep-learning reproducible-research gpu-acceleration communications 5g 6g link-level-simulation
Python
•
Other
•225•800•18•7•Updated Oct 1, 2024Oct 1, 2024
DoRA
Public
[ICML2024 (Oral)] Official PyTorch implementation of DoRA: Weight-Decomposed Low-Rank Adaptation
deep-neural-networks deep-learning lora commonsense-reasoning vision-and-language large-language-models parameter-efficient-tuning instruction-tuning large-vision-language-models parameter-efficient-fine-tuning
Python
•
Other
•40•615•9•1•Updated Oct 1, 2024Oct 1, 2024
nvdiffrast
Public
Nvdiffrast - Modular Primitives for High-Performance Differentiable Rendering
C++
•
Other
•155•1.4k•10•1•Updated Sep 30, 2024Sep 30, 2024