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

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Models and examples built with TensorFlow

Inference code for Llama models

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

The simplest, fastest repository for training/finetuning medium-sized GPTs.

CLI platform to experiment with codegen. Precursor to: https://lovable.dev

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

TensorFlow code and pre-trained models for BERT

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

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

OpenMMLab Detection Toolbox and Benchmark

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

The official Meta Llama 3 GitHub site

State-of-the-art 2D and 3D Face Analysis Project

Open standard for machine learning interoperability

gpt-oss-120b and gpt-oss-20b are two open-weight language models by OpenAI

An open-source, code-first Python toolkit for building, evaluating, and deploying sophisticated AI agents with flexibility and control.

Open source code for AlphaFold 2.

The pytest framework makes it easy to write small tests, yet scales to support complex functional testing

A PyTorch-based Speech Toolkit

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

A PyTorch implementation of EfficientNet

Minimal PyTorch implementation of YOLOv3

Code release for ConvNeXt model

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility