domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| open-mmlab/mmrazor MMRazor is OpenMMLab's model compression toolbox and benchmark built on PyTorch. It provides implementations of neural architecture search,… | 23 | 1682 | maintenance |
| karpathy/arxiv-sanity-lite A lightweight web application that periodically polls the arXiv API for new papers and lets users tag papers of interest. It recommends sim… | 32 | 1681 | maintenance |
| microsoft/EdgeML A library from Microsoft Research India implementing resource-efficient machine learning algorithms (Bonsai, ProtoNN, FastGRNN, EMI-RNN, DR… | 23 | 1681 | maintenance |
| CuriousAI/mean-teacher Reference implementations (TensorFlow and PyTorch) of the Mean Teacher semi-supervised learning method from the NIPS 2017 paper by Tarvaine… | 32 | 1678 | maintenance |
| rockyzhengwu/FoolNLTK FoolNLTK is a Python toolkit for Chinese natural language processing built on a BiLSTM model. It provides high-accuracy word segmentation, … | 32 | 1678 | maintenance |
| mit-han-lab/smoothquant SmoothQuant is a training-free post-training quantization (PTQ) method that enables W8A8 (8-bit weight, 8-bit activation) quantization for … | 32 | 1677 | maintenance |
| Lam1360/YOLOv3-model-pruning A PyTorch implementation of YOLOv3 channel pruning (network slimming) applied to hand detection on the Oxford Hand dataset. It provides spa… | 32 | 1676 | maintenance |
| argusswift/YOLOv4-pytorch A PyTorch re-implementation of YOLOv4 object detection with variants including attentive YOLOv4 (SEnet, CBAM, CoordAttention) and MobileNet… | 23 | 1676 | maintenance |
| lovoo/NSFWDetector A tiny 17 kB CoreML model and Swift library that scans images for nudity/porn content, returning a confidence score. Distributed via CocoaP… | 23 | 1676 | maintenance |
| YuliangXiu/ICON ICON is a PyTorch research implementation of a CVPR 2022 method that reconstructs detailed, animatable 3D clothed human avatars from 2D ima… | 23 | 1675 | maintenance |
| Kaixhin/Rainbow A PyTorch implementation of the Rainbow deep reinforcement learning agent, combining DQN improvements like prioritized experience replay, d… | 23 | 1673 | maintenance |
| uber/manifold Manifold is a model-agnostic visual debugging tool for machine learning from Uber. It helps ML practitioners identify which subsets of data… | 25 | 1672 | maintenance |
| topepo/caret caret (Classification And Regression Training) is an R package providing a unified interface for training, tuning, and evaluating classific… | 67 | 1669 | maintenance |
| plasticityai/magnitude Magnitude is a Python package and file format for fast, memory-efficient loading and querying of vector embeddings like word2vec, GloVe, an… | 23 | 1668 | maintenance |
| uber-research/deep-neuroevolution Distributed implementations of deep neuroevolution algorithms (ES, NS-ES, NSR-ES, DeepGA, Random Search) from Uber AI Labs research papers,… | 32 | 1667 | maintenance |
| natanielruiz/deep-head-pose Hopenet is a PyTorch deep learning model for fine-grained head pose estimation from images and video, without requiring facial keypoints. I… | 32 | 1666 | maintenance |
| datamllab/tods TODS is a full-stack automated machine learning system for outlier detection on multivariate time-series data, developed by DATA Lab at Ric… | 32 | 1666 | maintenance |
| akanazawa/hmr HMR (Human Mesh Recovery) is a TensorFlow implementation of the CVPR 2018 paper 'End-to-end Recovery of Human Shape and Pose', which regres… | 32 | 1666 | maintenance |
| autonomousvision/occupancy_networks Official PyTorch implementation of the CVPR 2019 paper 'Occupancy Networks: Learning 3D Reconstruction in Function Space'. It learns contin… | 32 | 1663 | maintenance |
| hfawaz/dl-4-tsc Companion repository for the paper 'Deep learning for time series classification: a review', providing TensorFlow 2 implementations of nine… | 32 | 1659 | maintenance |
| google-research/pegasus PEGASUS is Google Research's implementation of transformer encoder-decoder models pre-trained with the Gap Sentences Generation objective f… | 10 | 1655 | maintenance |
| rwightman/efficientdet-pytorch A PyTorch implementation of EfficientDet object detection, faithful to the original Google TensorFlow implementation with ported pretrained… | 23 | 1654 | maintenance |
| HonglinChu/SiamTrackers A PyTorch collection of Siamese-based visual object tracking models including SiamFC, SiamRPN++, SiamMask, Ocean, LightTrack, and the light… | 32 | 1653 | maintenance |
| jasonwei20/eda_nlp EDA (Easy Data Augmentation) is a Python library implementing four simple text augmentation operations—synonym replacement, random insertio… | 32 | 1652 | maintenance |
| benhamner/Metrics A multi-language library providing implementations of supervised machine learning evaluation metrics such as AUC, log loss, MAE, RMSE, and … | 32 | 1651 | maintenance |
| facebookresearch/CrypTen CrypTen is a PyTorch-based library for privacy-preserving machine learning built on secure multiparty computation. It exposes a CrypTensor … | 10 | 1649 | maintenance |
| vlfeat/vlfeat VLFeat is an open-source C library of popular computer vision algorithms specializing in image understanding and local feature extraction a… | 32 | 1648 | maintenance |
| invictus717/MetaTransformer Meta-Transformer is a research framework for unified multimodal learning that maps inputs from 12 modalities (text, images, point clouds, a… | 19 | 1647 | maintenance |
| amir-abdi/keras_to_tensorflow A Python CLI tool that converts trained Keras models (.h5 or .h5 + .) into frozen, inference-ready TensorFlow protobuf (.pb) files. It free… | 32 | 1646 | maintenance |
| kan-bayashi/ParallelWaveGAN Unofficial PyTorch implementations of non-autoregressive neural vocoders including Parallel WaveGAN, MelGAN, Multi-band MelGAN, HiFi-GAN, a… | 23 | 1645 | maintenance |
| gnobitab/RectifiedFlow Official PyTorch implementation of Rectified Flow, an ICLR 2023 Spotlight method for learning transport maps between distributions via stra… | 32 | 1644 | maintenance |
| ai-forever/ru-dalle ruDALL-E is a Python library for generating images from Russian text prompts using DALL-E-style transformer models like Malevich, Emojich, … | 23 | 1644 | maintenance |
| Coder-Yu/QRec QRec is a Python framework built on TensorFlow 1.x for quickly implementing and evaluating recommender system models. It ships with many st… | 32 | 1638 | maintenance |
| autonomio/talos Talos is a Python library for automating hyperparameter experiments and model evaluation with TensorFlow (tf.keras), Keras, and PyTorch mod… | 23 | 1636 | maintenance |
| facebookresearch/consistent_depth A research library from Facebook AI Research implementing Consistent Video Depth Estimation (SIGGRAPH 2020). It reconstructs dense, flicker… | 10 | 1634 | maintenance |
| tensorflow/mesh Mesh TensorFlow is a Python library and embedded language for specifying distributed tensor computations, letting users define how model di… | 10 | 1630 | maintenance |
| facebookresearch/pytorch_GAN_zoo A PyTorch toolbox of GAN implementations including Progressive Growing GAN, DCGAN, and StyleGAN (beta), plus tools like GDPP, AC-GAN condit… | 10 | 1627 | maintenance |
| facebookresearch/higher higher is a PyTorch library that enables higher-order gradients over losses spanning entire training loops rather than individual steps. It… | 10 | 1626 | maintenance |
| moses-smt/mosesdecoder Moses is a statistical machine translation (SMT) system that trains translation models from parallel corpora and decodes with efficient phr… | 29 | 1624 | maintenance |
| timoschick/pet Official implementation of Pattern-Exploiting Training (PET), a semi-supervised method that reformulates text classification and natural la… | 32 | 1622 | maintenance |
| Xtra-Computing/thundersvm ThunderSVM is a fast SVM library that accelerates support vector machine training on GPUs (via CUDA) and multi-core CPUs. It supports all L… | 23 | 1621 | maintenance |
| wangzhaode/mnn-llm mnn-llm is a C++ library for deploying large language models on-device using Alibaba's MNN inference engine, with demos for CLI, web, Andro… | 22 | 1616 | maintenance |
| dotnet/infer Infer.NET is a .NET framework for running Bayesian inference in graphical models, also usable for probabilistic programming. It compiles mo… | 74 | 1615 | maintenance |
| cdipaolo/goml goml is a machine learning library written entirely in Go, offering both traditional batch learning and online, stream-based learning via c… | 32 | 1614 | maintenance |
| allenai/bilm-tf A TensorFlow implementation of the bidirectional language model (biLM) used to compute ELMo deep contextualized word representations. It su… | 32 | 1612 | maintenance |
| experiencor/keras-yolo3 A Keras/TensorFlow implementation of YOLOv3 for object detection, supporting detection with pretrained weights, custom model training with … | 32 | 1608 | maintenance |
| Delta-ML/delta DELTA is a deep learning based end-to-end natural language and speech processing platform built on TensorFlow and Python 3. It provides one… | 10 | 1607 | maintenance |
| ialhashim/DenseDepth Official Keras/TensorFlow implementation (with experimental PyTorch and TF2 code) of the DenseDepth paper for high-quality monocular depth … | 32 | 1606 | maintenance |
| chandrikadeb7/Face-Mask-Detection A face mask detection system built with OpenCV and TensorFlow/Keras that uses deep learning (SSD MobileNetV2) to detect whether people are … | 23 | 1606 | maintenance |
| tensorflow/haskell Haskell bindings to the TensorFlow machine learning library, providing a typed API for building and training neural network models. It incl… | 10 | 1604 | maintenance |
| ycjuan/libffm LIBFFM is a C++ library implementing field-aware factorization machines (FFM), a model especially effective for click-through-rate (CTR) pr… | 10 | 1599 | maintenance |
| tmikolov/word2vec The original C implementation of word2vec, which trains word embeddings using the Continuous Bag-of-Words (CBOW) and Skip-gram neural netwo… | 32 | 1592 | maintenance |
| google-research/parti Parti is Google Research's Pathways Autoregressive Text-to-Image model, which generates high-fidelity photorealistic images from text by tr… | 32 | 1589 | maintenance |
| google/uis-rnn A Python library implementing the Unbounded Interleaved-State Recurrent Neural Network (UIS-RNN) algorithm for segmenting and clustering se… | 10 | 1588 | maintenance |
| lufficc/SSD A high-quality, fast, modular reference implementation of the SSD (Single Shot MultiBox Detector) object detection model in PyTorch. It sup… | 23 | 1585 | maintenance |
| PaddlePaddle/PGL Paddle Graph Learning (PGL) is an efficient and flexible graph learning framework built on PaddlePaddle. It supports heterogeneous graph le… | 23 | 1585 | maintenance |
| PrithivirajDamodaran/Gramformer Gramformer is a Python library that exposes interfaces to detect, highlight, and correct grammatical errors in natural language text, built… | 23 | 1584 | maintenance |
| jundongl/scikit-feature scikit-feature is an open-source Python feature selection library built on scikit-learn, NumPy, and SciPy. It implements around 40 popular … | 32 | 1583 | maintenance |
| D-X-Y/AutoDL-Projects AutoDL-Projects is a PyTorch library implementing automated deep learning algorithms, including neural architecture search (NAS) methods li… | 32 | 1581 | maintenance |
| melodyguan/enas Authors' TensorFlow implementation of Efficient Neural Architecture Search (ENAS), which discovers neural network architectures via paramet… | 32 | 1578 | maintenance |
| Temporal Segment Networks (TSN) Official code and pretrained models for Temporal Segment Networks (TSN), a deep learning framework for video action recognition published a… | 32 | 1577 | maintenance |
| maxpumperla/elephas Elephas is a Python library that extends Keras to run distributed deep learning training on Apache Spark. It serializes Keras models on the… | 23 | 1577 | maintenance |
| Kotlin/kotlindl KotlinDL is a high-level deep learning framework written in Kotlin and inspired by Keras, built on top of the TensorFlow Java API and ONNX … | 23 | 1576 | maintenance |
| vturrisi/solo-learn solo-learn is a Python library of state-of-the-art self-supervised methods for unsupervised visual representation learning, built on PyTorc… | 65 | 1573 | maintenance |
| cmdbug/YOLOv5_NCNN A mobile demo application that deploys the ncnn inference framework on Android and iOS, running a variety of computer vision models includi… | 32 | 1572 | maintenance |
| HIPS/Spearmint Spearmint is a Python package for Bayesian optimization that automatically tunes experiment parameters to minimize an objective in as few r… | 32 | 1572 | maintenance |
| kumar-shridhar/PyTorch-BayesianCNN A PyTorch library implementing Bayesian convolutional neural networks with variational inference via Bayes by Backprop. It provides drop-in… | 32 | 1570 | maintenance |
| AlexHex7/Non-local_pytorch A PyTorch implementation of the Non-local Neural Block from the paper 'Non-local Neural Networks', providing multiple variants (concatenati… | 32 | 1564 | maintenance |
| tobegit3hub/advisor Advisor is an open-source implementation of Google Vizier, a hyperparameter tuning system for black-box optimization. It provides a server … | 32 | 1563 | maintenance |
| NVlabs/noise2noise Official TensorFlow implementation of the Noise2Noise ICML 2018 paper, which trains image restoration networks using only corrupted (noisy)… | 32 | 1563 | maintenance |
| facebookresearch/DeepSDF DeepSDF is Facebook Research's official implementation of the CVPR 2019 paper on learning continuous signed distance functions for 3D shape… | 10 | 1563 | maintenance |
| msracver/FCIS FCIS is the official MXNet implementation of the CVPR 2017 paper 'Fully Convolutional Instance-aware Semantic Segmentation', which won firs… | 32 | 1561 | maintenance |
| QingyongHu/RandLA-Net Official TensorFlow implementation of RandLA-Net, a neural architecture for efficient semantic segmentation of large-scale 3D point clouds,… | 32 | 1560 | maintenance |
| bfelbo/DeepMoji DeepMoji is a deep learning model trained on 1.2 billion tweets with emojis to understand emotional language, with code for scoring texts, … | 32 | 1554 | maintenance |
| vlawhern/arl-eegmodels A collection of well-validated convolutional neural network models (EEGNet, DeepConvNet, ShallowConvNet) for EEG signal classification, imp… | 32 | 1552 | maintenance |
| vt-vl-lab/FGVC FGVC is a PyTorch implementation of the ECCV 2020 paper 'Flow-edge Guided Video Completion'. It completes missing regions in videos by extr… | 32 | 1551 | maintenance |
| pytorch/QNNPACK QNNPACK is a mobile-optimized C library of high-performance kernels for 8-bit quantized neural network operators such as convolution, pooli… | 10 | 1551 | maintenance |
| DevashishPrasad/CascadeTabNet CascadeTabNet is a PyTorch/mmdetection implementation of a CVPR 2020 paper for end-to-end table detection and structure recognition from im… | 32 | 1549 | maintenance |
| Tencent/TurboTransformers A fast, user-friendly C++/Python runtime for transformer model inference (BERT, ALBERT, RoBERTa, GPT2, decoders) on CPU and GPU. It works a… | 37 | 1548 | maintenance |
| JiaRenChang/PSMNet PSMNet is the official PyTorch implementation of the CVPR 2018 paper 'Pyramid Stereo Matching Network' for stereo depth estimation. It uses… | 32 | 1548 | maintenance |
| caogang/wgan-gp A PyTorch implementation of the 'Improved Training of Wasserstein GANs' (WGAN-GP) paper, covering toy datasets, MNIST, CIFAR-10, and charac… | 32 | 1547 | maintenance |
| Javacr/PyQt5-YOLOv5 A desktop GUI application built with PyQt5 that wraps YOLOv5 (v6.1) object detection models. It supports running detection on images, video… | 32 | 1547 | maintenance |
| YoYo000/MVSNet MVSNet is a deep learning architecture for depth map inference from unstructured multi-view images, and R-MVSNet is its recurrent extension… | 32 | 1546 | maintenance |
| JingyunLiang/VRT VRT is the official PyTorch implementation of the paper 'VRT: A Video Restoration Transformer', a transformer-based model for video restora… | 23 | 1546 | maintenance |
| google-research/big_transfer Official repository for the Big Transfer (BiT) paper, providing ResNet models pre-trained on ImageNet and ImageNet-21k for transfer learnin… | 10 | 1541 | maintenance |
| dandelin/ViLT Official PyTorch code for the ICML 2021 paper ViLT, a vision-and-language transformer that performs multimodal pre-training without convolu… | 23 | 1538 | maintenance |
| AxeldeRomblay/MLBox MLBox is a Python automated machine learning (AutoML) library that handles data preprocessing, feature selection, leak detection, and hyper… | 23 | 1536 | maintenance |
| gpleiss/efficient_densenet_pytorch A memory-efficient PyTorch implementation of DenseNets that uses gradient checkpointing to reduce feature map memory consumption from quadr… | 32 | 1535 | maintenance |
| poloclub/ganlab GAN Lab is an interactive, browser-based visualization tool for learning and experimenting with Generative Adversarial Networks. Built with… | 56 | 1534 | maintenance |
| AsuharietYgvar/AppleNeuralHash2ONNX A Python tool that converts Apple's NeuralHash perceptual hashing model (used in Apple's CSAM Detection system) into ONNX format. It includ… | 32 | 1534 | maintenance |
| yongzhuo/nlp_xiaojiang A Chinese natural language processing toolkit covering retrieval-based chatbots, text classification, NER (BERT+BiLSTM+CRF), sentence embed… | 32 | 1534 | maintenance |
| bes-dev/stable_diffusion.openvino A Python CLI implementation of Stable Diffusion text-to-image generation optimized for Intel CPUs and GPUs via OpenVINO. It supports text-t… | 32 | 1533 | maintenance |
| crownpku/Rasa_NLU_Chi A Chinese-language fork of Rasa NLU that turns Chinese natural language into structured intent and entity data using MITIE, Jieba tokenizat… | 32 | 1531 | maintenance |
| Stability-AI/sd3.5 A minimal inference-only reference implementation of Stable Diffusion 3.5 and SD3, including text encoders, VAE decoder, MM-DiT core, and S… | 22 | 1529 | maintenance |
| yu4u/age-gender-estimation A Keras/TensorFlow implementation of a convolutional neural network that estimates age and gender from face images, trained on the IMDB-WIK… | 23 | 1520 | maintenance |
| zhan-xu/RigNet RigNet is a PyTorch implementation of the SIGGRAPH 2020 paper on neural rigging for articulated 3D characters. It takes a character mesh as… | 32 | 1519 | maintenance |
| shaoxiongji/federated-learning A PyTorch implementation of the FedAvg federated learning algorithm reproducing the McMahan et al. AISTATS 2017 paper. It includes experime… | 32 | 1515 | maintenance |
| tanluren/yolov3-channel-and-layer-pruning A Python toolkit built on ultralytics/yolov3 that implements channel pruning, layer pruning, and knowledge distillation for YOLOv3/v4 (incl… | 32 | 1515 | maintenance |
| IBM/pytorch-seq2seq An open-source framework for building sequence-to-sequence (seq2seq) models in PyTorch, with modular and extensible components for models, … | 10 | 1515 | maintenance |
| megvii-model/ShuffleNet-Series A collection of ShuffleNet-series efficient convolutional neural network models (V1, V2, V2+, Large, ExLarge) plus NAS-derived backbones li… | 32 | 1513 | maintenance |