PyTorch

HuggingFace Transformershuggingface/transformers

A model-definition framework for state-of-the-art machine learning models across text, vision, audio, and multimodal tasks.

JAX, Flax & Transformershuggingface/transformers

A model-definition framework for state-of-the-art machine learning models across text, vision, audio, and multimodal tasks.

funnlpfighting41love/funNLP

A comprehensive collection of Chinese NLP resources, datasets, tools, and pre-trained models for developers and researchers.

Stable DiffusionCompVis/stable-diffusion

A latent text-to-image diffusion model that generates detailed images from text prompts, running on GPUs with at least 10GB VRAM.

annotated_deep_learning_paper_implementationslabmlai/annotated_deep_learning_paper_implementations

A collection of 60+ annotated PyTorch implementations of deep learning papers with side-by-side explanatory notes.

unslothunslothai/unsloth

A web UI and optimization library for running and fine-tuning open-source AI models locally with 2x faster training and 70% less VRAM.

YOLOv8ultralytics/ultralytics

A cutting-edge framework for training and deploying state-of-the-art YOLO models for object detection, segmentation, classification, and pose estimation.

Yolov5ultralytics/yolov5

A state-of-the-art PyTorch-based computer vision model for object detection, segmentation, and classification.

TTScoqui-ai/TTS

A deep learning toolkit for Text-to-Speech generation with pretrained models in over 1100 languages and tools for training.

Colossal-AI - An Integrated Large-scale Model Training System with Efficient Parallelization Techniqueshpcaitech/ColossalAI

A unified deep learning system for efficient large-scale model training and inference with advanced parallelism strategies.

FastChatlm-sys/FastChat

An open platform for training, serving, and evaluating large language model based chatbots.

Barksuno-ai/bark

A transformer-based text-to-audio model that generates realistic multilingual speech, music, and sound effects.

timmrwightman/pytorch-image-models

A comprehensive collection of PyTorch image models, layers, utilities, and training scripts for computer vision research and applications.

DragGANXingangPan/DragGAN

Interactive point-based manipulation tool for editing GAN-generated images by dragging points to target positions.

detectron2facebookresearch/detectron2

A PyTorch-based platform for state-of-the-art object detection, segmentation, and visual recognition tasks.

diffusershuggingface/diffusers

A modular PyTorch library for state-of-the-art diffusion models to generate images, audio, and 3D molecular structures.

spacyexplosion/spaCy

Industrial-strength Natural Language Processing library for Python, featuring pretrained pipelines for 70+ languages and production-ready training.

Pytorch Tutorial by Yunjey Choiyunjey/pytorch-tutorial

A collection of concise PyTorch tutorials for deep learning researchers, with most models implemented in under 30 lines of code.

pytorch-lightningLightning-AI/lightning

A deep learning framework to pretrain and finetune any AI model at any scale with zero code changes.

pytorch-lightningLightning-AI/pytorch-lightning

A deep learning framework to pretrain and finetune any AI model on any hardware with zero code changes.

PyTorch LightningPyTorchLightning/pytorch-lightning

A PyTorch wrapper that automates engineering boilerplate for scalable AI model training and deployment.

EasyOCRJaidedAI/EasyOCR

A ready-to-use OCR Python library supporting 80+ languages and popular writing scripts like Latin, Chinese, Arabic, and Cyrillic.

sglangsgl-project/sglang

A high-performance serving framework for large language models and multimodal models, delivering low-latency and high-throughput inference.

Jupyter Notebooksfastai/fastai

A deep learning library built on PyTorch that provides high-level components for rapid results and low-level components for research flexibility.

fastaifastai/fastai

A deep learning library built on PyTorch that provides high-level components for rapid results and low-level components for research flexibility.

MLXml-explore/mlx

An array framework for machine learning on Apple silicon with unified memory and dynamic graph construction.

Fluxblack-forest-labs/flux

Minimal inference code for running FLUX.1 open-weight models for image generation and editing.

shapslundberg/shap

A unified Python library for explaining any machine learning model's predictions using Shapley values from game theory.

fastbookfastai/fastbook

The fastai book, published as Jupyter Notebooks, provides an introduction to deep learning, fastai, and PyTorch.

PyTorch Geometricpyg-team/pytorch_geometric

A PyTorch library for building and training Graph Neural Networks (GNNs) on structured and irregular data.

GitHub repositoryrusty1s/pytorch_geometric

A comprehensive library for building and training Graph Neural Networks (GNNs) with PyTorch.

deepfaceserengil/deepface

A lightweight Python library for face recognition and facial attribute analysis (age, gender, emotion, race) with a unified API.

WhisperXm-bain/whisperX

Fast automatic speech recognition with accurate word-level timestamps and speaker diarization, built on OpenAI's Whisper.

Microsoft RecommendersMicrosoft/Recommenders

A repository of examples, utilities, and best practices for building and deploying production-ready recommendation systems.

FinGPTAI4Finance-Foundation/FinGPT

An open-source framework for financial large language models, enabling cost-effective fine-tuning for tasks like sentiment analysis and forecasting.

Buzzchidiwilliams/Buzz

An offline desktop application for transcribing and translating audio/video files, live recordings, and YouTube links using OpenAI's Whisper.