function: deep-learning

2653 products, primary matches first, then adoption-weighted; health v2 shown.

Product	Health v2	Stars	Maturity
meta-pytorch/tnt TNT (torchtnt) is a lightweight library from Meta providing tools and utilities for PyTorch training workflows. It offers building blocks f…	76	1721	active
One-2-3-45/One-2-3-45 One-2-3-45 is the official PyTorch implementation of a NeurIPS 2023 paper that converts any single image into a full 360-degree 3D textured…	29	1718	stable
VAST-AI-Research/UniRig UniRig is the official implementation of a SIGGRAPH 2025 framework that automatically rigs 3D models by predicting skeletons and skinning w…	58	1717	active
MaliosDark/wifi-3d-fusion WiFi-3D-Fusion is an open-source research project that estimates 3D human pose from WiFi CSI (Channel State Information) signals using deep…	27	1716	active
kingoflolz/mesh-transformer-jax A JAX/Haiku library implementing model-parallel training and inference of transformer models using xmap/pjit operators, similar to Megatron…	32	6380	maintenance
feizc/FluxMusic FluxMusic is the official PyTorch implementation of a rectified flow Transformer model for text-to-music generation, from the paper 'Flux t…	23	1711	active
zai-org/ImageReward ImageReward is the first general-purpose human preference reward model for text-to-image generation, trained on 137k expert comparison pair…	52	1702	stable
lsdefine/simple_GRPO A minimal (~200 lines) Python implementation of GRPO reinforcement learning for training LLMs to develop r1-like reasoning, built on PyTorc…	44	1702	active
PaddlePaddle/PaddleVideo PaddleVideo is a video understanding toolkit built on PaddlePaddle, offering state-of-the-art models for action recognition, temporal actio…	26	1702	active
sihyun-yu/REPA REPA is the official PyTorch implementation of the ICLR 2025 paper 'Representation Alignment for Generation', a regularization technique th…	27	1700	active
kyleskom/NBA-Machine-Learning-Sports-Betting A Python application that predicts NBA game winners and totals (over/under) using XGBoost and neural network models trained on historical t…	66	1696	active
BindsNET/bindsnet BindsNET is a Python package for simulating spiking neural networks (SNNs) built on PyTorch tensor functionality, running on CPUs or GPUs. …	84	1695	active
tensorpack/tensorpack Tensorpack is a high-level neural network training interface built on graph-mode TensorFlow, focused on training speed and flexibility for …	23	6286	maintenance
lucidrains/alphafold3-pytorch A PyTorch implementation of Google DeepMind's Alphafold 3 model for protein and molecular structure prediction. It is a research-oriented r…	68	1691	active
elixir-nx/axon Axon is a neural network library for Elixir built on top of Nx numerical definitions. It provides a functional API, a high-level model crea…	79	1688	active
williamyang1991/DualStyleGAN Official PyTorch implementation of DualStyleGAN, a CVPR 2022 model for exemplar-based high-resolution (1024px) portrait style transfer. It …	32	1683	stable
scverse/scvi-tools scvi-tools is a Python library for deep probabilistic modeling and analysis of single-cell and spatial omics data, built on PyTorch, PyTorc…	93	1681	stable
birdnet-team/BirdNET-Analyzer BirdNET-Analyzer is an open-source Python tool from the Cornell Lab of Ornithology that uses deep learning to identify bird species from au…	80	1681	active
intelligent-machine-learning/dlrover DLRover is an automatic distributed deep learning system that manages training of large AI models on Kubernetes and Ray clusters. It provid…	79	1680	active
takuseno/d3rlpy d3rlpy is a Python library for offline and online deep reinforcement learning built on PyTorch, offering state-of-the-art algorithms throug…	52	1679	active
pytorch-tabular/pytorch_tabular PyTorch Tabular is a Python library providing a unified, high-level interface for deep learning models on tabular data, built on PyTorch an…	67	1677	active
franciszzj/Leffa Leffa is a diffusion-based framework for controllable person image generation, supporting virtual try-on and pose transfer via a regulariza…	40	1672	active
shubham-goel/4D-Humans 4DHumans is a Python research codebase implementing HMR 2.0, a transformer-based model for 3D human mesh recovery from single images, plus …	59	1671	active
jd-opensource/JoyAI-Video-Edit JoyAI-Video-Edit is a real-time, instruction-guided video editing system that applies natural-language edits to live or uploaded video stre…	57	1671	active
zihangdai/xlnet XLNet is a generalized autoregressive pretraining method for language representation learning, built on Transformer-XL. This repo provides …	32	6188	maintenance
tkarras/progressive_growing_of_gans Official TensorFlow implementation of the ICLR 2018 NVIDIA paper 'Progressive Growing of GANs', which trains generators and discriminators …	32	6179	maintenance
Gen-Verse/MMaDA MMaDA is an open-source family of multimodal large diffusion language models that unify textual reasoning, multimodal understanding, and te…	49	1668	active
nwojke/deep_sort A Python implementation of Deep SORT, a multi-object tracking algorithm that extends SORT with a deep appearance descriptor for robust data…	36	6169	maintenance
bytedance/Sa2VA Sa2VA is a family of research models and codebases from ByteDance that combine SAM-2 with multimodal LLMs for pixel-level grounded understa…	70	1666	active
JiuhaiChen/BLIP3o Official implementation of the BLIP3o-Series, a unified autoregressive-plus-diffusion model for text-to-image generation and editing. It co…	44	1666	active
mit-han-lab/torchquantum TorchQuantum is a PyTorch-based framework for quantum computing simulation, supporting statevector and pulse-level simulation on GPUs with …	64	1662	active
thunil/TecoGAN TecoGAN is the official source code for a temporally coherent GAN for video super-resolution, published at SIGGRAPH/ACM TOG. It includes in…	32	6141	maintenance
thtrieu/darkflow Darkflow is a Python library that translates Darknet's YOLO neural network definitions to TensorFlow, enabling real-time object detection a…	32	6139	maintenance
Data-Centric-AI-Community/fg-data-synthetic A Python library for generating synthetic tabular and time-series data using state-of-the-art generative models such as GANs (including CTG…	72	1654	active
ZJU4HealthCare/HealthGPT HealthGPT is a medical multimodal large language model family unifying medical image comprehension and generation via heterogeneous knowled…	63	1654	active
mhamilton723/FeatUp FeatUp is a model-agnostic framework that upsamples the spatial resolution of deep neural network features by 16-32x without changing their…	16	1654	active
taki0112/UGATIT Official TensorFlow implementation of U-GAT-IT, an unsupervised image-to-image translation model using attention modules and adaptive layer…	32	6116	maintenance
maziarraissi/PINNs The original reference implementation of Physics-Informed Neural Networks (PINNs), which train neural networks to solve and discover nonlin…	62	6111	maintenance
lightly-ai/lightly-train LightlyTrain is a Python framework for training computer vision models, covering pretraining of vision foundation models (DINOv2/v3) on unl…	85	1652	active
sunlabuiuc/PyHealth PyHealth is an open-source Python toolkit for clinical deep learning, unifying healthcare datasets (EHRs, physiological signals, medical im…	91	1650	active
bytedance/DreamO DreamO is the official PyTorch implementation of a unified framework for image customization, built on FLUX diffusion models. It supports t…	35	1649	active
XueZeyue/DanceGRPO Official implementation of DanceGRPO, a framework applying Group Relative Policy Optimization (GRPO) to fine-tune visual generation models …	40	1648	active
JIA-Lab-research/ControlNeXt ControlNeXt is the official implementation of a controllable generation method for images and videos, built on Stable Diffusion XL, Stable …	24	1646	active
tenstorrent/tt-metal TT-Metal is Tenstorrent's open-source software stack containing TT-NN, a Python and C++ neural network operator library, and TT-Metalium, a…	97	1640	active
CoinCheung/BiSeNet A PyTorch implementation of the BiSeNet V1 and V2 real-time semantic segmentation models, with pretrained weights for Cityscapes, COCO-Stuf…	57	1638	active
yenchenlin/nerf-pytorch A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the original paper's results for synthesizing novel views of comp…	32	6044	maintenance
luanfujun/deep-painterly-harmonization Research code implementing the 'Deep Painterly Harmonization' algorithm, which seamlessly blends a pasted object into a painting's style us…	32	6042	maintenance
Picsart-AI-Research/StreamingT2V StreamingT2V is a research codebase (CVPR 2025) implementing an autoregressive technique that turns short text-to-video diffusion models li…	31	1630	active
Lucchetto/SuperImage SuperImage is an Android application that upscales low-resolution images using a Real-ESRGAN neural network running on-device via the MNN d…	10	1630	active
facebookresearch/DomainBed DomainBed is a PyTorch suite containing benchmark datasets and algorithms for domain generalization research. It implements many algorithms…	10	1630	active
yoshitomo-matsubara/torchdistill torchdistill is a modular, configuration-driven PyTorch framework for knowledge distillation and general deep learning experiments, requiri…	86	1629	active
Fantasy-AMAP/fantasy-talking FantasyTalking is a research codebase and model for generating realistic talking portrait videos from a single image and an audio clip, bui…	48	1628	active
InterDigitalInc/CompressAI CompressAI is a PyTorch library and evaluation platform for end-to-end learned data compression research. It provides custom layers, entrop…	73	1627	active
apple/ml-mobileclip Official implementation of Apple's MobileCLIP and MobileCLIP2, fast image-text contrastive (CLIP-style) models optimized for mobile latency…	60	1627	active
ZiqiaoPeng/SyncTalk SyncTalk is the official PyTorch implementation of a CVPR 2024 paper that synthesizes speech-driven, synchronized talking head videos using…	46	1626	active
guochengqian/Magic123 Magic123 is the official PyTorch implementation of an ICLR 2024 paper that generates high-quality textured 3D meshes from a single unposed …	90	1623	stable
gomlx/gomlx GoMLX is an accelerated machine learning and math framework for Go, comparable to PyTorch/JAX/TensorFlow. It offers differentiable operator…	95	1621	active
bowang-lab/scGPT scGPT is a foundation model for single-cell multi-omics built with a generative transformer, providing pretrained checkpoints and a Python …	67	1620	active
PaddlePaddle/PaddleSlim PaddleSlim is an open-source library built on PaddlePaddle for deep learning model compression and architecture search. It provides low-bit…	50	1612	active
open-gigaai/giga-world-0 GigaWorld-0 is a unified world model framework that acts as a data engine for Vision-Language-Action (VLA) learning in embodied AI. It comb…	41	1612	active
PKU-Alignment/safe-rlhf Beaver is a modular open-source framework from Peking University for training large language models with SFT, RLHF, and Safe RLHF (constrai…	53	1611	active
gorgonia/gorgonia Gorgonia is a Go library for machine learning that lets you define and evaluate mathematical equations over multidimensional arrays using a…	23	5929	maintenance
Robbyant/lingbot-depth LingBot-Depth is a PyTorch-based model and toolkit for masked depth modeling that transforms incomplete, noisy depth sensor data into metri…	56	1610	active
chainer/chainer Chainer is a Python-based deep learning framework that pioneered the define-by-run approach with dynamic computational graphs and automatic…	23	5924	maintenance
dmlc/gluon-cv GluonCV is a deep learning toolkit providing state-of-the-art computer vision model implementations with 170+ pre-trained models. It suppor…	23	5916	maintenance
meituan/YOLOv6 YOLOv6 is a single-stage object detection framework implemented in PyTorch, designed for industrial applications with a family of pretraine…	23	5895	maintenance
ml4a/ml4a ml4a is a Python library and collection of Jupyter notebooks for making art with machine learning. It wraps popular deep learning models li…	32	1602	active
time-series-foundation-models/lag-llama Lag-Llama is the first open-source foundation model for probabilistic time series forecasting, built on a transformer architecture. It prov…	37	1601	active
bilibili/ailab Bilibili's AI lab repository, best known for Real-CUGAN, a deep learning model for anime image super-resolution (upscaling). It provides pr…	23	5881	maintenance
autonomousvision/transfuser Official PyTorch implementation of TransFuser, a transformer-based multi-modal sensor fusion model for end-to-end autonomous driving, publi…	53	1599	stable
X-LANCE/AniTalker AniTalker is the official PyTorch implementation of an ACM MM 2024 paper that animates a single static portrait into a vivid talking-face v…	24	1598	active
SonyResearch/micro_diffusion Official implementation of Sony Research's micro-budget approach to training large-scale text-to-image diffusion transformer models from sc…	24	1594	active
UbiquitousLearning/mllm MLLM is a fast, lightweight multimodal LLM inference engine written in C++ for mobile and edge devices, with backends for ARM CPU, Qualcomm…	73	1593	active
google/uncertainty-baselines A library of high-quality, minimal-dependency implementations of standard and state-of-the-art uncertainty and robustness methods for deep …	77	1592	active
alibaba/Pai-Megatron-Patch Pai-Megatron-Patch is an open-source deep learning training toolkit from Alibaba Cloud for large-scale training and inference of LLMs and V…	56	1591	active
Tencent-Hunyuan/HunyuanWorld-Voyager HunyuanWorld-Voyager is a video diffusion framework from Tencent Hunyuan that generates world-consistent RGBD video and 3D point-cloud sequ…	52	1590	active
microsoft/Semi-supervised-learning USB (Unified Semi-supervised learning Benchmark) is a PyTorch-based codebase from Microsoft for semi-supervised learning across computer vi…	65	1588	active
MoonshotAI/Kimi-Linear Kimi Linear is a hybrid linear attention architecture (Kimi Delta Attention, based on Gated DeltaNet) released by Moonshot AI with 48B-para…	40	1588	active
FoundationVision/Infinity Infinity is a bitwise autoregressive text-to-image generation model (CVPR 2025 Oral) with released training and inference code, checkpoints…	56	1587	active
Drexubery/ViewCrafter ViewCrafter is a research codebase that uses video diffusion models to synthesize high-fidelity novel views of scenes from a single or spar…	49	1587	active
zju3dv/EasyVolcap EasyVolcap is a PyTorch-based library for accelerating neural volumetric video research, covering volumetric video capture, reconstruction,…	27	1587	active
DeepGraphLearning/torchdrug TorchDrug is a PyTorch-based machine learning library for drug discovery, covering graph neural networks, deep generative models, and reinf…	23	1587	active
Tencent-Hunyuan/HY-WorldPlay HY-WorldPlay (HY-World 1.5) is Tencent Hunyuan's open-source framework for interactive 3D world modeling, generating explorable 3D scenes f…	55	1586	active
pytorch/FBGEMM FBGEMM is a collection of highly optimized low-precision matrix multiplication and convolution kernels for server-side deep learning infere…	93	1584	active
XPixelGroup/HAT HAT (Hybrid Attention Transformer) is a PyTorch implementation of a state-of-the-art transformer model for image super-resolution and resto…	32	1583	stable
microsoft/MMdnn MMdnn is a Microsoft toolkit for converting, visualizing, and diagnosing deep learning models across frameworks such as TensorFlow, PyTorch…	39	5805	maintenance
meta-pytorch/torchtune Torchtune is a PyTorch-native library for authoring, post-training, and experimenting with large language models. It provides hackable trai…	69	5801	maintenance
AlibabaResearch/DAMO-ConvAI The official codebase for Alibaba DAMO Academy's Conversational AI research, containing implementations of models from their published pape…	71	1580	active
SalesforceAIResearch/uni2ts Uni2TS is a PyTorch library for unified pre-training, fine-tuning, inference, and evaluation of universal time series forecasting transform…	60	1580	active
RWKV/rwkv.cpp A C++ port of the RWKV language model to the ggml tensor library, providing FP32, FP16, and quantized INT4/INT5/INT8 inference focused on C…	37	1579	active
tensorflow/model-optimization The TensorFlow Model Optimization Toolkit (tfmot) is a Python library providing tools to optimize machine learning models for deployment, i…	82	1578	stable
DiffEqML/torchdyn Torchdyn is a PyTorch library dedicated to numerical deep learning, providing tools for neural differential equations, implicit models, and…	23	1578	active
NVlabs/sionna Sionna is an open-source, GPU-accelerated, differentiable Python library from NVIDIA for research on communication systems. It comprises Si…	83	1577	active
BloodAxe/pytorch-toolbelt A Python library of PyTorch extensions providing building blocks for fast R&D prototyping, including encoder-decoder architectures, special…	44	1574	active
Tencent/DepthCrafter DepthCrafter is a diffusion-based video depth estimation model from Tencent AI Lab that generates temporally consistent long depth sequence…	38	1574	active
gcorso/DiffDock DiffDock is a deep learning implementation of a diffusion generative model for molecular docking, predicting how small molecule ligands bin…	31	1569	active
Xilinx/brevitas Brevitas is a PyTorch library for neural network quantization supporting both post-training quantization (PTQ) and quantization-aware train…	91	1567	active
Chinese-Text-Classification-Pytorch A PyTorch-based collection of ready-to-run Chinese text classification models including TextCNN, TextRNN, FastText, TextRCNN, BiLSTM with a…	32	5728	maintenance
sdv-dev/CTGAN CTGAN is a Python library of deep learning based synthetic data generators for single-table tabular data, implementing the CTGAN conditiona…	80	1562	active
evo-design/evo Evo is a 7-billion-parameter DNA foundation model built on the StripedHyena architecture, trained on ~300 billion tokens of prokaryotic who…	71	1560	active