AutoGPT is the vision of accessible AI for everyone, to use and to build on. Our mission is to provide the tools, so that you can focus on what matters.

Stable Diffusion web UI

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

Python sample codes for robotics algorithms.

Python package built to ease deep learning on graph, on top of existing DL frameworks.

🐍 Geometric Computer Vision Library for Spatial AI

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

Turn your two-bit doodles into fine artworks with deep neural networks, generate seamless textures from photos, transfer style from one image to another, perform example-based upscaling, but wait..…

A collaboration friendly studio for NeRFs

Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image …

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

Object detection, 3D detection, and pose estimation using center point detection:

A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.

Neural style in TensorFlow! 🎨

Code for robust monocular depth estimation described in "Ranftl et. al., Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfer, TPAMI 2022"

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

A fork of OpenAI Baselines, implementations of reinforcement learning algorithms

Model summary in PyTorch similar to `model.summary()` in Keras

Depth Pro: Sharp Monocular Metric Depth in Less Than a Second.

An all-in-one Docker image for deep learning. Contains all the popular DL frameworks (TensorFlow, Theano, Torch, Caffe, etc.)

SuperGlue: Learning Feature Matching with Graph Neural Networks (CVPR 2020, Oral)

Semantic Segmentation Architectures Implemented in PyTorch

The project is an official implement of our ECCV2018 paper "Simple Baselines for Human Pose Estimation and Tracking(https://arxiv.org/abs/1804.06208)"

3D plotting and mesh analysis through a streamlined interface for the Visualization Toolkit (VTK)

On the Variance of the Adaptive Learning Rate and Beyond

Code for "NeuralRecon: Real-Time Coherent 3D Reconstruction from Monocular Video", CVPR 2021 oral

Dense Prediction Transformers

A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility