Stars
State-of-the-art Image & Video CLIP, Multimodal Large Language Models, and More!
The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use th…
A list of awesome compiler projects and papers for tensor computation and deep learning.
A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
[NeurIPS 2022 Spotlight] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training
Code release for SLIP Self-supervision meets Language-Image Pre-training
A deep learning library for video understanding research.
Ongoing research training transformer models at scale
Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research techniques without significant engineering overhead.Specifi…
3D mesh stylization driven by a text input in PyTorch
Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!
PyTorch android examples of usage in applications
Graph Neural Network Library for PyTorch
Machine learning metrics for distributed, scalable PyTorch applications.
NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
A library for efficient similarity search and clustering of dense vectors.
This is the code for Mark Rober's Domino Robot project. You can find a bunch more info about the project, including details about the software architecture on my website.
Set of Python bindings to C++ libraries which provides full HW acceleration for video decoding, encoding and GPU-accelerated color space and pixel format conversions
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
Tooling for the Common Objects In 3D dataset.
A collection of tools for neural compression enthusiasts.
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Transformer related optimization, including BERT, GPT
Write PyTorch controllers, test them in simulation, and seamlessly transfer to real-time hardware.