function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| chengzeyi/stable-fast Stable Fast is an ultra lightweight performance optimization framework for HuggingFace Diffusers pipelines on NVIDIA GPUs. It achieves SOTA… | 23 | 1304 | maintenance |
| nlpyang/PreSumm PreSumm is the official PyTorch implementation of the EMNLP 2019 paper 'Text Summarization with Pretrained Encoders', providing BERT-based … | 32 | 1303 | maintenance |
| jannerm/diffuser Official code for the ICML 2022 paper 'Planning with Diffusion for Flexible Behavior Synthesis', which plans by denoising trajectories with… | 32 | 1302 | maintenance |
| nok/sklearn-porter A Python library that transpiles trained scikit-learn estimators into source code for C, Go, Java, JavaScript, PHP, and Ruby. It targets em… | 23 | 1302 | maintenance |
| eliorc/node2vec A Python library implementing the node2vec algorithm for learning low-dimensional embeddings of nodes and edges in graphs. It integrates wi… | 43 | 1301 | maintenance |
| digantamisra98/Mish Official repository for the Mish activation function, a self-regularized non-monotonic neural activation function published at BMVC 2020. I… | 74 | 1299 | maintenance |
| ShichenLiu/SoftRas SoftRas is a PyTorch-based differentiable renderer that treats rasterization as a differentiable aggregating process over mesh triangles, e… | 56 | 1299 | maintenance |
| foolwood/DaSiamRPN PyTorch implementation of DaSiamRPN, an ECCV 2018 distractor-aware Siamese network for visual object tracking, winner of the VOT-18 real-ti… | 32 | 1299 | maintenance |
| vietnh1009/Super-mario-bros-PPO-pytorch A PyTorch implementation of the Proximal Policy Optimization (PPO) algorithm that trains an agent to play Super Mario Bros via OpenAI Gym. … | 32 | 1298 | maintenance |
| NVlabs/DG-Net DG-Net is a PyTorch implementation of the CVPR 2019 (Oral) paper 'Joint Discriminative and Generative Learning for Person Re-identification… | 32 | 1298 | maintenance |
| SeanLee97/xmnlp xmnlp is an out-of-the-box Chinese natural language processing toolkit in Python built on ONNX Runtime. It provides lexical analysis (segme… | 23 | 1297 | maintenance |
| unitaryai/detoxify Detoxify is a Python library providing pre-trained transformer models that classify toxic comments, built on PyTorch Lightning and Hugging … | 76 | 1296 | maintenance |
| Netflix/vectorflow Vectorflow is a minimalist neural network library written in D, optimized for sparse data and single-machine training environments. It is d… | 32 | 1295 | maintenance |
| google-research/multilingual-t5 Code and resources for mT5, a massively multilingual text-to-text transformer pretrained on the mC4 corpus covering 101 languages. It repro… | 10 | 1294 | maintenance |
| tensorflow/text TensorFlow Text is a library of text-processing classes and ops for TensorFlow 2.x, providing tokenization, Unicode normalization, n-grams,… | 90 | 1293 | maintenance |
| rocketlaunchr/dataframe-go dataframe-go is a lightweight Go library providing DataFrame and Series data structures for statistics, machine-learning, and data manipula… | 32 | 1292 | maintenance |
| bubbliiiing/deeplabv3-plus-pytorch A PyTorch implementation of the DeepLabv3+ semantic segmentation model with MobileNetV2 and Xception backbones. It includes scripts for tra… | 23 | 1292 | maintenance |
| HumanAIGC/OutfitAnyone OutfitAnyone is an Alibaba research project providing ultra-high quality AI virtual try-on, letting users visualize any clothing on any per… | 26 | 5984 | experimental |
| NVlabs/alias-free-gan The official project page and code for Alias-Free GAN, an NVIDIA research method for aliasing-free generative adversarial networks. The imp… | 32 | 1289 | maintenance |
| charlesXu86/Chatbot_CN Chatbot_CN is a Chinese multi-turn, task-oriented chatbot platform targeting the finance and legal domains (with chitchat support), combini… | 32 | 1289 | maintenance |
| zeusees/License-Plate-Detector A YOLOv5-based license plate detection model trained on the CCPD dataset and proprietary data, supporting many Chinese plate types. It prov… | 32 | 1288 | maintenance |
| palash1992/GEM GEM is a Python library offering a unified framework for static graph embedding algorithms, including node2vec, HOPE, SDNE, Graph Factoriza… | 23 | 1288 | maintenance |
| apple/ml-neuman Official reference implementation of NeuMan (ECCV 2022), which reconstructs an animatable human and the background scene from a single vide… | 32 | 1287 | maintenance |
| Khrylx/PyTorch-RL A PyTorch library implementing deep reinforcement learning policy gradient algorithms (TRPO, PPO, A2C) and Generative Adversarial Imitation… | 32 | 1286 | maintenance |
| kwea123/ngp_pl A PyTorch + CUDA implementation of NVIDIA's Instant-NGP focused on NeRF, trained with PyTorch Lightning for fast, high-quality neural radia… | 23 | 1286 | maintenance |
| weihua916/powerful-gnns Official PyTorch implementation of the experiments from the ICLR 2019 paper 'How Powerful are Graph Neural Networks?', introducing the Grap… | 32 | 1285 | maintenance |
| SelfExplainML/PiML-Toolbox PiML is a Python toolbox for developing, diagnosing, and validating inherently interpretable machine learning models such as GLMs, GAMs, de… | 29 | 1285 | maintenance |
| oscarknagg/few-shot A PyTorch repository with clean, tested implementations that reproduce few-shot learning and meta-learning research papers such as Prototyp… | 32 | 1284 | maintenance |
| YuanxunLu/LiveSpeechPortraits A PyTorch implementation of the SIGGRAPH Asia 2021 paper 'Live Speech Portraits', which generates photorealistic personalized talking-head … | 32 | 1283 | maintenance |
| iskandr/fancyimpute A Python library offering a variety of multivariate imputation and matrix completion algorithms such as KNN, SoftImpute, and NuclearNormMin… | 23 | 1282 | maintenance |
| NVIDIA-Merlin/Transformers4Rec Transformers4Rec is a Python library by NVIDIA for sequential and session-based recommendation using transformer architectures, bridging NL… | 67 | 1281 | maintenance |
| harvardnlp/seq2seq-attn A Torch (Lua) implementation of sequence-to-sequence models with LSTM encoder-decoders and optional attention for neural machine translatio… | 32 | 1280 | maintenance |
| chonyy/AI-basketball-analysis An AI-powered web app and API that analyzes basketball shots and shooting form from uploaded videos using object detection and OpenPose pos… | 32 | 1279 | maintenance |
| benedekrozemberczki/CapsGNN A PyTorch implementation of the Capsule Graph Neural Network (CapsGNN) from ICLR 2019, which uses capsule networks and routing mechanisms t… | 23 | 1278 | maintenance |
| kaiyinzhou/BERT-NER A Python implementation that fine-tunes Google's BERT for named entity recognition using the CoNLL-2003 dataset, built on TensorFlow. It pr… | 32 | 1277 | maintenance |
| SullyChen/Autopilot-TensorFlow A TensorFlow implementation of Nvidia's end-to-end self-driving steering angle prediction paper (arXiv 1604.07316) with some modifications.… | 32 | 1276 | maintenance |
| j96w/DenseFusion DenseFusion is the official PyTorch implementation of the paper '6D Object Pose Estimation by Iterative Dense Fusion', which estimates the … | 32 | 1276 | maintenance |
| YavorGIvanov/sam.cpp A pure C/C++ implementation of Meta's Segment Anything Model (SAM) for image segmentation inference, built on the ggml tensor library. It c… | 28 | 1276 | maintenance |
| dragnet-org/dragnet Dragnet is a Python library that uses machine learning models to extract the main article content, and optionally user comments, from HTML … | 36 | 1275 | maintenance |
| pfnet/pfrl PFRL is a PyTorch-based deep reinforcement learning library implementing state-of-the-art algorithms such as DQN, Rainbow, PPO, SAC, and A3… | 54 | 1274 | maintenance |
| elanmart/cbp-translate A demo application that live-translates foreign-language speech in videos into subtitles, mimicking the Cyberpunk 2077 translation effect. … | 32 | 1274 | maintenance |
| s9roll7/ebsynth_utility An AUTOMATIC1111 WebUI extension that creates stylized or edited videos by combining Stable Diffusion img2img with EbSynth keyframe propaga… | 31 | 1274 | maintenance |
| TRI-ML/packnet-sfm Official PyTorch implementation of PackNet and related self-supervised monocular depth estimation methods from Toyota Research Institute's … | 23 | 1274 | maintenance |
| ahrm/UnstableFusion UnstableFusion is a PyQt5 desktop frontend for Stable Diffusion that supports text-to-image generation, inpainting, img2img, and outpaintin… | 23 | 1271 | maintenance |
| marcbelmont/cnn-watermark-removal A TensorFlow implementation of a fully convolutional neural network that removes transparent watermark overlays from images. It trains on s… | 32 | 1270 | maintenance |
| DeepGraphLearning/graphvite GraphVite is a general, high-performance graph embedding engine written in C++ with Python bindings, supporting node embedding, knowledge g… | 32 | 1269 | maintenance |
| joel-simon/ganbreeder Ganbreeder is a collaborative web application for discovering and 'breeding' images using BigGAN models, where users mix images and share t… | 32 | 1269 | maintenance |
| TheEconomist/us-potus-model Code for The Economist's dynamic multilevel Bayesian model to forecast US presidential elections, written in R and Stan. It aggregates stat… | 32 | 1268 | maintenance |
| HendrikStrobelt/LSTMVis LSTMVis is a visual analysis toolbox for inspecting hidden state dynamics of LSTM (and other RNN) networks. It provides a browser-based int… | 23 | 1266 | maintenance |
| tensorlayer/HyperPose HyperPose is a library for building high-performance custom human pose estimation applications. It combines a C++ inference engine with Ten… | 23 | 1264 | maintenance |
| kennymckormick/pyskl PYSKL is a PyTorch-based toolbox for skeleton-based human action recognition, built on MMAction2. It provides official implementations of P… | 53 | 1263 | maintenance |
| TimoBolkart/voca VOCA is a speech-driven 3D facial animation framework that synthesizes realistic character face animations from an arbitrary speech signal … | 32 | 1263 | maintenance |
| linkedin/detext DeText is a deep neural text understanding framework from LinkedIn for NLP ranking, classification, and language generation tasks. It uses … | 23 | 1263 | maintenance |
| salesforce/pytorch-qrnn A PyTorch implementation of the Quasi-Recurrent Neural Network (QRNN) from Salesforce Research, offering LSTM-like accuracy with 2-17x fast… | 10 | 1263 | maintenance |
| openpifpaf/openpifpaf OpenPifPaf is a PyTorch library implementing Composite Fields for semantic keypoint detection and spatio-temporal association, primarily fo… | 23 | 1262 | maintenance |
| UKPLab/EasyNMT EasyNMT is a Python library providing easy access to state-of-the-art neural machine translation for 100+ languages using models like Opus-… | 23 | 1261 | maintenance |
| kuleshov/audio-super-res A Python/TensorFlow implementation of neural network models for audio super resolution, upsampling low-resolution audio to higher sampling … | 32 | 1260 | maintenance |
| google-research/planet Open-source implementation of the PlaNet agent, a purely model-based reinforcement learning algorithm that solves control tasks from pixels… | 10 | 1260 | maintenance |
| Yujun-Shi/DragDiffusion DragDiffusion is the official CVPR 2024 research code for interactive point-based image editing using pretrained diffusion models. It provi… | 19 | 1259 | maintenance |
| harvardnlp/im2markup A deep learning system (built on Torch) that converts images of rendered text into presentational markup such as LaTeX or HTML, using a CNN… | 32 | 1258 | maintenance |
| DrSleep/tensorflow-deeplab-resnet A TensorFlow re-implementation of the DeepLab-ResNet model for semantic image segmentation, trained and evaluated on the PASCAL VOC dataset… | 10 | 1258 | maintenance |
| uclaml/SPIN Official implementation of Self-Play Fine-Tuning (SPIN), a method that improves language models by having them iteratively generate and dis… | 26 | 1254 | maintenance |
| ycjuan/kaggle-2014-criteo The winning '3 Idiots' solution code for the 2014 Kaggle Criteo Display Advertising Challenge, built around field-aware factorization machi… | 10 | 1254 | maintenance |
| maciejkula/glove-python A toy Python/Cython implementation of the GloVe algorithm for training dense word vector embeddings by factorizing the log of the word co-o… | 32 | 1253 | maintenance |
| davidmrau/mixture-of-experts A PyTorch re-implementation of the sparsely-gated Mixture-of-Experts layer from the paper 'Outrageously Large Neural Networks' by Shazeer e… | 32 | 1253 | maintenance |
| GoGoDuck912/Self-Correction-Human-Parsing A deep learning toolkit for human parsing (semantic segmentation of clothing and body parts in images), with pretrained models on LIP, ATR,… | 32 | 1253 | maintenance |
| Fantasy-Studio/Paint-by-Example Paint by Example is a PyTorch implementation of exemplar-based image editing with diffusion models, letting users fill masked regions of an… | 32 | 1252 | maintenance |
| clovaai/CutMix-PyTorch Official PyTorch implementation of CutMix, an image data augmentation regularizer that cuts and pastes patches between training images whil… | 32 | 1251 | maintenance |
| ARM-software/ML-KWS-for-MCU TensorFlow models and training scripts for keyword spotting (wake-word detection) on Arm Cortex-M microcontrollers, accompanying the 'Hello… | 32 | 1250 | maintenance |
| kamalkraj/BERT-NER A PyTorch library for training and running named entity recognition (NER) models based on Google's BERT, evaluated on the CoNLL-2003 datase… | 32 | 1249 | maintenance |
| leftthomas/SRGAN A PyTorch implementation of SRGAN, the CVPR 2017 generative adversarial network for photo-realistic single-image super-resolution. It inclu… | 32 | 1249 | maintenance |
| CQFIO/PhotographicImageSynthesis A TensorFlow implementation of the ICCV 2017 paper 'Photographic Image Synthesis with Cascaded Refinement Networks', which synthesizes phot… | 32 | 1248 | maintenance |
| daijifeng001/R-FCN MATLAB implementation of R-FCN, a region-based fully convolutional object detection framework described in a NIPS 2016 paper. It builds on … | 32 | 1248 | maintenance |
| RedaOps/ann-visualizer A Python library that generates presentable graph diagrams of Keras neural network architectures using Graphviz. It supports visualizing bo… | 10 | 1248 | maintenance |
| XiangLi1999/Diffusion-LM Diffusion-LM is the official research code for the paper 'Diffusion-LM Improves Controllable Text Generation', implementing a diffusion-bas… | 32 | 1246 | maintenance |
| KMnP/vpt Official PyTorch implementation of Visual Prompt Tuning (VPT), an ECCV 2022 method for parameter-efficient fine-tuning of vision transforme… | 32 | 1244 | maintenance |
| NeymarL/ChineseChess-AlphaZero An open-source implementation of AlphaZero/AlphaGo Zero reinforcement learning methods applied to Chinese chess (Xiangqi). It includes self… | 23 | 1244 | maintenance |
| mravanelli/SincNet SincNet is a PyTorch neural architecture that processes raw audio waveforms using parametrized sinc band-pass filters in the first convolut… | 32 | 1243 | maintenance |
| SUC-DriverOld/MSST-WebUI A WebUI application for running inference with Music-Source-Separation-Training (MSST) and UVR/VR models, with preset processing flows, mod… | 69 | 1242 | maintenance |
| tf-encrypted/tf-encrypted TF Encrypted is a Python framework for privacy-preserving machine learning in TensorFlow, enabling training and prediction on encrypted dat… | 32 | 1242 | maintenance |
| ShannonAI/service-streamer Service Streamer is a Python middleware that queues discrete web service requests into mini-batches for deep learning model inference, impr… | 32 | 1241 | maintenance |
| lfz/DSB2017 The winning solution of team 'grt123' for the 2017 Data Science Bowl (DSB2017), a deep learning pipeline for detecting lung cancer from CT … | 32 | 1241 | maintenance |
| AlgoTraders/stock-analysis-engine A distributed stock analysis and backtesting framework that ingests automated pricing data from IEX Cloud, Tradier, and FinViz and runs tho… | 32 | 1241 | maintenance |
| ranahanocka/point2mesh Point2Mesh is a PyTorch implementation of a SIGGRAPH 2020 technique that reconstructs watertight surface meshes from input point clouds by … | 32 | 1240 | maintenance |
| VeritasYin/STGCN_IJCAI-18 Reference implementation of Spatio-Temporal Graph Convolutional Networks (STGCN) from the IJCAI 2018 paper, a deep learning framework for t… | 71 | 1239 | maintenance |
| autonomousvision/giraffe Official PyTorch implementation of GIRAFFE, a CVPR 2021 (oral, best paper award) generative model that represents scenes as compositional g… | 32 | 1238 | maintenance |
| BR-IDL/PaddleViT PaddleViT is a collection of state-of-the-art Vision Transformer and MLP model implementations for PaddlePaddle 2.1+, covering image classi… | 23 | 1238 | maintenance |
| otaha178/Emotion-recognition A Python application that performs real-time facial emotion recognition from a webcam feed using a convolutional neural network. It display… | 32 | 1237 | maintenance |
| openseg-group/openseg.pytorch Official PyTorch implementations of semantic segmentation models OCNet, OCRNet, and SegFix, achieving state-of-the-art results on benchmark… | 23 | 1237 | maintenance |
| facebookresearch/simsiam A PyTorch implementation of SimSiam, a simple Siamese-based self-supervised representation learning method from Facebook AI Research (CVPR … | 10 | 1237 | maintenance |
| jfzhang95/pytorch-video-recognition A PyTorch library implementing C3D, R3D, and R2Plus1D models for video action recognition, with training scripts for UCF101 and HMDB51 data… | 32 | 1236 | maintenance |
| ultralytics/JSON2YOLO A legacy Python toolkit that converts JSON-format annotation datasets (COCO, LabelMe, Labelbox, VoTT, INFOLKS, ATH) into the YOLO format fo… | 67 | 1235 | maintenance |
| jakesnell/prototypical-networks Reference PyTorch implementation of Prototypical Networks for few-shot classification from the NeurIPS 2017 paper. It includes training and… | 32 | 1234 | maintenance |
| AGI-Edgerunners/LLM-Adapters LLM-Adapters is a Python framework extending HuggingFace's PEFT library that integrates various adapter types (LoRA, series/parallel adapte… | 30 | 1234 | maintenance |
| cmu-db/ottertune OtterTune is a tool from the CMU Database Group that automatically finds good settings for a DBMS's configuration knobs using machine learn… | 10 | 1233 | maintenance |
| pollen-robotics/dtw A small Python module implementing Dynamic Time Warping (DTW), a similarity measure between temporal sequences. It offers a basic pure-Pyth… | 23 | 1232 | maintenance |
| qubvel/classification_models A Keras/TensorFlow Keras library providing ImageNet-pretrained image classification model architectures such as ResNet, SE-ResNet, DenseNet… | 23 | 1231 | maintenance |
| linkedin/luminol Luminol is a lightweight Python library for time series analysis that provides anomaly detection and correlation between time series. It as… | 49 | 1230 | maintenance |
| yuanxiaosc/Entity-Relation-Extraction A TensorFlow and BERT based pipeline for joint entity and relation extraction, built as a solution for the 2019 Language and Intelligence C… | 32 | 1230 | maintenance |
| shobrook/BitVision BitVision is a terminal-based real-time charting and trading dashboard for the Bitstamp Bitcoin exchange, built with Node.js and Python. It… | 32 | 1229 | maintenance |