domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| TorchCraft/TorchCraft TorchCraft is a bridge library connecting the Torch machine learning framework to StarCraft: Brood War via a BWAPI module that streams game… | 10 | 1398 | maintenance |
| safety-research/bloom Bloom is a Python tool that automatically generates behavioral evaluation suites for LLMs, probing target models for behaviors like sycopha… | 55 | 1396 | maintenance |
| JasperSnoek/spearmint Spearmint is a Python package for Bayesian optimization of machine learning algorithms, iteratively tuning parameters to minimize an object… | 32 | 1392 | maintenance |
| innnky/emotional-vits Emotional VITS is a voice synthesis (TTS) model based on VITS that enables emotion-controllable speech generation without requiring manual … | 32 | 1391 | maintenance |
| Maluuba/nlg-eval A Python library and CLI tool that computes unsupervised automated metrics (BLEU, METEOR, ROUGE, CIDEr, SPICE, SkipThought, and embedding-b… | 23 | 1391 | maintenance |
| 198808xc/Pangu-Weather Official implementation of Pangu-Weather, a deep learning model for fast, accurate medium-range global weather forecasting using 3D neural … | 22 | 1390 | maintenance |
| facebookresearch/LAMA LAMA is a research probe for analyzing the factual and commonsense knowledge stored in pretrained language models such as BERT, RoBERTa, GP… | 10 | 1390 | maintenance |
| yao8839836/text_gcn The official implementation of Text GCN, a Graph Convolutional Network approach for text classification published at AAAI 2019. It builds w… | 32 | 1389 | maintenance |
| yinboc/liif LIIF is the official PyTorch implementation of the CVPR 2021 paper 'Learning Continuous Image Representation with Local Implicit Image Func… | 32 | 1389 | maintenance |
| chrisdonahue/wavegan Official TensorFlow implementation of WaveGAN, a generative adversarial network that learns to synthesize raw audio waveforms from examples… | 32 | 1383 | maintenance |
| DeepGraphLearning/KnowledgeGraphEmbedding A PyTorch toolkit implementing the RotatE knowledge graph embedding model along with TransE, ComplEx, DistMult, and pRotatE. It trains and … | 32 | 1383 | maintenance |
| bytedance/music_source_separation A PyTorch implementation of music source separation that splits songs into sources like vocals and accompaniment. It ships pretrained check… | 10 | 1383 | maintenance |
| zhanlaoban/EDA_NLP_for_Chinese A Python implementation of the EDA (Easy Data Augmentation) paper adapted for Chinese text corpora. It augments labeled text classification… | 32 | 1382 | maintenance |
| wouterkool/attention-learn-to-route A PyTorch implementation of the attention-based neural model from the ICLR 2019 paper 'Attention, Learn to Solve Routing Problems!', traine… | 32 | 1382 | maintenance |
| jrieke/traingenerator Traingenerator is a Streamlit-based web app that generates custom template code for machine learning projects with PyTorch and scikit-learn… | 32 | 1381 | maintenance |
| amazon-science/patchcore-inspection Official implementation of PatchCore, a deep-learning method for industrial image anomaly detection and localization from Roth et al. (2021… | 32 | 1379 | maintenance |
| PSPNet Reference implementation of the Pyramid Scene Parsing Network (PSPNet), a CVPR 2017 semantic segmentation model that won the ImageNet Scene… | 32 | 1378 | maintenance |
| tensorflow/ecosystem A collection of templates and connectors for integrating TensorFlow with other open-source frameworks such as Kubernetes, Kubeflow, Spark, … | 10 | 1377 | maintenance |
| takuya-takeuchi/FaceRecognitionDotNet A C# port of the popular face_recognition library providing a simple facial recognition API for .NET. It supports face detection, recogniti… | 23 | 1376 | maintenance |
| tinkoff-ai/CORL CORL is a research-oriented Python library providing thoroughly benchmarked single-file implementations of state-of-the-art offline and off… | 10 | 1373 | maintenance |
| tensorboy/pytorch_Realtime_Multi-Person_Pose_Estimation A PyTorch implementation of the CVPR'17 Realtime Multi-Person 2D Pose Estimation (OpenPose/rtpose) model. It provides pretrained weights, d… | 32 | 1371 | maintenance |
| Cysu/open-reid Open-ReID is a lightweight Python library for person re-identification built on PyTorch, providing a uniform dataset interface, models, and… | 32 | 1370 | maintenance |
| zalandoresearch/pytorch-ts PyTorchTS is a PyTorch-based probabilistic time series forecasting library that uses GluonTS as its backend API for loading, transforming, … | 23 | 1370 | maintenance |
| tensor-compiler/taco TACO (The Tensor Algebra Compiler) is a C++ library that computes sparse and dense tensor algebra and linear algebra expressions, generatin… | 30 | 1367 | maintenance |
| mdeff/cnn_graph Reference implementation of the NIPS 2016 paper 'Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering' (ChebNet),… | 32 | 1366 | maintenance |
| JonasGeiping/cramming A research framework for pretraining BERT-type language models from scratch on a single consumer GPU in one day, replicating the paper 'Cra… | 22 | 1366 | maintenance |
| hiyouga/FastEdit FastEdit is a Python library and CLI tool for editing large language models, injecting fresh or customized factual knowledge into pretraine… | 19 | 1366 | maintenance |
| BorealisAI/advertorch AdverTorch is a Python toolbox for adversarial robustness research built on PyTorch. It provides modules for generating adversarial perturb… | 32 | 1364 | maintenance |
| taoxugit/AttnGAN A PyTorch implementation of AttnGAN, a fine-grained text-to-image generative adversarial network with multi-stage attention mechanisms from… | 32 | 1363 | maintenance |
| sail-sg/poolformer PoolFormer is a PyTorch implementation of the CVPR 2022 paper 'MetaFormer Is Actually What You Need for Vision', which replaces attention i… | 23 | 1363 | maintenance |
| nvidia-cosmos/cosmos-predict2.5 NVIDIA Cosmos-Predict2.5 is a family of world foundation models (WFMs) that generate video predictions of future world states for physical … | 72 | 1362 | maintenance |
| atulapra/Emotion-detection A Python application that performs real-time facial emotion detection from a webcam feed using a CNN trained on the FER-2013 dataset. It us… | 32 | 1359 | maintenance |
| LetheSec/HuggingFace-Download-Accelerator A Python wrapper script around huggingface-cli and hf_transfer that enables high-speed downloads of HuggingFace models and datasets, with b… | 27 | 1359 | maintenance |
| davidcelis/recommendable Recommendable is a Ruby gem that adds a like/dislike-based recommendation engine to Ruby applications using Jaccardian similarity and memor… | 23 | 1354 | maintenance |
| bumble-tech/private-detector Bumble's Private Detector is a pretrained TensorFlow image classifier based on EfficientNet-v2 that detects lewd images. The repo provides … | 23 | 1352 | maintenance |
| ShuLiu1993/PANet A PyTorch re-implementation of PANet (Path Aggregation Network), the CVPR 2018 paper that won 1st place in the COCO 2017 Instance Segmentat… | 32 | 1347 | maintenance |
| Replicable-MARL/MARLlib MARLlib is a unified multi-agent reinforcement learning library built on Ray and RLlib with PyTorch, providing implementations of many MARL… | 23 | 1345 | maintenance |
| microsoft/X-Decoder Official PyTorch implementation of X-Decoder, a generalized decoding model from CVPR 2023 that unifies pixel-level segmentation, image-leve… | 22 | 1345 | maintenance |
| dorarad/gansformer GANformer is a research implementation of a generative adversarial transformer that uses a bipartite attention structure for efficient high… | 23 | 1344 | maintenance |
| PeizeSun/SparseR-CNN Sparse R-CNN is a PyTorch implementation (built on Detectron2) of the CVPR 2021 / PAMI 2023 paper 'End-to-End Object Detection with Learnab… | 23 | 1343 | maintenance |
| ebhy/budgetml BudgetML is a Python library that deploys machine learning models as secured HTTPS API endpoints on cheap GCP preemptible instances in unde… | 23 | 1343 | maintenance |
| HobbitLong/CMC Official PyTorch implementation of Contrastive Multiview Coding (CMC), a self-supervised visual representation learning method that contras… | 32 | 1340 | maintenance |
| timojl/clipseg CLIPSeg is a Python implementation of the CVPR 2022 paper 'Image Segmentation Using Text and Image Prompts', enabling zero-shot segmentatio… | 32 | 1338 | maintenance |
| rbgirshick/yacs YACS (Yet Another Configuration System) is a lightweight Python library for defining and managing system configurations, originally built f… | 23 | 1338 | maintenance |
| ikostrikov/pytorch-a3c A PyTorch implementation of the Asynchronous Advantage Actor Critic (A3C) reinforcement learning algorithm from the DeepMind paper 'Asynchr… | 32 | 1336 | maintenance |
| ahmetozlu/tensorflow_object_counting_api An open-source framework built on TensorFlow and Keras that simplifies developing object counting systems. It supports cumulative counting,… | 23 | 1334 | maintenance |
| GuitarML/SmartGuitarAmp SmartGuitarAmp is a JUCE-based audio plugin (VST3/AU/AAX) that uses neural network models (WaveNet-style) to emulate a real tube guitar amp… | 23 | 1331 | maintenance |
| kaz-Anova/StackNet StackNet is a Java-based meta-modelling framework that implements Wolpert's stacked generalization across multiple layers, resembling a fee… | 32 | 1326 | maintenance |
| google/sg2im A PyTorch research implementation of the CVPR 2018 paper 'Image Generation from Scene Graphs' by Johnson et al. It converts a structured sc… | 10 | 1325 | maintenance |
| CharlesShang/DCNv2 A PyTorch implementation of Deformable Convolutional Networks v2 (DCNv2), providing CUDA/CPU operators for deformable convolution as a drop… | 71 | 1323 | maintenance |
| facebookresearch/moco-v3 A PyTorch implementation of MoCo v3, a self-supervised contrastive learning method for ResNet and Vision Transformer (ViT) models. It inclu… | 10 | 1322 | maintenance |
| WisconsinAIVision/yolact_edge YolactEdge is a PyTorch implementation of a real-time instance segmentation model optimized for edge devices like the NVIDIA Jetson AGX Xav… | 32 | 1321 | maintenance |
| CSTR-Edinburgh/merlin Merlin is a toolkit from the University of Edinburgh's CSTR for building deep neural network models for statistical parametric speech synth… | 23 | 1321 | maintenance |
| YelpArchive/MOE MOE (Metric Optimization Engine) is a global, black box optimization engine built by Yelp for optimizing expensive, time-consuming objectiv… | 10 | 1321 | maintenance |
| msracver/Deep-Feature-Flow Official MXNet implementation of Deep Feature Flow (CVPR 2017), an end-to-end framework for video recognition such as object detection and … | 32 | 1315 | maintenance |
| szagoruyko/wide-residual-networks Reference implementation of Wide Residual Networks (WRN), a ResNet variant that trades depth for width to train faster and achieve state-of… | 32 | 1315 | maintenance |
| lda-project/lda A Python library implementing latent Dirichlet allocation (LDA) topic modeling via collapsed Gibbs sampling, with a scikit-learn-style API.… | 32 | 1314 | maintenance |
| facebookresearch/svoice SVoice is a PyTorch implementation of the ICML paper 'Voice Separation with an Unknown Number of Multiple Speakers' from Facebook AI Resear… | 10 | 1314 | maintenance |
| Renovamen/Speech-Emotion-Recognition A Python library implementing speech emotion recognition with Keras/TensorFlow 2 using LSTM, CNN, SVM, and MLP models. It extracts audio fe… | 32 | 1312 | maintenance |
| EricGuo5513/momask-codes Official PyTorch implementation of MoMask (CVPR 2024), a masked modeling framework that generates 3D human motion sequences from text promp… | 27 | 1312 | maintenance |
| d-li14/involution Official PyTorch implementation of the involution neural operator from the CVPR 2021 paper 'Involution: Inverting the Inherence of Convolut… | 32 | 1310 | maintenance |
| tysam-code/hlb-CIFAR10 A single-file PyTorch implementation that trains a neural network to 94% accuracy on CIFAR-10 in under 6.3 seconds on a single A100 GPU, fo… | 22 | 1310 | maintenance |
| NVlabs/prismer Official PyTorch implementation of Prismer, a data- and parameter-efficient vision-language model that ensembles pre-trained task-specific … | 30 | 1309 | maintenance |
| 920232796/bert_seq2seq A lightweight PyTorch framework for fine-tuning pretrained language models (BERT, RoBERTa, Nezha, GPT2, T5, BART) on Chinese NLP tasks usin… | 32 | 1308 | maintenance |
| chickenbestlover/RNN-Time-series-Anomaly-Detection A PyTorch implementation of an RNN-based time-series anomaly detector using a two-stage strategy of multi-step prediction followed by anoma… | 32 | 1305 | maintenance |
| chengzeyi/stable-fast Stable Fast is an ultra lightweight performance optimization framework for HuggingFace Diffusers pipelines on NVIDIA GPUs. It achieves SOTA… | 24 | 1304 | maintenance |
| kakaobrain/pororo PORORO is a Python library from Kakao Brain that provides unified access to dozens of pretrained neural models for natural language process… | 10 | 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 |
| 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 |
| 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 |
| 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 |
| 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 |
| 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 |
| snap-research/articulated-animation Official research code for the CVPR 2021 paper 'Motion Representations for Articulated Animation' by Snap Research. It animates a static so… | 43 | 1277 | 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 |
| 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 |