PyTorch

yolactdbolya/yolact

A fully convolutional neural network for real-time instance segmentation, achieving high speed and accuracy on COCO.

GitHub repositoryNVIDIAGameWorks/kaolin

A PyTorch library providing GPU-accelerated tools for 3D deep learning, including differentiable rendering and geometric operations.

ignitepytorch/ignite

A high-level library for training and evaluating neural networks in PyTorch with a flexible engine and event system.

pysotSTVIR/pysot

A PyTorch-based research platform implementing state-of-the-art single object tracking algorithms like SiamRPN and SiamMask.

RecBoleRUCAIBox/RecBole

A unified, comprehensive, and efficient Python/PyTorch library for reproducing and developing recommendation algorithms.

BioGPTmicrosoft/BioGPT

A domain-specific generative language model pre-trained on biomedical literature for text generation and mining tasks.

Deep High-Resolution-Netleoxiaobin/deep-high-resolution-net.pytorch

Official PyTorch implementation of HRNet for human pose estimation, maintaining high-resolution representations through parallel multi-scale fusions.

DrQAfacebookresearch/DrQA

A PyTorch system for open-domain question answering by retrieving and reading documents, originally applied to Wikipedia.

Hugging Face Diffusion Models Coursehuggingface/diffusion-models-class

A free course teaching diffusion models theory and hands-on implementation using Hugging Face's Diffusers library.

hloccvg/Hierarchical-Localization

A modular Python toolbox for state-of-the-art 6-DoF visual localization using hierarchical image retrieval and feature matching.

Evolutionary Scale Modeling (ESM)facebookresearch/esm

A collection of transformer protein language models for predicting structure, function, and designing proteins from sequences.

GPyTorchcornellius-gp/gpytorch

A highly efficient, scalable Gaussian process library implemented in PyTorch with GPU acceleration and modular design.

lightly - A computer vision framework for self-supervised learninglightly-ai/lightly

A Python library for self-supervised learning on images, providing a modular PyTorch-like framework with support for modern SSL models.

Horizonfacebookresearch/Horizon

An end-to-end platform for applied reinforcement learning and contextual bandits, built with PyTorch for production decision-making systems.

ReAgentfacebookresearch/ReAgent

An end-to-end platform for applied reinforcement learning and contextual bandits, originally developed at Facebook for production recommendation systems.

DI-engineopendilab/DI-engine

A comprehensive Python-first reinforcement learning framework with modular abstractions for decision intelligence applications.

BoTorchpytorch/botorch

A modular library for Bayesian optimization built on PyTorch, enabling efficient optimization of expensive black-box functions.

ML Workspaceml-tooling/ml-workspace

A web-based IDE for machine learning and data science with pre-installed libraries and tools, deployable via Docker.

pomegranatejmschrei/pomegranate

A Python library for probabilistic modeling built on PyTorch, offering modular distributions, GPU support, and flexible model composition.

pytrackingvisionml/pytracking

A PyTorch-based framework for visual object tracking and video object segmentation, featuring implementations of state-of-the-art trackers like TaMOs, RTS, and DiMP.

Ethen's Notebook Collectionethen8181/machine-learning

A comprehensive collection of machine learning tutorials and implementations in Python, covering algorithms from scratch to production deployment.

TensorWatchmicrosoft/tensorwatch

A debugging and visualization tool for data science, deep learning, and reinforcement learning in Jupyter Notebook.

SRGANtensorlayer/srgan

Implementation of SRGAN for photo-realistic single image super-resolution using generative adversarial networks.

Colorizationrichzhang/colorization

Automatic and interactive image colorization using deep neural networks, with PyTorch models for ECCV 2016 and SIGGRAPH 2017 papers.

PyTorch-BigGraphfacebookresearch/PyTorch-BigGraph

A distributed system for learning graph embeddings from large-scale graphs with billions of entities and trillions of edges.

OpenFoldaqlaboratory/openfold

A trainable, memory-efficient PyTorch reproduction of AlphaFold 2 for protein structure prediction.

pyslamluigifreda/pyslam

A hybrid Python/C++ Visual SLAM pipeline supporting monocular, stereo, and RGB-D cameras with modern features, loop closure, and dense reconstruction.

Catalyst: High-level utils for PyTorch DL & RL research. It was developed with a focus on reproducibility, fast experimentation and code/ideas reusingcatalyst-team/catalyst

A PyTorch framework for deep learning research and development, focusing on reproducibility and rapid experimentation.

Ask-AnythingOpenGVLab/Ask-Anything

An open-source framework for building multimodal AI systems that enable large language models to understand and chat about videos and images.

flownet2-pytorchNVIDIA/flownet2-pytorch

PyTorch implementation of FlowNet 2.0 for optical flow estimation using deep neural networks.

TTS WebUIrsxdalv/TTS-WebUI

A unified web interface for text-to-speech, voice cloning, and audio generation with support for dozens of AI models.

RFdiffusionRosettaCommons/RFdiffusion

An open-source diffusion model for generating and designing protein structures, including binders, symmetric oligomers, and motif-scaffolded proteins.

pytorch_geometric_temporalbenedekrozemberczki/pytorch_geometric_temporal

A PyTorch library for spatiotemporal signal processing with dynamic and temporal graph neural networks.

ESM3evolutionaryscale/esm

A multimodal protein language model for generative protein design and engineering by jointly reasoning over sequence, structure, and function.

TorchAudiopytorch/audio

An audio library for PyTorch providing data manipulation, transformations, and dataset loaders for machine learning applications.

computer-vision-in-actionCharmve/computer-vision-in-action

An interactive online learning platform for computer vision with a comprehensive Chinese ebook, code, and community.