domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| floodsung/LearningToCompare_FSL Official PyTorch implementation of the Relation Network for Few-Shot Learning from the CVPR 2018 paper 'Learning to Compare'. It includes t… | 32 | 1075 | maintenance |
| JosephKJ/OWOD OWOD is the official PyTorch implementation of ORE (Open World Object Detector), a CVPR 2021 Oral paper introducing the Open World Object D… | 32 | 1074 | maintenance |
| ayoolaolafenwa/PixelLib PixelLib is a Python library for performing semantic and instance segmentation of objects in images and videos with just a few lines of cod… | 23 | 1072 | maintenance |
| huihut/OpenCV-MinGW-Build A repository of prebuilt OpenCV binaries (32-bit and 64-bit) compiled with MinGW on Windows, covering versions from 3.3.1 through 4.5.5 inc… | 23 | 1072 | maintenance |
| microsoft/VideoX VideoX is a collection of Microsoft's video cross-modal understanding models, including X-CLIP for video-language recognition, 2D-TAN and M… | 32 | 1071 | maintenance |
| gusye1234/LightGCN-PyTorch A PyTorch implementation of LightGCN, a simplified graph convolution network for collaborative filtering recommendation from the SIGIR 2020… | 10 | 1071 | maintenance |
| MatthieuCourbariaux/BinaryNet BinaryNet is a research codebase for training deep neural networks whose weights and activations are constrained to +1 or -1, reproducing t… | 32 | 1068 | maintenance |
| openai/blocksparse A Python package of TensorFlow ops and custom CUDA GPU kernels for block-sparse matrix multiplication and convolution, plus related ops lik… | 10 | 1068 | maintenance |
| EdVince/Stable-Diffusion-NCNN A C++ implementation of Stable Diffusion using the NCNN inference framework, supporting both txt2img and img2img. It runs on x86 Windows ex… | 23 | 1067 | maintenance |
| ntasfi/PyGame-Learning-Environment PyGame Learning Environment (PLE) is a Python library providing a reinforcement learning environment with a suite of PyGame-based games, mi… | 32 | 1066 | maintenance |
| omerbt/MultiDiffusion Official PyTorch implementation of MultiDiffusion (ICML 2023), a training-free framework that fuses multiple diffusion paths over a pre-tra… | 31 | 1066 | maintenance |
| NVIDIA-AI-IOT/trt_pose trt_pose is a Python library from NVIDIA for real-time human pose estimation accelerated with TensorRT, targeting NVIDIA Jetson and other N… | 23 | 1066 | maintenance |
| caicloud/cyclone Cyclone is a workflow engine and end-to-end pipeline solution built entirely on native Kubernetes resources, with no extra dependencies. It… | 23 | 1064 | maintenance |
| facebookresearch/mbrl-lib MBRL-Lib is a PyTorch-based toolbox from Facebook Research for developing model-based reinforcement learning algorithms. It provides interc… | 10 | 1064 | maintenance |
| deepmedic/deepmedic DeepMedic is an efficient multi-scale 3D convolutional neural network for segmenting 3D medical scans such as MRI and CT. It is a Python-ba… | 23 | 1063 | maintenance |
| ternaus/TernausNet TernausNet is a PyTorch implementation of the U-Net architecture with a VGG11 encoder pre-trained on ImageNet for image segmentation. It wa… | 32 | 1062 | maintenance |
| open-mmlab/mmflow MMFlow is an open-source optical flow toolbox and benchmark built on PyTorch as part of the OpenMMLab project. It provides a unified, modul… | 23 | 1061 | maintenance |
| wubinzzu/NeuRec NeuRec is an open-source Python library for recommender systems built on TensorFlow, offering 33 state-of-the-art neural recommendation mod… | 23 | 1061 | maintenance |
| jsyoon0823/TimeGAN Reference implementation of TimeGAN, a generative adversarial network framework for generating synthetic time-series data, published at Neu… | 62 | 1060 | maintenance |
| zhaoweicai/cascade-rcnn A C++/Caffe implementation of Cascade R-CNN and other popular two-stage object detection frameworks such as Faster R-CNN, R-FCN, and FPN. I… | 32 | 1060 | maintenance |
| OFA-Sys/ONE-PEACE ONE-PEACE is a general multimodal representation model that jointly encodes vision, audio, and language modalities without initializing fro… | 29 | 1060 | maintenance |
| philipperemy/keract Keract is a Python library that extracts layer activations (outputs) and gradients from TensorFlow/Keras models as NumPy arrays, with helpe… | 30 | 1059 | maintenance |
| BurguerJohn/Dain-App Dain-App is a video frame interpolation application built on the DAIN (Depth-Aware Video Frame Interpolation) deep learning model. It provi… | 23 | 1059 | maintenance |
| chenyuntc/PyTorchText A PyTorch implementation of multiple text classification models (TextCNN, TextRNN/LSTM, RCNN, FastText, inception CNN) that won 1st place i… | 32 | 1056 | maintenance |
| RexYing/gnn-model-explainer Official implementation of GNNExplainer, a NeurIPS 2019 method for explaining predictions of graph neural networks by learning soft masks o… | 32 | 1056 | maintenance |
| huggingface/optimum-quanto Optimum Quanto is a PyTorch quantization backend for Hugging Face Optimum that quantizes model weights (int2/int4/int8/float8) and activati… | 70 | 1054 | maintenance |
| raminmh/CfC Reference implementations of Closed-form Continuous-time (CfC) neural networks, a fast closed-form approximation of liquid time-constant ne… | 23 | 1054 | maintenance |
| keras-team/keras-cv KerasCV is a library of modular computer vision components built on Keras 3 that work natively with TensorFlow, JAX, or PyTorch. It provide… | 10 | 1054 | maintenance |
| aliutkus/speechmetrics A Python library that wraps several objective speech quality metrics (MOSNet, BSSEval, STOI, PESQ, SRMR, SISDR) behind a unified API. It su… | 32 | 1053 | maintenance |
| Edresson/YourTTS YourTTS is a zero-shot multi-speaker text-to-speech and voice conversion model built on VITS, implemented in the Coqui TTS framework. It su… | 23 | 1053 | maintenance |
| mpaepper/llm_agents A small Python library for building agents controlled by large language models, inspired by LangChain but implemented from scratch in very … | 43 | 1052 | maintenance |
| ashkamath/mdetr MDETR (Modulated Detection) is a PyTorch research codebase for end-to-end multi-modal object detection that grounds free-form text queries … | 32 | 1052 | maintenance |
| damo-cv/TransReID Official PyTorch implementation of TransReID, an ICCV 2021 paper applying vision transformers to object re-identification. It provides trai… | 32 | 1052 | maintenance |
| lhelontra/tensorflow-on-arm A set of shell scripts and Docker-based build tooling for cross-compiling TensorFlow wheels for ARM boards like Raspberry Pi and Odroid. It… | 23 | 1052 | maintenance |
| jondurbin/airoboros Airoboros is a Python library implementing a heavily modified version of the Self-Instruct paper to generate high-quality synthetic instruc… | 20 | 1051 | maintenance |
| caoscott/SReC SReC is the official PyTorch implementation of the paper 'Lossless Image Compression through Super-Resolution', which frames lossless image… | 32 | 1050 | maintenance |
| yangxudong/deeplearning A collection of deep learning model training, evaluation, and prediction code implemented with TensorFlow's high-level Estimator API, with … | 32 | 1048 | maintenance |
| microsoft/SimMIM Official PyTorch implementation of SimMIM, a simple framework for masked image modeling (self-supervised visual pre-training) from Microsof… | 32 | 1048 | maintenance |
| uber/fiber Fiber is a Python distributed computing library for modern computer clusters that exposes the same API as Python's standard multiprocessing… | 10 | 1047 | maintenance |
| yuval-alaluf/restyle-encoder Official PyTorch implementation of ReStyle, a residual-based StyleGAN encoder that inverts real images into GAN latent codes via iterative … | 32 | 1046 | maintenance |
| thunlp/OpenDelta OpenDelta is a Python library for parameter-efficient tuning (delta tuning) of pretrained language models, letting users attach small train… | 23 | 1046 | maintenance |
| pkhungurn/talking-head-anime-3-demo Demo programs for the Talking Head(?) Anime 3 project, which animates an anime character from a single image using machine learning. It inc… | 32 | 1045 | maintenance |
| facebookresearch/cc_net CCNet is a Python pipeline from Facebook AI Research for downloading, deduplicating, and cleaning Common Crawl web data into high-quality m… | 10 | 1044 | maintenance |
| YuwenXiong/py-R-FCN A Python implementation of R-FCN (Region-based Fully Convolutional Networks) for object detection, modified from the official MATLAB code a… | 32 | 1043 | maintenance |
| haltakov/natural-language-image-search A demo application that searches Unsplash photos using natural language descriptions, powered by OpenAI's CLIP model over the ~2M photo Uns… | 23 | 1042 | maintenance |
| asingh33/CNNGestureRecognizer A desktop application that recognizes hand gestures from webcam video using a convolutional neural network built with Keras, TensorFlow/The… | 60 | 1041 | maintenance |
| IBM/MAX-Image-Resolution-Enhancer An IBM Model Asset Exchange project that deploys an SRGAN-based image super-resolution model as a web service in a Docker container. It ups… | 42 | 1041 | maintenance |
| thuml/Anomaly-Transformer Official PyTorch implementation of the Anomaly Transformer model from the ICLR 2022 Spotlight paper on unsupervised time series anomaly det… | 32 | 1041 | maintenance |
| descriptinc/melgan-neurips Official PyTorch implementation of MelGAN, a GAN-based non-autoregressive vocoder that inverts mel-spectrograms into raw audio waveforms fo… | 32 | 1040 | maintenance |
| pykaldi/pykaldi PyKaldi is a Python scripting layer providing wrappers for the C++ APIs of the Kaldi speech recognition toolkit and OpenFst library. It ena… | 47 | 1039 | maintenance |
| MaybeShewill-CV/CRNN_Tensorflow A TensorFlow implementation of CRNN (CNN + Bi-LSTM + CTC loss) for scene text recognition, based on the Shi et al. paper. It includes pretr… | 32 | 1039 | maintenance |
| huggingface/pytorch-pretrained-BigGAN A PyTorch reimplementation of DeepMind's BigGAN generator with pretrained weights at 128, 256, and 512 pixel resolutions, plus scripts to c… | 23 | 1039 | maintenance |
| google-research/morph-net MorphNet is a TensorFlow library for resource-constrained learning of deep network structure. It adds regularizers during training that ind… | 73 | 1038 | maintenance |
| yashbhalgat/HashNeRF-pytorch A pure PyTorch implementation of NVIDIA's Instant-NGP multi-resolution hash encoding for training NeRFs, built on top of NeRF-pytorch. It e… | 32 | 1037 | maintenance |
| google-research/deeplab2 DeepLab2 is a TensorFlow library from Google Research providing a unified, state-of-the-art codebase for dense pixel labeling tasks such as… | 10 | 1037 | maintenance |
| craffel/pretty-midi A Python library providing utility functions and classes for conveniently handling, analyzing, creating, and manipulating MIDI data. It sup… | 57 | 1036 | maintenance |
| Xwin-LM/Xwin-LM Xwin-LM is an open-source project for LLM alignment technologies including supervised fine-tuning, reward models, reject sampling, and RLHF… | 28 | 1036 | maintenance |
| sicara/tf-explain tf-explain is a Python library implementing interpretability methods (Grad CAM, occlusion sensitivity, vanilla gradients, etc.) for tf.kera… | 23 | 1036 | maintenance |
| Kismuz/btgym BTGym is a Python library that wraps the Backtrader algorithmic trading engine in an OpenAI Gym environment API, enabling event-driven back… | 32 | 1035 | maintenance |
| pesser/stable-diffusion The development repository for Stable Diffusion and Latent Diffusion Models, containing research code, training scripts, and pretrained mod… | 32 | 1033 | maintenance |
| goberoi/faceit A Python script that simplifies swapping faces in videos using the deepfakes/faceswap library, with training data sourced from YouTube vide… | 32 | 1032 | maintenance |
| ClementPinard/SfmLearner-Pytorch A PyTorch reimplementation of SfMLearner (Zhou et al., CVPR 2017), which jointly learns monocular depth estimation and camera ego-motion fr… | 67 | 1031 | maintenance |
| xingyizhou/ExtremeNet Official PyTorch implementation of ExtremeNet, a CVPR 2019 bottom-up object detection method that detects four extreme points and one cente… | 32 | 1031 | maintenance |
| vincentherrmann/pytorch-wavenet A PyTorch implementation of the WaveNet neural audio generation architecture with fast generation support. It includes dataset creation fro… | 32 | 1031 | maintenance |
| dandelionsllm/pandallm Panda is an open-source project for overseas Chinese large language models, providing PandaLLM model weights (continued pretraining of LLaM… | 30 | 1031 | maintenance |
| ArrowLuo/CLIP4Clip Official PyTorch implementation of the CLIP4Clip paper, a video-text retrieval model that transfers CLIP knowledge to end-to-end video clip… | 23 | 1031 | maintenance |
| graphdeeplearning/graphtransformer Reference implementation of the Graph Transformer architecture from the paper 'A Generalization of Transformer Networks to Graphs' (DLG-AAA… | 32 | 1030 | maintenance |
| qubvel/ttach TTAch is a Python library for image test time augmentation (TTA) with PyTorch. It wraps existing models to apply augmentations like flips, … | 23 | 1030 | maintenance |
| google-deepmind/scalable_agent A TensorFlow implementation of IMPALA (Importance Weighted Actor-Learner Architectures) for scalable distributed deep reinforcement learnin… | 32 | 1029 | maintenance |
| mimno/Mallet MALLET (MAchine Learning for LanguagE Toolkit) is a Java-based package for statistical natural language processing, including document clas… | 86 | 1028 | maintenance |
| zepen/predict_Lottery_ticket A Python application that scrapes historical draw data for Chinese lotteries (双色球/Double Color Ball and 大乐透/Super Lotto) and trains deep le… | 34 | 1027 | maintenance |
| microsoft/Llama-2-Onnx Microsoft's optimized ONNX export of Meta's Llama 2 models (7B and 13B, pretrained and fine-tuned, float16/float32), distributed via Git su… | 28 | 1026 | maintenance |
| edvardHua/PoseEstimationForMobile A TensorFlow-based library implementing CPM and Hourglass models with MobileNetV2 inverted residual modules for real-time single-person hum… | 32 | 1024 | maintenance |
| lmb-freiburg/flownet2 A Caffe fork implementing FlowNet 2.0, a deep CNN for optical flow estimation from image pairs, released with the CVPR 2017 paper. It inclu… | 32 | 1024 | maintenance |
| dwofk/fast-depth FastDepth is the official PyTorch implementation of the ICRA 2019 paper 'FastDepth: Fast Monocular Depth Estimation on Embedded Systems' fr… | 32 | 1024 | maintenance |
| cxxnet ps-lite is a lightweight, efficient C++ implementation of the parameter server framework for distributed machine learning. It exposes simpl… | 10 | 1024 | maintenance |
| awslabs/multi-model-server Multi Model Server (MMS) is a tool for serving deep learning model inference over HTTP endpoints, supporting models from any ML/DL framewor… | 10 | 1024 | maintenance |
| turtlesoupy/this-word-does-not-exist A project that trains a GPT-2 variant to invent fake English words with generated definitions and example sentences, powering the thiswordd… | 72 | 1023 | maintenance |
| NaturalIntelligence/imglab ImgLab is a browser-based image annotation tool for labeling objects and landmark points to train object detectors like dlib. It supports m… | 76 | 1021 | maintenance |
| sniklaus/sepconv-slomo A reference PyTorch implementation of Video Frame Interpolation via Adaptive Separable Convolution, which generates intermediate frames bet… | 43 | 1021 | maintenance |
| Gumpest/YOLOv5-Multibackbone-Compression A YOLOv5-based toolbox for swapping in lightweight or high-accuracy backbones (TPH-YOLOv5, GhostNet, ShuffleNetV2, MobileNetV3-Small, Effic… | 32 | 1020 | maintenance |
| neeru1207/AI_Sudoku A Python desktop application with a Tkinter GUI that extracts a Sudoku puzzle from a photo using OpenCV image processing and solves it. Dig… | 32 | 1020 | maintenance |
| facebookresearch/CompilerGym CompilerGym is a Python library of OpenAI Gym-compatible reinforcement learning environments for compiler and program optimization tasks, c… | 10 | 1020 | maintenance |
| tensorflow/similarity TensorFlow Similarity is a Python library built on TensorFlow for training and serving similarity and metric learning models, including sel… | 10 | 1020 | maintenance |
| bayesiains/nflows nflows is a comprehensive PyTorch library of normalizing flows for density estimation and generative modeling. It provides invertible trans… | 23 | 1019 | maintenance |
| brain-research/self-attention-gan A TensorFlow implementation of Self-Attention GANs for reproducing results from the paper 'Self-Attention Generative Adversarial Networks' … | 10 | 1019 | maintenance |
| RyanZotti/Self-Driving-Car A hobby project that turns a toy Ferrari into a self-driving car using a Raspberry Pi, a PS3 controller for data collection, and a locally-… | 32 | 1017 | maintenance |
| facebookresearch/barlowtwins A PyTorch implementation of Barlow Twins, a self-supervised learning method that trains visual representations by reducing redundancy betwe… | 10 | 1016 | maintenance |
| wpeebles/gangealing Official PyTorch implementation of GANgealing, a CVPR 2022 method that trains a Spatial Transformer to densely align images using GAN-gener… | 32 | 1012 | maintenance |
| tensorflow/neural-structured-learning Neural Structured Learning (NSL) is a TensorFlow framework for training neural networks with structured signals, either explicit graphs or … | 65 | 1011 | maintenance |
| PAIR-code/what-if-tool The What-If Tool (WIT) is a visual interface from Google's PAIR team for probing black-box classification and regression ML models without … | 62 | 1011 | maintenance |
| AMLab-Amsterdam/AttentionDeepMIL A PyTorch implementation of the paper 'Attention-based Deep Multiple Instance Learning' (Ilse, Tomczak & Welling, 2018). It provides code t… | 32 | 1011 | maintenance |
| kakaobrain/kogpt KakaoBrain's KoGPT, a Korean Generative Pre-trained Transformer (GPT) model with 6B parameters, distributed via Hugging Face with inference… | 23 | 1011 | maintenance |
| zju3dv/OnePose OnePose is the official PyTorch implementation of the CVPR 2022 paper 'One-Shot Object Pose Estimation without CAD Models'. It estimates th… | 32 | 1010 | maintenance |
| allegro/allRank allRank is a PyTorch-based framework for training neural learning-to-rank (LTR) models. It provides pointwise, pairwise, and listwise loss … | 32 | 1010 | maintenance |
| xiaoyufenfei/Efficient-Segmentation-Networks A PyTorch reference implementation collection of lightweight, real-time semantic segmentation models such as ENet, ERFNet, LEDNet, Fast-SCN… | 32 | 1008 | maintenance |
| naver/splade SPLADE is a research library from NAVER for training, indexing, and retrieval with sparse neural search models based on BERT. It learns spa… | 23 | 1008 | maintenance |
| kevinzakka/spatial-transformer-network A TensorFlow implementation of Spatial Transformer Networks, a differentiable module that can be inserted into ConvNet architectures to add… | 32 | 1005 | maintenance |
| bubbliiiing/yolov8-pytorch A PyTorch implementation of the YOLOv8 object detection model with training, prediction, and evaluation scripts. It supports training on cu… | 21 | 1005 | maintenance |
| NATSpeech/NATSpeech A PyTorch framework for non-autoregressive text-to-speech (NAR-TTS), containing official implementations of PortaSpeech (NeurIPS 2021) and … | 23 | 1004 | maintenance |