NVIDIA Research Projects

All

335 repositories

gbrl_sb3
Public
GBRL-based Actor-Critic algorithms implemented in stable-baselines3
Python
•
Other
•2•25•0•0•Updated Dec 11, 2024Dec 11, 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
•168•2.1k•53•3•Updated Dec 11, 2024Dec 11, 2024
ProtoMotions
Public
Python
•
Other
•27•367•7•1•Updated Dec 10, 2024Dec 10, 2024
GatedDeltaNet
Public
Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
Python
•
Other
•8•35•0•0•Updated Dec 10, 2024Dec 10, 2024
RADIO
Public
Official repository for "AM-RADIO: Reduce All Domains Into One"
Jupyter Notebook
•
Other
•36•838•22•3•Updated Dec 10, 2024Dec 10, 2024
Sana
Public
SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformer
Python
•
Other
•74•1.6k•32•2•Updated Dec 10, 2024Dec 10, 2024
catk
Public
Closed-Loop Supervised Fine-Tuning of Tokenized Traffic Models
autonomous-driving imitation-learning traffic-simulation waymo-open-dataset supervised-finetuning next-token-prediction
Python
•
Other
•0•16•0•0•Updated Dec 10, 2024Dec 10, 2024
earth2grid
Public
Python
•
Apache License 2.0
•2•16•5•0•Updated Dec 10, 2024Dec 10, 2024
CircuitOps
Public
Python
•
Apache License 2.0
•14•72•3•1•Updated Dec 9, 2024Dec 9, 2024
COAT
Public
Python
•1•42•0•0•Updated Dec 9, 2024Dec 9, 2024
scene_synthesizer
Public
Python package to create manipulation scenes.
Python
•
Apache License 2.0
•2•17•0•0•Updated Dec 9, 2024Dec 9, 2024
edm2
Public
EDM2 and Autoguidance -- Official PyTorch implementation
Python
•
Other
•23•557•8•0•Updated Dec 9, 2024Dec 9, 2024
hymba
Public
Python
•7•112•1•0•Updated Dec 8, 2024Dec 8, 2024
MaskLLM
Public
[NeurIPS 24 Spotlight] MaskLLM: Learnable Semi-structured Sparsity for Large Language Models
Python
•
Other
•13•128•3•0•Updated Dec 7, 2024Dec 7, 2024
EdgeRunner
Public
EdgeRunner: Auto-regressive Auto-encoder for Efficient Mesh Generation
mesh auto-regressive image-to-mesh point-cloud-to-mesh
Python
•
Other
•5•130•2•0•Updated Dec 6, 2024Dec 6, 2024
distillnerf
Public
[NeurIPS 2024] DistillNeRF: Perceiving 3D Scenes from Single-Glance Images by Distilling Neural Fields and Foundation Model Features
Python
•
Other
•0•15•0•0•Updated Dec 5, 2024Dec 5, 2024
6dof-graspnet
Public
Implementation of 6-DoF GraspNet with tensorflow and python. This repo has been tested with python 2.7 and tensorflow 1.12.
Python
•
MIT License
•58•220•17•0•Updated Dec 4, 2024Dec 4, 2024
MinVIS
Public
Python
•
Other
•22•265•7•0•Updated Dec 4, 2024Dec 4, 2024
flip
Public
A tool for visualizing and communicating the errors in rendered images.
C++
•
Other
•42•526•1•0•Updated Dec 3, 2024Dec 3, 2024
DRAIL
Public
[NeurIPS'24] The Official PyTorch implementation of DRAIL
reinforcement-learning robotics imitation-learning adversarial-machine-learning diffusion-models generative-adversarial-imitation-learning
Python
•
Other
•3•19•0•0•Updated Dec 3, 2024Dec 3, 2024
collab-sim
Public
Python
•
Other
•0•4•0•0•Updated Dec 2, 2024Dec 2, 2024
nvTorchCam
Public
Python
•
Apache License 2.0
•5•114•0•0•Updated Dec 2, 2024Dec 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
•238•830•22•7•Updated Dec 2, 2024Dec 2, 2024
fova-depth
Public
Python
•
Other
•1•10•1•0•Updated Dec 2, 2024Dec 2, 2024
watch-it-move
Public
Jupyter Notebook
•
Other
•9•65•1•0•Updated Nov 30, 2024Nov 30, 2024
rule-hierarchies
Public
Python
•
Other
•2•12•0•0•Updated Nov 26, 2024Nov 26, 2024
curobo
Public
CUDA Accelerated Robot Library
robotics cuda motion-planning pytorch
Python
•
Other
•127•808•47•2•Updated Nov 25, 2024Nov 25, 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
•104•344•54•2•Updated Nov 25, 2024Nov 25, 2024
Differentiable-Global-Router
Public
Python
•
Apache License 2.0
•2•12•0•0•Updated Nov 25, 2024Nov 25, 2024
TokenBench
Public
A Video Tokenizer Evaluation Dataset
Python
•
Apache License 2.0
•3•54•0•0•Updated Nov 23, 2024Nov 23, 2024