pytorch

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

Stable Diffusion web UI

The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface. The fastest local inference engine in the world.

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

A high-throughput and memory-efficient inference and serving engine for LLMs

🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠

Deep Learning for humans

Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

Clone a voice in 5 seconds to generate arbitrary speech in real-time

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

We write your reusable computer vision tools. 💜

Learn how to develop, deploy and iterate on production-grade ML applications.

🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

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

VoxCPM2: Tokenizer-Free TTS for Multilingual Speech Generation, Creative Voice Design, and True-to-Life Cloning

GFPGAN aims at developing Practical Algorithms for Real-world Face Restoration.

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.

🚀Clone a voice in 5 seconds to generate arbitrary speech in real-time

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

Visualizer for neural network, deep learning and machine learning models

Non-autoregressive System 1 decision engine. Typed choice, score and yes/no decisions over any text in a single forward pass, in 100+ languages, with a router that picks the right checkpoint per request.

OpenMMLab Detection Toolbox and Benchmark

PyTorch Tutorial for Deep Learning Researchers

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

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

Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.

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

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

The fastai deep learning library

SoftVC VITS Singing Voice Conversion

GUI for a Vocal Remover that uses Deep Neural Networks.

deep learning for image processing including classification and object-detection etc.

JARVIS, a system to connect LLMs with ML community. Paper: https://arxiv.org/pdf/2303.17580.pdf

Image-to-Image Translation in PyTorch

Graph Neural Network Library for PyTorch

ncnn is a high-performance neural network inference framework optimized for the mobile platform

🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

Image inpainting tool powered by SOTA AI Model. Remove any unwanted object, defect, people from your pictures or erase and replace(powered by stable diffusion) any thing on your pictures.

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

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

pytorch handbook是一本开源的书籍，目标是帮助那些希望和使用PyTorch进行深度学习开发和研究的朋友快速入门，其中包含的Pytorch教程全部通过测试保证可以成功运行

Open standard for machine learning interoperability

FinGPT: Open-Source Financial Large Language Models! Revolutionize 🔥 We release the trained model on HuggingFace.

Open-source speech recognition toolkit for training, inference, streaming ASR, VAD, punctuation, speaker diarization pipelines, and OpenAI-compatible/MCP serving.

Machine Learning Engineering Open Book

本项目将《动手学深度学习》(Dive into Deep Learning)原书中的MXNet实现改为PyTorch实现。

Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.

Welcome to the Llama Cookbook! This is your go to guide for Building with Llama: Getting started with Inference, Fine-Tuning, RAG. We also show you how to solve end to end problems using Llama model family and using them on various provider services

Private AI platform for agents, assistants and enterprise search. Built-in Agent Builder, Deep research, Document analysis, Multi-model support, and API connectivity for agents.

[NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer

深度学习入门教程, 优秀文章, Deep Learning Tutorial

A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.

pix2tex: Using a ViT to convert images of equations into LaTeX code.

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

Nano vLLM

The Cloud Native AI Platform

stable diffusion webui colab

Bringing Old Photo Back to Life (CVPR 2020 oral)

Natural Language Processing Tutorial for Deep Learning Researchers

A theoretical reconstruction of the Claude Mythos architecture, built from first principles using the available research literature.

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT LLM also contains components to create Python and C++ runtimes that orchestrate the inference execution in a performant way.

Learn ML engineering for free in 4 months! Register here 👇🏼

RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RNN and transformer - great performance, linear time, constant space (no kv-cache), fast training, infinite ctx_len, and free sentence embedding.

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

A very simple framework for state-of-the-art Natural Language Processing (NLP)

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

An open source implementation of CLIP.

Convert Machine Learning Code Between Frameworks

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

Open3D: A Modern Library for 3D Data Processing

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

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

人工智能学习路线图，整理近200个实战案例与项目，免费提供配套教材，零基础入门，就业实战！包括：Python，数学，机器学习，数据分析，深度学习，计算机视觉，自然语言处理，PyTorch tensorflow machine-learning,deep-learning data-analysis data-mining mathematics data-science artificial-intelligence python tensorflow tensorflow2 caffe keras pytorch algorithm numpy pandas matplotlib seaborn nlp cv等热门领域

深度学习入门开源书，基于TensorFlow 2.0案例实战。Open source Deep Learning book, based on TensorFlow 2.0 framework.

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

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)

🏄 Scalable embedding, reasoning, ranking for images and sentences with CLIP

The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.

超轻量级中文ocr，支持竖排文字识别, 支持ncnn、mnn、tnn推理 ( dbnet(1.8M) + crnn(2.5M) + anglenet(378KB)) 总模型仅4.7M

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

LMCache: Supercharge Your LLM with the Fastest KV Cache Layer

A collaboration friendly studio for NeRFs

A PyTorch-based Speech Toolkit

An open-source NLP research library, built on PyTorch.

Python 开源项目之「自学编程之路」，保姆级教程：AI实验室、宝藏视频、数据结构、学习指南、机器学习实战、深度学习实战、网络爬虫、大厂面经、程序人生、资源分享。

Low-code framework for building custom LLMs, neural networks, and other AI models

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

This repository contains demos I made with the Transformers library by HuggingFace.

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

🐍 Geometric Computer Vision Library for Spatial AI

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

Real-time, local speech-to-text with streaming ASR, speaker diarization, translation, and OpenAI/Deepgram-compatible APIs.

An elegant PyTorch deep reinforcement learning library.

Large Language Model Text Generation Inference

Fast and Accurate ML in 3 Lines of Code

Neural building blocks for speaker diarization: speech activity detection, speaker change detection, overlapped speech detection, speaker embedding