domain: deep-learning
2771 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| yifan123/flow_grpo Flow-GRPO is the official PyTorch implementation of a NeurIPS 2025 paper that trains flow matching models (e.g., SD3.5, FLUX.1, Qwen-Image,… | 56 | 2498 | active |
| sthalles/SimCLR A PyTorch reference implementation of SimCLR, a self-supervised contrastive learning framework for learning visual representations from unl… | 23 | 2491 | stable |
| ppogg/YOLOv5-Lite YOLOv5-Lite is a lightweight object detection model family evolved from YOLOv5, with models as small as ~900KB (int8) that run 10-15+ FPS o… | 23 | 2487 | active |
| thuml/Autoformer Autoformer is the official PyTorch implementation of the NeurIPS 2021 paper 'Autoformer: Decomposition Transformers with Auto-Correlation f… | 36 | 2481 | stable |
| xuebinqin/U-2-Net Official PyTorch implementation of U^2-Net, a nested U-structure deep network for salient object detection, published in Pattern Recognitio… | 32 | 9853 | maintenance |
| alexjc/neural-doodle A Python implementation of Semantic Style Transfer (Champandard, 2016) based on the Neural Patches algorithm. It turns rough doodles into r… | 10 | 9852 | maintenance |
| pykeio/ort ort is a safe, fast Rust binding for Microsoft's ONNX Runtime, providing hardware-accelerated machine learning inference and training for O… | 89 | 2476 | active |
| GaParmar/img2img-turbo A research library implementing one-step image-to-image translation models (CycleGAN-Turbo and pix2pix-turbo) built on SD-Turbo diffusion m… | 41 | 2476 | active |
| jadore801120/attention-is-all-you-need-pytorch A PyTorch implementation of the Transformer model from the paper 'Attention Is All You Need', including training and translation scripts fo… | 32 | 9784 | maintenance |
| Lightning-AI/torchmetrics TorchMetrics is a PyTorch-native library of machine learning metrics that work in distributed and scalable training pipelines. It provides … | 90 | 2461 | stable |
| facebookresearch/pifuhd PIFuHD is a PyTorch implementation of a CVPR 2020 research model that reconstructs high-resolution 3D human body meshes from a single 2D im… | 10 | 9737 | maintenance |
| data-infra/cube-studio CubeStudio is an open-source, cloud-native, all-in-one AI platform covering the full machine learning lifecycle (MLOps/MaaS/LLMOps), includ… | 80 | 2448 | active |
| TorchIO-project/torchio TorchIO is a Python library for loading, augmenting, and processing 3D medical images (MRI, CT) within PyTorch deep learning pipelines. It … | 93 | 2439 | active |
| google/XNNPACK XNNPACK is a highly optimized C library of floating-point neural network inference operators for ARM, x86, WebAssembly, and RISC-V platform… | 77 | 2435 | active |
| shenweichen/DeepMatch DeepMatch is a Python library of deep matching models for recommendations and advertising, built on TensorFlow/Keras. It lets users train m… | 72 | 2433 | active |
| marin-community/marin Marin is an open-source Python framework and research program for training foundation models, covering the full pipeline from data curation… | 78 | 2428 | active |
| roboflow/inference Roboflow Inference is a Python library and self-hostable inference server for deploying computer vision models on any computer or edge devi… | 91 | 2427 | active |
| tflearn/tflearn TFLearn is a modular deep learning library providing a higher-level, Keras-like API on top of TensorFlow for building and training neural n… | 23 | 9576 | maintenance |
| wolny/pytorch-3dunet A PyTorch implementation of 3D U-Net and its variants (residual, squeeze-and-excitation) for volumetric semantic segmentation, with 2D U-Ne… | 63 | 2416 | active |
| HuCaoFighting/Swin-Unet Official PyTorch implementation of Swin-Unet, a U-shaped pure Transformer model for medical image segmentation, published at ECCV 2022 Medi… | 41 | 2416 | stable |
| google/tunix Tunix is a lightweight JAX-based library for post-training large language models, supporting supervised fine-tuning, preference optimizatio… | 83 | 2415 | active |
| AI-Hypercomputer/maxtext MaxText is a high-performance, scalable open-source LLM training library written in pure Python/JAX, targeting Google Cloud TPUs and GPUs. … | 97 | 2408 | active |
| resemble-ai/resemble-enhance Resemble Enhance is an AI-powered Python tool that improves speech quality through denoising and enhancement, using a denoiser module and a… | 17 | 2397 | active |
| Alibaba-Quark/LiveAvatar LiveAvatar is an open-source implementation of an ECCV 2026 paper for streaming, real-time, infinite-length audio-driven avatar video gener… | 61 | 2386 | active |
| ailia-ai/ailia-models A collection of 400+ pre-trained state-of-the-art AI models (object detection, pose estimation, speech recognition, LLMs, image generation,… | 77 | 2385 | active |
| google/neural-tangents Neural Tangents is a Python library built on JAX for defining, training, and evaluating neural networks of both finite and infinite width. … | 10 | 2383 | stable |
| microsoft/Olive Olive is Microsoft's AI model optimization toolkit for the ONNX Runtime, automating finetuning, conversion, quantization, and compression o… | 91 | 2382 | active |
| FMInference/FlexLLMGen FlexLLMGen is a high-throughput generation engine for running large language models on a single GPU with limited memory. It uses IO-efficie… | 10 | 9352 | maintenance |
| apple/axlearn AXLearn is a Python deep learning library built on JAX and XLA for developing and training large-scale models, with an object-oriented conf… | 71 | 2372 | active |
| OpenGVLab/InternVideo InternVideo is a series of open-source video foundation models for multimodal video understanding, spanning generative and discriminative l… | 72 | 2368 | active |
| jik876/hifi-gan The official PyTorch implementation of HiFi-GAN, a generative adversarial network that converts mel-spectrograms into high-fidelity 22.05 k… | 32 | 2367 | stable |
| tencent-ailab/V-Express V-Express is a Python research project from Tencent AI Lab that generates talking head portrait videos from a reference image, audio, and V… | 25 | 2360 | active |
| facebookresearch/perception_models Meta's Perception Models repository hosting state-of-the-art image, video, and audio encoders (Perception Encoder, PE) and a multimodal lan… | 54 | 2353 | active |
| mlech26l/ncps A Python package providing PyTorch and TensorFlow/Keras implementations of Neural Circuit Policies (NCPs), including liquid time-constant (… | 23 | 2345 | active |
| ufoym/imbalanced-dataset-sampler A PyTorch dataset sampler (ImbalancedDatasetSampler, pip package torchsampler) that rebalances class distributions during training by overs… | 64 | 2326 | stable |
| google-deepmind/optax Optax is a gradient processing and optimization library for JAX, offering composable building blocks like optimizers and loss functions. It… | 82 | 2325 | stable |
| labmlai/labml A Python library for tracking and monitoring deep learning experiments, with a self-hostable server app for viewing metrics and hardware us… | 30 | 2325 | active |
| facebookresearch/schedule_free A PyTorch library implementing schedule-free optimizers (SGD, AdamW, RAdam variants) that remove the need for learning rate schedules or sp… | 67 | 2323 | active |
| PKU-YuanGroup/MoE-LLaVA MoE-LLaVA is an open-source Mixture-of-Experts based sparse large vision-language model, released with the MoE-Tuning training strategy fro… | 31 | 2322 | active |
| iver56/audiomentations Audiomentations is a Python library for audio data augmentation with an API inspired by albumentations. It provides fast CPU-based waveform… | 67 | 2314 | stable |
| SkyworkAI/Matrix-Game Matrix-Game is Skywork AI's open-source series of interactive world foundation models that generate real-time, streaming video in response … | 52 | 2314 | active |
| facebookresearch/ImageBind A PyTorch library from Meta AI implementing ImageBind, a model that learns a joint embedding space across six modalities: images, text, aud… | 54 | 9064 | maintenance |
| IDEA-Research/detrex detrex is an open-source PyTorch-based research platform and toolbox for DETR-style Transformer detection algorithms, built on top of Detec… | 41 | 2306 | active |
| frgfm/torch-cam TorchCAM is a Python library that extracts class activation maps (CAMs) from PyTorch CNN classifiers, supporting many CAM variants such as … | 72 | 2304 | active |
| XiaomiMiMo/MiMo Xiaomi's MiMo is a 7B-parameter reasoning language model trained from pretraining through posttraining with reinforcement learning, release… | 30 | 2299 | active |
| traveller59/spconv SpConv is a spatially sparse convolution library for deep learning on 3D point clouds and sparse tensors, distributed as PyPI packages with… | 32 | 2291 | active |
| huggingface/picotron Picotron is a minimalist, hackable distributed training framework for pre-training Llama-like large language models using 4D parallelism (d… | 40 | 2289 | active |
| aixcoder-plugin/aiXcoder-7B Official repository for aiXcoder-7B, an open-weights 7B-parameter code large language model trained on 1.2T tokens for code completion, gen… | 29 | 2271 | active |
| aigc-apps/EasyAnimate EasyAnimate is an end-to-end Python pipeline for high-resolution, long video and image generation based on transformer diffusion (DiT) mode… | 20 | 2270 | active |
| 666DZY666/micronet micronet is a Python library for deep neural network model compression and deployment built on PyTorch. It provides quantization (QAT, PTQ,… | 41 | 2266 | active |
| ashawkey/stable-dreamfusion A PyTorch implementation of Dreamfusion that generates 3D models from text prompts or images using NeRF combined with Stable Diffusion guid… | 23 | 8854 | maintenance |
| hkchengrex/MMAudio MMAudio is a PyTorch-based model for generating synchronized audio from video and/or text inputs, using multimodal joint training across au… | 43 | 2264 | active |
| aws/sagemaker-python-sdk The SageMaker Python SDK is an open-source Python library for training and deploying machine learning models on Amazon SageMaker. It suppor… | 95 | 2261 | active |
| RosettaCommons/RoseTTAFold RoseTTAFold is the official implementation of a deep learning system for predicting protein structures and interactions using a three-track… | 23 | 2258 | stable |
| LAION-AI/CLAP CLAP (Contrastive Language-Audio Pretraining) is a Python library from LAION that learns joint audio and text embeddings using contrastive … | 42 | 2256 | stable |
| THU-MIG/yoloe YOLOE is the official PyTorch implementation of an open-vocabulary object detection and segmentation model presented at ICCV 2025. It unifi… | 32 | 2256 | active |
| fishaudio/Bert-VITS2 Bert-VITS2 is a text-to-speech model implementation combining the VITS2 architecture with multilingual BERT embeddings, written in Python. … | 63 | 8796 | maintenance |
| CoinCheung/pytorch-loss A PyTorch library providing a collection of loss functions (focal loss, triplet loss, AMSoftmax, label-smooth CE, dice loss, lovasz-softmax… | 32 | 2252 | active |
| facebookresearch/fvcore fvcore is a lightweight Python core library providing common functionality shared across FAIR's computer vision frameworks such as Detectro… | 76 | 2250 | stable |
| Alpha-VLLM/Lumina-T2X Lumina-T2X is a unified framework for text-to-any-modality generation built on flow-based large diffusion transformers. It supports generat… | 28 | 2250 | active |
| azavea/raster-vision Raster Vision is an open source Python library and low-code framework for building computer vision models on satellite, aerial, and other l… | 61 | 2240 | active |
| NVIDIA/vid2vid A PyTorch implementation of NVIDIA's video-to-video synthesis method for generating high-resolution (e.g., 2048x1024) photorealistic videos… | 32 | 8692 | maintenance |
| facebookresearch/DiT Official PyTorch implementation of Diffusion Transformers (DiT) from the paper 'Scalable Diffusion Models with Transformers', including mod… | 10 | 8689 | maintenance |
| NVlabs/MambaVision MambaVision is NVIDIA's official PyTorch implementation of a hybrid Mamba-Transformer vision backbone, published at CVPR 2025. It provides … | 49 | 2224 | active |
| ellisdg/3DUnetCNN A PyTorch library for building, training, and applying 3D U-Net convolutional neural networks for medical image segmentation. It provides c… | 45 | 2224 | active |
| dstackai/dstack dstack is an open-source, vendor-agnostic control plane for GPU provisioning and orchestration that works across GPU clouds, Kubernetes, an… | 95 | 2221 | active |
| ahmedfgad/GeneticAlgorithmPython PyGAD is an open-source Python 3 library for implementing the genetic algorithm to optimize single- and multi-objective problems. It can al… | 80 | 2220 | active |
| lifeiteng/vall-e An unofficial PyTorch implementation of VALL-E, a zero-shot text-to-speech model that treats TTS as a conditional language modeling task ov… | 40 | 2215 | active |
| aigc-apps/VideoX-Fun VideoX-Fun is a Python-based video generation pipeline built on Diffusion Transformer models (CogVideoX-Fun, Wan-Fun) that generates videos… | 67 | 2210 | active |
| lucidrains/lion-pytorch A PyTorch implementation of the Lion optimizer (Evolved Sign Momentum), discovered by Google Brain via genetic algorithms and claimed to ou… | 62 | 2199 | active |
| kubeflow/trainer Kubeflow Trainer is a Kubernetes-native platform for distributed AI model training and LLM fine-tuning across frameworks like PyTorch, JAX,… | 95 | 2198 | active |
| NX-AI/xlstm Official PyTorch implementation of xLSTM, an extended Long Short-Term Memory recurrent architecture with exponential gating and matrix memo… | 59 | 2198 | active |
| tensorflow/tfx TensorFlow Extended (TFX) is an end-to-end, Google-production-scale platform for building and deploying production machine learning pipelin… | 91 | 2190 | stable |
| tensorflow/quantum TensorFlow Quantum is a Python framework for hybrid quantum-classical machine learning that integrates Cirq quantum circuits with TensorFlo… | 84 | 2180 | active |
| Harry24k/adversarial-attacks-pytorch Torchattacks is a PyTorch library providing implementations of adversarial attacks to generate adversarial examples against deep learning m… | 23 | 2177 | active |
| ByteDance-Seed/VeOmni VeOmni is a PyTorch-native framework for single- and multi-modal model pre-training and post-training, with a modular, trainer-free design … | 82 | 2173 | active |
| MoonshotAI/MoBA MoBA (Mixture of Block Attention) is a PyTorch implementation of a trainable block-sparse attention mechanism for long-context large langua… | 27 | 2169 | active |
| facebookresearch/mae A PyTorch/GPU re-implementation of the Masked Autoencoders (MAE) paper for self-supervised vision learning. It includes pre-training code, … | 10 | 8370 | maintenance |
| galilai-group/stable-worldmodel A Python library providing a unified platform for reproducible world model research, covering data collection, training, and evaluation via… | 81 | 2156 | active |
| Tencent-Hunyuan/HunyuanVideo-Avatar HunyuanVideo-Avatar is Tencent's open-source model and inference code for high-fidelity audio-driven human animation, generating talking av… | 45 | 2156 | active |
| google/gin-config Gin-config is a lightweight Python configuration framework based on dependency injection. Functions or classes decorated with @gin.configur… | 63 | 2155 | stable |
| baowenbo/DAIN DAIN is a PyTorch-based deep learning model for depth-aware video frame interpolation, synthesizing intermediate frames between existing on… | 32 | 8316 | maintenance |
| google/trax Trax is an end-to-end deep learning library built on JAX and TensorFlow that focuses on clear code and speed, developed and maintained by t… | 10 | 8306 | maintenance |
| ViTAE-Transformer/ViTPose Official PyTorch implementation of ViTPose and ViTPose++, Vision Transformer models for human and generic body pose estimation from NeurIPS… | 59 | 2138 | stable |
| xinsir6/ControlNetPlus ControlNet++ is an all-in-one ControlNet model and architecture supporting 10+ control types for text-to-image generation and image editing… | 23 | 2138 | active |
| utkuozbulak/pytorch-cnn-visualizations A PyTorch library implementing a wide range of convolutional neural network visualization and interpretability techniques, including Grad-C… | 32 | 8233 | maintenance |
| lukemelas/EfficientNet-PyTorch A PyTorch implementation of the EfficientNet convolutional neural network family with pretrained ImageNet weights. It provides a simple pip… | 23 | 8222 | maintenance |
| cvxpy/cvxpylayers CVXPYlayers is a Python library for constructing differentiable convex optimization layers in PyTorch, JAX, and MLX using CVXPY. It solves … | 87 | 2126 | active |
| tairov/llama2.mojo A single-file implementation of Llama 2 transformer inference written in pure Mojo, leveraging SIMD and vectorization for fast CPU inferenc… | 57 | 2125 | active |
| yyfz/Pi3 Pi3 (π³) is a feed-forward neural network for visual geometry reconstruction that eliminates the need for a fixed reference view, using a p… | 59 | 2122 | active |
| LiheYoung/Depth-Anything Depth Anything is a monocular depth estimation foundation model trained on 1.5M labeled and 62M+ unlabeled images, released as a Python lib… | 26 | 8195 | maintenance |
| GAIR-NLP/daVinci-MagiHuman daVinci-MagiHuman is an open-source 15B-parameter single-stream transformer foundation model that jointly generates synchronized audio and … | 49 | 2113 | active |
| deepspeedai/DeepSpeed-MII DeepSpeed-MII is a Python library for high-throughput, low-latency large language model inference, built on DeepSpeed. It provides blocked … | 43 | 2111 | active |
| 3DTopia/LGM LGM is the official PyTorch implementation of an ECCV 2024 Oral paper that generates high-resolution 3D models from text prompts or single-… | 26 | 2111 | active |
| fangwei123456/spikingjelly SpikingJelly is an open-source deep learning framework for Spiking Neural Networks (SNNs) built on PyTorch. It provides a beginner-friendly… | 77 | 2110 | active |
| River-Zhang/ICEdit ICEdit (In-Context Edit) is a research framework for instruction-based image editing built on large-scale Diffusion Transformers, using a L… | 45 | 2102 | active |
| Open-Reasoner-Zero/Open-Reasoner-Zero Open-Reasoner-Zero is an open-source implementation of large-scale reinforcement learning training for reasoning-oriented language models, … | 31 | 2099 | active |
| eloialonso/diamond DIAMOND is a Python implementation of a reinforcement learning agent trained entirely inside a diffusion-based world model, presented as a … | 24 | 2096 | active |
| RenderKit/oidn Intel Open Image Denoise is an open-source library of high-performance, deep-learning-based denoising filters for images rendered with ray … | 94 | 2089 | stable |
| patrick-kidger/diffrax Diffrax is a JAX-based library providing numerical differential equation solvers for ODEs, SDEs, and CDEs. It is fully autodifferentiable a… | 80 | 2089 | active |