NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
- tensorrt
- nvidia
- deep-learning
- inference
47,133 repositories indexed
97–120 of 704 · Most stars
NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
A MNIST-like fashion product database. Benchmark :point_down:
Software that can generate photos from paintings, turn horses into zebras, perform style transfer, and more.
🏄 Scalable embedding, reasoning, ranking for images and sentences with CLIP
A standard API for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)
Easy-to-use Speech Toolkit including Self-Supervised Learning model, SOTA/Streaming ASR with punctuation, Streaming TTS with text frontend, Speaker Verification System, End-to-End Speech Translation and Keyword Spotting. Won NAACL2022 Best Demo Award.
Hierarchical Reasoning Model Official Release
Collection of scripts to aggregate image data for the purposes of training an NSFW Image Classifier
An open-source, tool-augmented conversational language model from Fudan University
🐍 Geometric Computer Vision Library for Spatial AI
Lime: Explaining the predictions of any machine learning classifier
A collaboration friendly studio for NeRFs
pix2code: Generating Code from a Graphical User Interface Screenshot
A PyTorch-based Speech Toolkit
Low-code framework for building custom LLMs, neural networks, and other AI models
PyTorch implementation of the U-Net for image semantic segmentation with high quality images
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch
Refine high-quality datasets and visual AI models
Deep Learning in Javascript. Train Convolutional Neural Networks (or ordinary ones) in your browser.
Style transfer, deep learning, feature transform
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
OpenVINO™ is an open source toolkit for optimizing and deploying AI inference