PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.

DORA (Dataflow-Oriented Robotic Architecture) is middleware designed to streamline and simplify the creation of AI-based robotic applications. It offers low latency, composable, and distributed dat…

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T

The easiest way to serve AI apps and models - Build Model Inference APIs, Job queues, LLM apps, Multi-model pipelines, and more!

Sample code and notebooks for Generative AI on Google Cloud, with Gemini Enterprise Agent Platform

[ICCV 2023] OccNet: Scene as Occupancy

Official Code for DragGAN (SIGGRAPH 2023)

[CVPR 2023] An academic alternative to Tesla's occupancy network for autonomous driving.

The official implementation for the paper 'mmSampler: Efficient Frame Sampler for Multimodal Video Retrieval'.

[SIGIR 2022] CenterCLIP: Token Clustering for Efficient Text-Video Retrieval.

Berkeley Deep Drive-X (eXplanation) dataset

This repo contains documentation and code needed to use PACO dataset: data loaders and training and evaluation scripts for objects, parts, and attributes prediction models, query evaluation scripts…

Official implementation of arxiv paper "4K-NeRF: High Fidelity Neural Radiance Fields at Ultra High Resolutions"

Code release for "Learning Video Representations from Large Language Models"

[CVPR 2022] "MonoScene: Monocular 3D Semantic Scene Completion": 3D Semantic Occupancy Prediction from a single image

A framework for annotating 3D meshes using the predictions of a 2D semantic segmentation model.

Monocular Depth Estimation Toolbox based on MMSegmentation.

[ICLR 2023 Oral] Zero-Shot Image Restoration Using Denoising Diffusion Null-Space Model

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch

Cleanlab's open-source library is the standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

2DPASS: 2D Priors Assisted Semantic Segmentation on LiDAR Point Clouds (ECCV 2022) 🔥

Code Release for MeMViT Memory-Augmented Multiscale Vision Transformer for Efficient Long-Term Video Recognition, CVPR 2022

Versatile Diffusion: Text, Images and Variations All in One Diffusion Model, arXiv 2022 / ICCV 2023

[ICRA 2024] SuperFusion: Multilevel LiDAR-Camera Fusion for Long-Range HD Map Generation

GPU-ready Dockerfile to run Stability.AI stable-diffusion model v2 with a simple web interface. Includes multi-GPUs support.

Deep Learning Examples

OpenAI's Whisper dockerized and put behind FastAPI