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.

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.

🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), ga…

High-Resolution Image Synthesis with Latent Diffusion Models

Official Code for DragGAN (SIGGRAPH 2023)

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Let us control diffusion models!

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

OpenMMLab Detection Toolbox and Benchmark

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

Generative Models by Stability AI

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

Open-Sora: Democratizing Efficient Video Production for All

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

[NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

Official inference repo for FLUX.1 models

Rembg is a tool to remove images background

WebUI extension for ControlNet

🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"

Mamba SSM architecture

Generate 3D objects conditioned on text or images

A paper list of object detection using deep learning.

InstantID: Zero-shot Identity-Preserving Generation in Seconds 🔥

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.