Unified framework for robot learning built on NVIDIA Isaac Sim

An easy to use PyTorch to TensorRT converter

An educational AI robot based on NVIDIA Jetson Nano.

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T

A set of examples and demos for the pythonocc CAD package

High-Resolution Image Synthesis with Latent Diffusion Models

Easy to use neural networks for NVIDIA Jetson (and desktop too!)

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

Real-time pose estimation accelerated with NVIDIA TensorRT

Useful tensorrt plugin. For pytorch and mmdetection model conversion.

OpenMMLab Pose Estimation Toolbox and Benchmark.

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

Improved Wave-U-Net implemented in Pytorch

Code for CVPR 2021 paper: Anchor-Free Person Search

Pyroomacoustics is a package for audio signal processing for indoor applications. It was developed as a fast prototyping platform for beamforming algorithms in indoor scenarios.

TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.

Easy training on custom dataset. Various backends (MobileNet and SqueezeNet) supported. A YOLO demo to detect raccoon run entirely in brower is accessible at https://git.io/vF7vI (not on Windows).

NeMo: a toolkit for conversational AI

Segmentation models with pretrained backbones. PyTorch.

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas lik…

A Python library that enables the use of Jetson's GPIOs

A library that uses hardware acceleration to load sequences of video frames to facilitate machine learning training

caffe model convert to onnx model