NVIDIA Research Projects

All

349 repositories

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
•45•943•21•2•Updated Feb 27, 2025Feb 27, 2025
catk
Public
Closed-Loop Supervised Fine-Tuning of Tokenized Traffic Models. CVPR 2025.
autonomous-driving imitation-learning traffic-simulation waymo-open-dataset supervised-finetuning next-token-prediction cvpr2025
Python
•
Other
•3•62•1•0•Updated Feb 26, 2025Feb 26, 2025
ProtoMotions
Public
Python
•
Other
•40•526•13•2•Updated Feb 26, 2025Feb 26, 2025
HOVER
Public
HOVER
Python
•
Apache License 2.0
•7•130•0•0•Updated Feb 25, 2025Feb 25, 2025
RADIO
Public
Official repository for "AM-RADIO: Reduce All Domains Into One"
Python
•
Other
•38•925•28•3•Updated Feb 25, 2025Feb 25, 2025
X-MOBILITY
Public
X-MOBILITY
Python
•
Apache License 2.0
•1•15•0•0•Updated Feb 25, 2025Feb 25, 2025
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
•265•883•27•8•Updated Feb 25, 2025Feb 25, 2025
svraster
Public
Jupyter Notebook
•
Other
•8•281•5•0•Updated Feb 25, 2025Feb 25, 2025
MobilityGen
Public
Data Generation Pipeline for Mobility
Python
•
Apache License 2.0
•5•43•1•5•Updated Feb 25, 2025Feb 25, 2025
scene_synthesizer
Public
Python package to create manipulation scenes.
Python
•
Apache License 2.0
•15•145•2•0•Updated Feb 23, 2025Feb 23, 2025
GatedDeltaNet
Public
[ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
Python
•
Other
•9•137•2•0•Updated Feb 23, 2025Feb 23, 2025
COAT
Public
[ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
Python
•
Apache License 2.0
•4•90•1•0•Updated Feb 22, 2025Feb 22, 2025
FoundationStereo
Public
9•472•8•0•Updated Feb 20, 2025Feb 20, 2025
RVT
Public
Official Code for RVT-2 and RVT
robotics object-manipulation rvt rlbench rvt-2
Jupyter Notebook
•
Other
•41•316•14•0•Updated Feb 14, 2025Feb 14, 2025
Sana
Public
SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformer
Python
•
Apache License 2.0
•209•3.5k•41•3•Updated Feb 12, 2025Feb 12, 2025
QLIP
Public
[arXiv: 2502.05178] QLIP: Text-Aligned Visual Tokenization Unifies Auto-Regressive Multimodal Understanding and Generation
JavaScript
•1•55•2•0•Updated Feb 12, 2025Feb 12, 2025
NVBit
Public
23•235•72•0•Updated Feb 11, 2025Feb 11, 2025
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
•107•365•55•0•Updated Feb 10, 2025Feb 10, 2025
VILA
Public
VILA is a family of state-of-the-art vision language models (VLMs) for diverse multimodal AI tasks across the edge, data center, and cloud.
Python
•
Apache License 2.0
•239•3k•65•4•Updated Feb 10, 2025Feb 10, 2025
earth2grid
Public
Python
•
Apache License 2.0
•4•20•8•1•Updated Feb 7, 2025Feb 7, 2025
verilog-eval
Public
Verilog evaluation benchmark for large language model
SystemVerilog
•
Other
•35•226•3•3•Updated Feb 7, 2025Feb 7, 2025
FoundationPose
Public
[CVPR 2024 Highlight] FoundationPose: Unified 6D Pose Estimation and Tracking of Novel Objects
Python
•
Other
•254•1.8k•55•5•Updated Feb 4, 2025Feb 4, 2025
gbrl
Public
Gradient Boosting Reinforcement Learning (GBRL)
C++
•
Other
•6•101•0•0•Updated Feb 4, 2025Feb 4, 2025
gbrl_sb3
Public
GBRL-based Actor-Critic algorithms implemented in stable-baselines3
Python
•
Other
•2•32•0•0•Updated Feb 4, 2025Feb 4, 2025
tracks-to-4d
Public
Python
•1•44•4•0•Updated Feb 2, 2025Feb 2, 2025
collab-sim
Public
Python
•
Other
•0•14•0•0•Updated Feb 1, 2025Feb 1, 2025
CircuitOps
Public
Python
•
Apache License 2.0
•15•76•3•0•Updated Jan 30, 2025Jan 30, 2025
OmniDrive
Public
Python
•16•292•29•1•Updated Jan 29, 2025Jan 29, 2025
EAGLE
Public
Eagle Family: Exploring Model Designs, Data Recipes and Training Strategies for Frontier-Class Multimodal LLMs
demo eagle llama lmm nvdia huggingface gpt4 large-language-models llm mllm
Python
•38•614•20•0•Updated Jan 28, 2025Jan 28, 2025
neural_rx
Public
Real-Time Inference of 5G NR Multi-user MIMO Neural Receivers
deep-learning receiver mimo 5g-nr
Jupyter Notebook
•
Other
•10•47•2•0•Updated Jan 28, 2025Jan 28, 2025