jaybdub

jupyter_clickable_image_widget

jetbot

An educational AI robot based on NVIDIA Jetson Nano.

stablediffusion

High-Resolution Image Synthesis with Latent Diffusion Models

torch2trt

An easy to use PyTorch to TensorRT converter

AITemplate

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.

AlignPS

Code for CVPR 2021 paper: Anchor-Free Person Search

amirstan_plugin

Useful tensorrt plugin. For pytorch and mmdetection model conversion.

bubble_bop

A simple game for NVIDIA Jetson

caffe-onnx

caffe model convert to onnx model

jaybdub.github.io

jbvim

jcontainers

jetnet

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

jetson-containers

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T

jetson-gpio

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

keras-yolo2

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).

mmpose

OpenMMLab Pose Estimation Toolbox and Benchmark.

NeMo

NeMo: a toolkit for conversational AI

nvim_setup

NVISII

nvvl

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

orbit

Unified framework for robot learning built on NVIDIA Isaac Sim

pyroomacoustics

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.

pythonocc-demos

A set of examples and demos for the pythonocc CAD package

pytorch-ssd

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 like CoordConv.

segmentation_models.pytorch

Segmentation models with pretrained backbones. PyTorch.

Swin-Transformer

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

TensorRT

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

trt_pose

Real-time pose estimation accelerated with NVIDIA TensorRT

Wave-U-Net-Pytorch

Improved Wave-U-Net implemented in Pytorch