标签：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

单卡也能压榨出高吞吐的 LLM 推理引擎，PagedAttention 基本成了行业标准

🧑‍🏫 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.

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

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

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

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.

🤗 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

OpenMMLab Detection Toolbox and Benchmark

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.

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

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

The fastai deep learning library

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.

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.

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

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

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

Nano vLLM

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

The Cloud Native AI Platform

stable diffusion webui colab

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.