All Algorithms implemented in Python

🤗 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.

Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)

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

Unsloth Studio is a web UI for training and running open models like Qwen, DeepSeek, gpt-oss and Gemma locally.

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

⏬ Dumb downloader that scrapes the web

Rich is a Python library for rich text and beautiful formatting in the terminal.

AiLearning：数据分析+机器学习实战+线性代数+PyTorch+NLTK+TF2

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.

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

Machine Learning From Scratch. Bare bones NumPy implementations of machine learning models and algorithms with a focus on accessibility. Aims to cover everything from linear regression to deep lear…

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

A Gemini 2.5 Flash Level MLLM for Vision, Speech, and Full-Duplex Multimodal Live Streaming on Your Phone

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

Graph Neural Network Library for PyTorch

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

Image annotation with Python. Supports polygon, rectangle, circle, line, point, and AI-assisted annotation.

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

Use PEFT or Full-parameter to CPT/SFT/DPO/GRPO 600+ LLMs (Qwen3.5, DeepSeek-R1, GLM-5, InternLM3, Llama4, ...) and 300+ MLLMs (Qwen3-VL, Qwen3-Omni, InternVL3.5, Ovis2.5, GLM4.5v, Llava, Phi4, ...)…

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

[CVPR 2025 Best Paper Award] VGGT: Visual Geometry Grounded Transformer

This project aim to reproduce Sora (Open AI T2V model), we wish the open source community contribute to this project.

Enjoy the magic of Diffusion models!

Official repo for paper "Structured 3D Latents for Scalable and Versatile 3D Generation" (CVPR'25 Spotlight).

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.

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

Minimal and clean examples of machine learning algorithms implementations