domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| punica-ai/punica Punica is a Python system for serving many LoRA-finetuned LLMs from a single copy of the base model on one GPU, using a custom CUDA kernel … | 19 | 1175 | maintenance |
| julrog/nn_vis A Python tool that processes trained neural networks and renders them as decluttered 3D visualizations of their architecture and parameters… | 23 | 1174 | maintenance |
| MLNLP-World/AI-Paper-Collector A tool from the MLNLP community that automatically collects AI-related papers from sources like DBLP, ACL Anthology, NIPS, and OpenReview, … | 32 | 1173 | maintenance |
| ermongroup/SDEdit SDEdit is a PyTorch implementation of the ICLR 2022 paper 'Guided Image Synthesis and Editing with Stochastic Differential Equations'. It p… | 32 | 1168 | maintenance |
| WebSpiderUtils/verification_code A research repository documenting approaches and code for solving mainstream CAPTCHA systems such as Geetest, NetEase Yidun, and Aliyun CAP… | 32 | 1165 | maintenance |
| alexandre01/deepsvg Official PyTorch code for the NeurIPS 2020 paper 'DeepSVG: A Hierarchical Generative Network for Vector Graphics Animation'. It provides a … | 32 | 1164 | maintenance |
| xiangwang1223/knowledge_graph_attention_network TensorFlow implementation of KGAT (Knowledge Graph Attention Network), a KDD 2019 paper on knowledge-aware personalized recommendation. It … | 32 | 1164 | maintenance |
| google/active-learning A Python module for running experiments comparing different active learning algorithms on benchmark datasets. It provides a main experiment… | 10 | 1163 | maintenance |
| facebookresearch/meshrcnn Mesh R-CNN is Facebook AI Research's official implementation of the ICCV 2019 paper, a model that detects objects in images and predicts th… | 60 | 1161 | maintenance |
| cheungdaven/DeepRec DeepRec is an open-source Python/TensorFlow toolkit implementing deep learning based recommendation models for rating prediction, top-N ite… | 32 | 1160 | maintenance |
| wyharveychen/CloserLookFewShot Reference PyTorch implementation of the ICLR 2019 paper 'A Closer Look at Few-shot Classification', providing a testbed for empirical evalu… | 32 | 1159 | maintenance |
| FALCONN-LIB/FALCONN FALCONN is a C++ library (with a Python wrapper) implementing locality-sensitive hashing (LSH) data structures for fast nearest neighbor se… | 23 | 1159 | maintenance |
| Paperspace/DataAugmentationForObjectDetection A Python library implementing data augmentation transforms (flipping, scaling, rotation, shearing, translation, resizing) that adjust both … | 32 | 1156 | maintenance |
| abhishekkrthakur/tez Tez is a lightweight, simple trainer library for PyTorch that simplifies training loops while keeping users close to native PyTorch. It sup… | 23 | 1155 | maintenance |
| DeepGraphLearning/RecommenderSystems A Python research library collecting the authors' recommender systems code, covering sequential, feature-based (CTR), and social recommenda… | 32 | 1154 | maintenance |
| Xharlie/pointnerf Point-NeRF is a research implementation of a point-based neural radiance field method (CVPR 2022 Oral) that models scenes with neural 3D po… | 32 | 1154 | maintenance |
| pkhungurn/talking-head-anime-2-demo Demo programs for the Talking Head Anime from a Single Image 2 project, which animates an anime character from a single image. It includes … | 32 | 1154 | maintenance |
| uber-research/PPLM PPLM (Plug and Play Language Model) is a research implementation for controlled text generation that steers the topic and attributes of GPT… | 32 | 1152 | maintenance |
| PengyiZhang/SlimYOLOv3 SlimYOLOv3 is a research implementation of channel-pruned YOLOv3 object detectors optimized for real-time inference on UAVs and embedded de… | 32 | 1152 | maintenance |
| ZhuiyiTechnology/roformer RoFormer is an MLM pre-trained language model built on rotary position embeddings (RoPE), a relative position encoding method with strong t… | 32 | 1152 | maintenance |
| NVIDIA-Merlin/NVTabular NVTabular is a GPU-accelerated feature engineering and preprocessing library for tabular data, built to handle terabyte-scale datasets for … | 60 | 1151 | maintenance |
| andrewkirillov/AForge.NET AForge.NET is an open-source C# framework for computer vision and artificial intelligence, comprising libraries such as AForge.Imaging, AFo… | 32 | 1151 | maintenance |
| HackerPoet/Composer Composer is a neural network-based application that generates video game music. It is written in Python and produces chiptune-style game so… | 32 | 1151 | maintenance |
| lensacom/sparkit-learn Sparkit-learn provides scikit-learn's API and functionality on top of PySpark, operating on distributed RDDs of numpy arrays and sparse mat… | 23 | 1150 | maintenance |
| bubbliiiing/yolov5-pytorch A PyTorch implementation of the YOLOv5 (v5.0) object detection model with full training, prediction, and evaluation pipelines. It is design… | 23 | 1149 | maintenance |
| showlab/Show-1 Show-1 is a research codebase implementing a text-to-video generation model that combines pixel and latent diffusion models, published at I… | 46 | 1148 | maintenance |
| tensorflow/benchmarks A benchmark framework for TensorFlow, containing PerfZero for running and reporting TensorFlow benchmarks and legacy CNN benchmark scripts … | 10 | 1148 | maintenance |
| joe-siyuan-qiao/DetectoRS Official PyTorch implementation of DetectoRS, a state-of-the-art object detection and instance segmentation model using Recursive Feature P… | 32 | 1147 | maintenance |
| tech-srl/code2vec Official TensorFlow implementation of the code2vec model from the POPL'2019 paper, which learns distributed vector representations of code … | 32 | 1147 | maintenance |
| AimeeLee77/keyword_extraction A Python project implementing Chinese text keyword extraction using three methods: TF-IDF, TextRank, and Word2Vec word clustering. It inclu… | 32 | 1145 | maintenance |
| eBay/bayesian-belief-networks A Python framework for defining Bayesian Belief Networks and other graphical models as pure Python functions, with support for discrete and… | 10 | 1142 | maintenance |
| patil-suraj/question_generation An open-source study and library for neural question generation using pre-trained seq2seq transformer models like T5 via Hugging Face trans… | 32 | 1140 | maintenance |
| google-research/mixmatch Reference implementation of MixMatch, a holistic semi-supervised learning algorithm from a Google Research paper. It includes dataset prepa… | 10 | 1139 | maintenance |
| HRNet/HRNet-Facial-Landmark-Detection Official PyTorch implementation of HRNet-based facial landmark detection from the TPAMI paper 'Deep High-Resolution Representation Learning… | 32 | 1138 | maintenance |
| IBM/Dromedary Dromedary is an open-source self-aligned language model trained with minimal human supervision using the principle-driven SELF-ALIGN pipeli… | 49 | 1137 | maintenance |
| arpitbansal297/Cold-Diffusion-Models Official PyTorch implementation of Cold Diffusion, a research paper showing that diffusion-style generative models can invert arbitrary ima… | 32 | 1136 | maintenance |
| karfly/learnable-triangulation-pytorch Official PyTorch implementation of the ICCV 2019 paper 'Learnable Triangulation of Human Pose', providing algebraic and volumetric learnabl… | 23 | 1135 | maintenance |
| Timthony/self_drive A self-driving RC car project based on Raspberry Pi and TensorFlow/Keras. It collects camera images while a human drives the car on a taped… | 32 | 1134 | maintenance |
| harvardnlp/pytorch-struct Torch-Struct is a PyTorch library of tested, GPU-accelerated implementations of core structured prediction algorithms such as CRFs, HMMs, H… | 23 | 1133 | maintenance |
| puzzledqs/BBox-Label-Tool A simple Python Tkinter desktop application for drawing and saving object bounding box annotations on images. It stores labels as text file… | 32 | 1132 | maintenance |
| Cloud-CV/Fabrik Fabrik is a web-based collaborative IDE for building, visualizing, and designing neural networks via a drag-and-drop interface. It supports… | 32 | 1131 | maintenance |
| fff-rs/juice Juice is a Rust machine learning framework ('The Hacker's Machine Learning Engine') built on the Coaster hardware abstraction layer, suppor… | 32 | 1131 | maintenance |
| KinWaiCheuk/nnAudio nnAudio is a PyTorch-based audio processing toolbox that computes spectrograms (STFT, Mel, CQT, MFCC) using 1D convolutional neural network… | 60 | 1127 | maintenance |
| yu-takagi/StableDiffusionReconstruction Research codebase reproducing Takagi and Nishimoto's CVPR 2023 method for reconstructing images a person viewed from fMRI brain activity us… | 30 | 1127 | maintenance |
| julvo/reloading A small Python library that reloads the source code of loops or functions on each iteration or invocation without losing runtime state. It … | 23 | 1127 | maintenance |
| ray-project/llmperf LLMPerf is a Python library for benchmarking and validating the performance of LLM APIs. It runs load tests measuring inter-token latency a… | 10 | 1126 | maintenance |
| microsoft/Windows-Machine-Learning Microsoft's Windows Machine Learning samples and tools repository, providing a high-performance ONNX inference API powered by ONNX Runtime … | 39 | 1123 | maintenance |
| bearpaw/pytorch-pose A PyTorch toolkit implementing a general pipeline for 2D single-human pose estimation, with training, inference, and evaluation interfaces … | 32 | 1121 | maintenance |
| irolaina/FCRN-DepthPrediction Reference implementation and pretrained models for FCRN (Deeper Depth Prediction with Fully Convolutional Residual Networks), predicting de… | 32 | 1118 | maintenance |
| Harmonai-org/sample-generator A set of tools and Jupyter notebooks for training generative diffusion models on arbitrary audio samples, built around Dance Diffusion. It … | 32 | 1118 | maintenance |
| Res2Net/Res2Net-PretrainedModels Official PyTorch implementation of Res2Net, a multi-scale CNN backbone architecture published in TPAMI, with ImageNet-pretrained model weig… | 32 | 1115 | maintenance |
| mit-han-lab/gan-compression A PyTorch research codebase implementing GAN Compression, a CVPR 2020 method that reduces the computation of conditional GANs like pix2pix,… | 23 | 1115 | maintenance |
| microsoft/MASS MASS is Microsoft's PyTorch implementation of Masked Sequence to Sequence Pre-training for language generation tasks. It provides pre-train… | 10 | 1115 | maintenance |
| thuml/TimesNet TimesNet is the official code release for the ICLR 2023 paper on temporal 2D-variation modeling for general time series analysis. It transf… | 31 | 1114 | maintenance |
| LeonLok/Multi-Camera-Live-Object-Tracking A Python application that performs live multi-camera object detection, tracking, and counting using YOLO v4 and Deep SORT, with a Flask web… | 32 | 1111 | maintenance |
| hotshotco/Hotshot-XL Hotshot-XL is an AI text-to-GIF model built to work alongside Stable Diffusion XL, generating 1-second GIFs at 8 FPS. It supports any fine-… | 27 | 1111 | maintenance |
| rosinality/style-based-gan-pytorch A PyTorch implementation of the StyleGAN paper (A Style-Based Generator Architecture for Generative Adversarial Networks), including traini… | 32 | 1110 | maintenance |
| biubug6/Face-Detector-1MB-with-landmark A PyTorch-based ultra-lightweight face detection library (~1MB models) with 5-point facial landmark detection, offering slim, RFB, and Reti… | 32 | 1109 | maintenance |
| fudan-zvg/SETR SETR (SEgmentation TRansformers) is the official PyTorch implementation of the CVPR 2021 / IJCV 2024 paper 'Rethinking Semantic Segmentatio… | 32 | 1108 | maintenance |
| novak-99/MLPP ML++ is a C++ machine learning library providing regression models, dynamically sized neural networks, statistics, and optimization algorit… | 23 | 1108 | maintenance |
| aim-uofa/AdelaiDepth AdelaiDepth is an open-source toolbox for monocular depth prediction and 3D scene reconstruction from single images, containing research pr… | 32 | 1105 | maintenance |
| vietnh1009/QuickDraw A Python implementation of Google's Quick, Draw! game that recognizes hand-drawn sketches using a CNN trained on the Quick Draw dataset. It… | 32 | 1104 | maintenance |
| msracver/Relation-Networks-for-Object-Detection Official MXNet implementation of the CVPR 2018 paper 'Relation Networks for Object Detection', which adds an attention-based relation modul… | 32 | 1104 | maintenance |
| keroro824/HashingDeepLearning SLIDE is a C++ research codebase implementing locality-sensitive hashing based training of deep neural networks, from the paper 'In Defense… | 32 | 1104 | maintenance |
| PaddlePaddle/Paddle.js Paddle.js is a browser-based deep learning inference engine for Baidu PaddlePaddle models, running via WebGL, WebGPU, or WebAssembly backen… | 23 | 1104 | maintenance |
| pfnet-research/sngan_projection Official Chainer implementation of GANs using spectral normalization and projection discriminators for conditional image generation on Imag… | 32 | 1103 | maintenance |
| Flode-Labs/vid2densepose A Python tool that applies the DensePose model to videos, producing color-coded part-index visualizations for each frame. Its output is des… | 26 | 1102 | maintenance |
| rlancemartin/auto-evaluator A lightweight Streamlit-based evaluation tool for LLM question-answering chains built with LangChain. It auto-generates question-answer pai… | 30 | 1101 | maintenance |
| auspicious3000/autovc AUTOVC is a PyTorch implementation of a many-to-many non-parallel voice conversion framework that performs zero-shot voice style transfer u… | 32 | 1100 | maintenance |
| hhaAndroid/mmdetection-mini A minimal, heavily annotated reimplementation of the mmdetection object detection framework, built from scratch for learning purposes. It m… | 32 | 1100 | maintenance |
| jacobgil/vit-explain A PyTorch library implementing explainability methods for Vision Transformers, including Attention Rollout and Gradient Attention Rollout. … | 32 | 1100 | maintenance |
| EleutherAI/math-lm Repository for Llemma, an open language model for mathematics, hosting training code, data preprocessing scripts, and evaluation tooling. I… | 31 | 1097 | maintenance |
| ethz-asl/segmap SegMap is a C++ library and ROS package providing a segment-based 3D map representation for robot localization, environment reconstruction,… | 32 | 1096 | maintenance |
| Tencent/TencentPretrain TencentPretrain is a PyTorch-based toolkit for pre-training and fine-tuning transformer models across modalities such as text, vision, and … | 32 | 1093 | maintenance |
| NVlabs/NVAE Official PyTorch implementation of NVAE, a deep hierarchical variational autoencoder for likelihood-based image generation from a NeurIPS 2… | 32 | 1092 | maintenance |
| dyh/unbox_yolov5_deepsort_counting A Python application that combines YOLOv5 object detection with DeepSORT multi-object tracking to detect, track, and count pedestrians and … | 32 | 1091 | maintenance |
| pkuliyi2015/sd-webui-stablesr A Stable Diffusion WebUI (Automatic1111) plugin that integrates StableSR, a diffusion-prior-based super-resolution method, for high-fidelit… | 29 | 1090 | maintenance |
| MatthewJA/Inverse-Reinforcement-Learning A Python library implementing selected inverse reinforcement learning (IRL) algorithms, including linear programming IRL, maximum entropy I… | 23 | 1090 | maintenance |
| maelfabien/Multimodal-Emotion-Recognition A real-time multimodal emotion recognition web app built with Flask that analyzes emotions from text, audio, and video inputs using deep le… | 32 | 1089 | maintenance |
| yu4u/noise2noise An unofficial, partial Keras implementation of the Noise2Noise paper for learning image restoration without clean training data. It trains … | 23 | 1088 | maintenance |
| ethanhe42/channel-pruning Reference implementation of the ICCV 2017 channel pruning method for accelerating very deep convolutional neural networks, using LASSO regr… | 23 | 1088 | maintenance |
| replit/ReplitLM Official repository with inference code, configs, and guides for the ReplitLM family of code-oriented language models, such as replit-code-… | 10 | 1087 | maintenance |
| sfzhang15/ATSS Official PyTorch implementation of ATSS (Adaptive Training Sample Selection), a CVPR 2020 Oral paper on object detection. It automatically … | 32 | 1086 | maintenance |
| emedvedev/attention-ocr A TensorFlow-based OCR model using a CNN plus seq2seq architecture with visual attention for text recognition in images. It ships as a Pyth… | 23 | 1086 | maintenance |
| ibayer/fastFM fastFM is a Python library for Factorization Machines with a scikit-learn compatible API, with performance-critical code written in C and w… | 23 | 1086 | maintenance |
| GOATmessi8/ASFF A PyTorch implementation of YOLOv3 with the Adaptively Spatial Feature Fusion (ASFF) module and optional MobileNetV2 backbone for single-sh… | 32 | 1085 | maintenance |
| PrincetonML/SIF A Python research library implementing the Smooth Inverse Frequency (SIF) weighting scheme for computing sentence embeddings, from the ICLR… | 32 | 1085 | maintenance |
| datumbox/datumbox-framework Datumbox is an open-source Machine Learning framework written in Java that enables rapid development of ML and statistical applications. It… | 32 | 1085 | maintenance |
| AstarLight/CPS-OCR-Engine A deep-learning-based OCR engine from SYSU DeepDriving Lab that recognizes 3755 printed Chinese characters (Level-1 character set) in elect… | 32 | 1083 | maintenance |
| rykov8/ssd_keras A Keras port of the SSD (Single Shot MultiBox Detector) object detection model, including pretrained weights ported from the original Caffe… | 10 | 1083 | maintenance |
| seongjunyun/Graph_Transformer_Networks Authors' PyTorch implementation of Graph Transformer Networks (GTN) from the NeurIPS 2019 paper, plus FastGTN with non-local operations. It… | 32 | 1082 | maintenance |
| wilson1yan/VideoGPT VideoGPT is a PyTorch implementation of a video generation model that combines a VQ-VAE with 3D convolutions and axial attention with a GPT… | 32 | 1082 | maintenance |
| rixwew/pytorch-fm A PyTorch library implementing a wide collection of factorization machine models (FM, FFM, DeepFM, xDeepFM, AutoInt, NFM, DCN, and more) fo… | 32 | 1081 | maintenance |
| akelleh/causality A Python library providing tools for causal inference and analysis from observational data, built as an extension of pandas DataFrames. It … | 27 | 1081 | maintenance |
| openai/automated-interpretability OpenAI's code and tools for automatically generating, simulating, and scoring explanations of neuron behavior in language models, based on … | 10 | 1081 | maintenance |
| rhiever/datacleaner A Python package and command-line tool that automatically cleans pandas DataFrames, handling missing values and encoding categorical variab… | 23 | 1079 | maintenance |
| pytorch/torchdynamo TorchDynamo is a Python-level JIT compiler that speeds up unmodified PyTorch programs by capturing Python bytecode into FX graphs. The proj… | 10 | 1079 | maintenance |
| locuslab/mpc.pytorch mpc.pytorch is a fast and differentiable model predictive control (MPC) solver implemented as a PyTorch library. It solves receding-horizon… | 32 | 1078 | maintenance |
| orobix/Prototypical-Networks-for-Few-shot-Learning-PyTorch A PyTorch implementation of Prototypical Networks for few-shot learning, including the prototypical loss and episode-based batch sampler. I… | 32 | 1078 | maintenance |
| comet-ml/kangas Kangas is a Python library and tool for exploring, analyzing, and visualizing large-scale multimedia datasets via its DataGrid format. It o… | 23 | 1077 | maintenance |