domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| open-mmlab/mmrotate MMRotate is an open-source PyTorch toolbox for rotated object detection, part of the OpenMMLab project. It provides modular components, mul… | 23 | 2163 | maintenance |
| SheffieldML/GPy GPy is a Python framework for Gaussian process modeling, providing kernels, regression, classification, and sparse GP methods. It is a rese… | 97 | 2161 | maintenance |
| bubbliiiing/yolov4-pytorch A PyTorch implementation of the YOLOv4 object detection model with full training, prediction, and evaluation scripts. It supports training … | 23 | 2160 | maintenance |
| ankush-me/SynthText SynthText is a Python tool for generating synthetic scene-text images with ground-truth bounding boxes, as described in the CVPR 2016 paper… | 32 | 2146 | maintenance |
| SeanNaren/deepspeech.pytorch A PyTorch implementation of the DeepSpeech2 speech recognition model, built on PyTorch Lightning, supporting training, testing, and inferen… | 23 | 2136 | maintenance |
| glouw/tinn Tinn is a tiny, dependency-free neural network library written in under 200 lines of C99, implementing a feedforward network with one hidde… | 32 | 2133 | maintenance |
| chuanqi305/MobileNet-SSD A Caffe implementation of the MobileNet-SSD object detection network with pretrained weights on the VOC0712 dataset achieving mAP of 0.727.… | 46 | 2127 | maintenance |
| rlworkgroup/garage garage is a Python toolkit for developing and evaluating reinforcement learning algorithms, with a library of state-of-the-art RL algorithm… | 23 | 2126 | maintenance |
| magenta/magenta-js Magenta.js is a collection of TypeScript libraries for running inference with pre-trained Magenta machine learning models directly in the b… | 72 | 2125 | maintenance |
| AIZOOTech/FaceMaskDetection An open-source face mask detection project providing a lightweight SSD-based model (1.01M parameters) with inference code for PyTorch, Tens… | 32 | 2125 | maintenance |
| atriumlts/subpixel A TensorFlow reimplementation of the efficient sub-pixel convolutional neural network (ESPCN) for single-image super-resolution, based on S… | 32 | 2123 | maintenance |
| songyouwei/ABSA-PyTorch PyTorch implementations of aspect-based sentiment analysis (ABSA) models, including attention-based and BERT-based approaches for aspect-te… | 10 | 2114 | maintenance |
| bubbliiiing/yolo3-pytorch A PyTorch implementation of the YOLOv3 object detection model with full training, prediction, and evaluation scripts. It supports training … | 23 | 2112 | maintenance |
| facebookresearch/SentEval SentEval is a Python library from Facebook Research for evaluating the quality of sentence embeddings via a suite of downstream transfer ta… | 10 | 2110 | maintenance |
| sfujim/TD3 The author's official PyTorch implementation of Twin Delayed Deep Deterministic Policy Gradients (TD3), a reinforcement learning algorithm … | 32 | 2106 | maintenance |
| asappresearch/sru SRU is a PyTorch library implementing the Simple Recurrent Unit, a highly parallelizable RNN that runs 10-16x faster than cuDNN LSTM with c… | 23 | 2106 | maintenance |
| bgshih/crnn An implementation of the Convolutional Recurrent Neural Network (CRNN), combining CNN, RNN, and CTC loss for image-based sequence recogniti… | 32 | 2105 | maintenance |
| dog-qiuqiu/Yolo-Fastest Yolo-Fastest is an ultra-lightweight YOLO-based object detection algorithm and model zoo, with only ~250 MFLOPs and a 666KB ncnn model. It … | 23 | 2103 | maintenance |
| jalammar/ecco Ecco is a Python library for exploring and explaining Transformer-based NLP language models through interactive visualizations inside Jupyt… | 23 | 2102 | maintenance |
| archinetai/audio-diffusion-pytorch A PyTorch library for audio generation using diffusion models, supporting unconditional and text-conditional generation, diffusion autoenco… | 23 | 2096 | maintenance |
| facebookresearch/swav PyTorch implementation of SwAV, a self-supervised method for pre-training convolutional networks without labels by swapping cluster assignm… | 10 | 2096 | maintenance |
| facebookresearch/ELF ELF is an end-to-end, lightweight and flexible C++/Python platform for game research, focused on real-time strategy games. It hosts multipl… | 10 | 2090 | maintenance |
| ai-forever/ru-gpts A repository of Russian GPT-3 language models (ruGPT3XL/Large/Medium/Small and ruGPT2Large) with usage and fine-tuning examples. It provide… | 32 | 2087 | maintenance |
| GantMan/nsfw_model A Keras/TensorFlow deep learning model that classifies images into NSFW categories (drawings, hentai, neutral, porn, sexy) with ~93% accura… | 23 | 2085 | maintenance |
| THUDM/P-tuning-v2 P-tuning v2 is a Python implementation of deep prompt tuning, applying trainable continuous prompts at every transformer layer so prompt tu… | 32 | 2078 | maintenance |
| UniversalDataTool/universal-data-tool Universal Data Tool is a web and desktop application for collaboratively annotating images, text, audio, and documents, supporting tasks li… | 28 | 2072 | maintenance |
| koth/kcws A deep learning Chinese word segmentation toolkit implemented in C++ with TensorFlow, supporting BiLSTM+CRF and IDCNN+CRF models, plus part… | 23 | 2069 | maintenance |
| RootKit-Org/AI-Aimbot A Python application that uses a YOLOv5-based computer vision model to detect humanoid targets on screen and assist aiming in FPS games lik… | 10 | 2067 | maintenance |
| ChenglongChen/tensorflow-DeepFM A TensorFlow implementation of the DeepFM model combining factorization machines and deep neural networks for click-through-rate (CTR) pred… | 32 | 2066 | maintenance |
| tianqiraf/DouZero_For_HappyDouDiZhu A Python desktop application that applies the DouZero reinforcement-learning Dou Dizhu (Chinese card game) AI to the popular Happy DouDiZhu… | 23 | 2065 | maintenance |
| flowdriveai/flowpilot Flowpilot is an open source driver assistance system built on top of openpilot that provides adaptive cruise control, automated lane center… | 23 | 2065 | maintenance |
| strawlab/python-pcl Python bindings for the Point Cloud Library (PCL), implemented with Cython. It exposes PCL's 3D point cloud processing algorithms (filterin… | 10 | 2056 | maintenance |
| danmacnish/cartoonify A Python application that converts photographs into hand-drawn-style cartoons using a neural network for object recognition combined with t… | 32 | 2052 | maintenance |
| marcotcr/checklist CheckList is a Python library for behavioral testing of NLP models, implementing the ACL 2020 paper 'Beyond Accuracy: Behavioral Testing of… | 32 | 2050 | maintenance |
| bytewax/bytewax Bytewax is a Python-first framework with a Rust-based distributed engine for stateful event and stream processing, inspired by Apache Flink… | 62 | 2047 | maintenance |
| amazon-science/auto-cot Official implementation of Auto-CoT (ICLR 2023), a method that automatically constructs chain-of-thought demonstrations for large language … | 32 | 2046 | maintenance |
| alibaba/AliceMind AliceMind is Alibaba's collection of pre-trained encoder-decoder language models and related NLP techniques, including StructBERT, PALM, VE… | 23 | 2041 | maintenance |
| salu133445/musegan MuseGAN is a TensorFlow-based generative adversarial network for generating polyphonic, multitrack music (bass, drums, guitar, piano, strin… | 32 | 2036 | maintenance |
| pkhungurn/talking-head-anime-demo Demo applications for the 'Talking Head Anime from a Single Image' research project, built with PyTorch. It includes a manual poser for ani… | 32 | 2025 | maintenance |
| WongKinYiu/ScaledYOLOv4 PyTorch implementation of Scaled-YOLOv4, which scales the Cross Stage Partial Network (CSP) across model sizes from tiny to P7 for object d… | 23 | 2024 | maintenance |
| xlang-ai/instructor-embedding INSTRUCTOR is an instruction-finetuned text embedding model and Python library that generates task-tailored embeddings by simply providing … | 32 | 2023 | maintenance |
| lucasb-eyer/pydensecrf PyDenseCRF is a Cython-based Python wrapper around Philipp Krähenbühl's fully-connected (dense) conditional random field implementation wit… | 32 | 2022 | maintenance |
| bytedance/piano_transcription A PyTorch implementation of ByteDance's high-resolution piano transcription system that converts piano audio recordings into MIDI files. It… | 10 | 2021 | maintenance |
| neulab/prompt2model Prompt2Model is a Python library that takes a natural language task description (like an LLM prompt) and automatically generates a small, s… | 21 | 2018 | maintenance |
| baidu/Senta Senta is Baidu's open-source sentiment analysis system built on the SKEP (Sentiment Knowledge Enhanced Pre-training) model. It provides pre… | 32 | 2015 | maintenance |
| open-mmlab/mmgeneration MMGeneration is a PyTorch-based toolkit from OpenMMLab for training and applying generative models, with a focus on GANs and support for di… | 23 | 2012 | maintenance |
| brannondorsey/PassGAN A Python command-line implementation of the PassGAN paper that uses a Wasserstein GAN to generate password guesses, with a pretrained model… | 23 | 2012 | maintenance |
| facebookresearch/Detic Detic is the official code release for the ECCV 2022 paper 'Detecting Twenty-thousand Classes using Image-level Supervision'. It is an open… | 32 | 2008 | maintenance |
| divelab/DIG DIG (Dive into Graphs) is a Python library built on PyTorch Geometric that provides a unified testbed for advanced graph deep learning rese… | 23 | 2007 | maintenance |
| robbiebarrat/art-DCGAN A modified Torch implementation of DCGAN focused on generating artworks, with support for 128x128 images and resuming training from checkpo… | 32 | 2003 | maintenance |
| WongKinYiu/yolor PyTorch implementation of the YOLOR paper 'You Only Learn One Representation: Unified Network for Multiple Tasks', a real-time object detec… | 23 | 2003 | maintenance |
| haitongli/knowledge-distillation-pytorch A PyTorch framework for running knowledge distillation experiments, supporting both shallow (teacher-to-small-CNN) and deep distillation on… | 32 | 2000 | maintenance |
| asg017/sqlite-vss sqlite-vss is a SQLite extension written in C++ that adds vector similarity search to SQLite using Faiss, via vss0 virtual tables and a vss… | 22 | 1998 | maintenance |
| HobbitLong/PyContrast PyContrast is a PyTorch library providing reference implementations of contrastive learning methods such as InstDis, CMC, and MoCo, along w… | 32 | 1993 | maintenance |
| NVlabs/edm Official PyTorch implementation of the NeurIPS 2022 paper 'Elucidating the Design Space of Diffusion-Based Generative Models' (EDM) by NVID… | 32 | 1991 | maintenance |
| salesforce/awd-lstm-lm A PyTorch toolkit for training LSTM and QRNN language models, implementing Salesforce Research's AWD-LSTM papers. It supports word-level an… | 10 | 1989 | maintenance |
| databricks/spark-deep-learning Deep Learning Pipelines for Apache Spark, now reduced to the HorovodRunner component for distributed deep learning training via Horovod on … | 23 | 1987 | maintenance |
| ZiyaoGeng/RecLearn RecLearn is a Python/TensorFlow2.x library implementing a wide collection of recommendation algorithms, covering both matching (Top-K recom… | 23 | 1986 | maintenance |
| apple/ml-cvnets CVNets is Apple's open-source PyTorch library for training computer vision networks, covering classification, detection, segmentation, vide… | 32 | 1983 | maintenance |
| Tencent/ObjectDetection-OneStageDet OneStageDet (OSD) is a PyTorch-based single-stage object detection framework from Tencent implementing YOLOv2 and YOLOv3 with plans to unif… | 45 | 1980 | maintenance |
| WuJie1010/Facial-Expression-Recognition.Pytorch A PyTorch implementation of CNN-based facial expression recognition achieving state-of-the-art accuracy on FER2013 (73.112%) and CK+ (94.64… | 32 | 1976 | maintenance |
| ICT-BDA/EasyML EasyML is a general-purpose dataflow-based machine learning platform where tasks are defined as directed acyclic graphs of operations. It i… | 23 | 1976 | maintenance |
| r9y9/deepvoice3_pytorch A PyTorch implementation of Deep Voice 3 and related convolutional neural network-based text-to-speech synthesis models. It includes traini… | 23 | 1975 | maintenance |
| SummitKwan/transparent_latent_gan TL-GAN is a Python/TensorFlow project that makes a GAN's latent space transparent by discovering feature axes, enabling controlled image sy… | 32 | 1973 | maintenance |
| google/nerfies Nerfies is the official JAX implementation of Deformable Neural Radiance Fields, which reconstructs non-rigidly deforming scenes from casua… | 10 | 1973 | maintenance |
| Hitachi-Automotive-And-Industry-Lab/semantic-segmentation-editor A web-based labeling tool for creating AI training datasets from bitmap images (.jpg/.png) and point clouds (.pcd). Built as a Meteor app w… | 23 | 1970 | maintenance |
| JavierAntoran/Bayesian-Neural-Networks A collection of PyTorch implementations of Bayesian neural network approximate inference methods, including Bayes by Backprop, MC Dropout, … | 32 | 1969 | maintenance |
| deepseek-ai/DeepSeek-MoE DeepSeekMoE 16B is a Mixture-of-Experts language model with fine-grained expert segmentation and shared expert isolation, released with bas… | 26 | 1969 | maintenance |
| nobody132/masr MASR is an end-to-end Mandarin Chinese automatic speech recognition project built on a gated convolutional neural network (similar to Wav2L… | 23 | 1968 | maintenance |
| MojoJolo/textteaser TextTeaser is an automatic text summarization algorithm written in Scala that combines natural language processing and machine learning to … | 32 | 1967 | maintenance |
| Music-and-Culture-Technology-Lab/omnizart Omnizart is a Python library and CLI for automatic music transcription, transcribing pitched instruments, vocal melody, chords, drum events… | 88 | 1964 | maintenance |
| yehengchen/Object-Detection-and-Tracking A collection of Python implementations combining YOLO-based object detection with SORT and DeepSORT multi-object tracking. It includes exam… | 32 | 1961 | maintenance |
| ml-jku/hopfield-layers A PyTorch library implementing modern Hopfield networks with continuous states, whose update rule is equivalent to transformer attention. I… | 32 | 1959 | maintenance |
| devzwy/open_nsfw_android An Android library for offline NSFW (pornographic) image detection, ported from Yahoo's open_nsfw project using TensorFlow Lite. It classif… | 10 | 1957 | maintenance |
| mit-han-lab/once-for-all Once-for-All (OFA) is a PyTorch library implementing the ICLR 2020 Once-for-All network, which trains a single supernet that can be special… | 23 | 1956 | maintenance |
| guillaumegenthial/sequence_tagging A TensorFlow implementation of a Bi-LSTM + CRF model with character embeddings for named entity recognition and sequence tagging. It trains… | 32 | 1952 | maintenance |
| hukaixuan19970627/yolov5_obb A PyTorch implementation of YOLOv5 extended for oriented (rotated) object detection using Circular Smooth Label (CSL) angle encoding. It pr… | 32 | 1947 | maintenance |
| peng-zhihui/DeepVision DeepVision is an Android-based computer vision algorithm inference and validation framework by peng-zhihui. It decouples data flow control,… | 32 | 1942 | maintenance |
| julius-speech/julius Julius is an open-source large vocabulary continuous speech recognition (LVCSR) decoder written in C, based on word N-gram language models … | 35 | 1933 | maintenance |
| google/sling SLING is a natural language frame semantics parser that annotates text with frame semantic graph representations using bi-directional LSTMs… | 10 | 1930 | maintenance |
| uTensor/uTensor uTensor is an extremely lightweight machine learning inference framework built on TensorFlow and optimized for Arm Cortex-M microcontroller… | 32 | 1928 | maintenance |
| danieldjohnson/biaxial-rnn-music-composition A Python implementation of a biaxial recurrent neural network (LSTM-based) trained to generate classical music from MIDI data. It includes … | 32 | 1926 | maintenance |
| S-LoRA/S-LoRA S-LoRA is a serving system for large language models that scales to thousands of concurrent LoRA adapters on one or more GPUs. It uses Unif… | 10 | 1923 | maintenance |
| Tencent/NeuralNLP-NeuralClassifier NeuralClassifier is a PyTorch-based open-source toolkit for neural text classification, with a focus on hierarchical multi-label classifica… | 56 | 1922 | maintenance |
| black0017/MedicalZooPytorch A PyTorch-based open-source library implementing state-of-the-art 3D (and some 2D) deep neural networks for multi-modal medical image segme… | 32 | 1917 | maintenance |
| appvision-ai/fast-bert Fast-Bert is a Python deep learning library for training and deploying BERT, RoBERTa, and XLNet based models for NLP tasks, starting with m… | 23 | 1917 | maintenance |
| Tiiiger/bert_score BERTScore is a PyTorch implementation of the BERTScore automatic evaluation metric for text generation, which scores generated text against… | 23 | 1916 | maintenance |
| Yvictor/TradingGym TradingGym is an OpenAI Gym-style Python toolkit providing trading environments for training reinforcement learning agents and backtesting … | 32 | 1910 | maintenance |
| DmitryUlyanov/Multicore-TSNE A multicore modification of Barnes-Hut t-SNE written in C++ with Python (CFFI) and Torch wrappers for fast dimensionality reduction of high… | 32 | 1909 | maintenance |
| WongKinYiu/PyTorch_YOLOv4 A PyTorch implementation of the YOLOv4 real-time object detection model, based on ultralytics/yolov3. It includes research extensions such … | 23 | 1908 | maintenance |
| Palashio/libra Libra is a Python autoML library that lets users build, train, and evaluate machine learning models with one-line natural-language-style qu… | 40 | 1907 | maintenance |
| snap-stanford/GraphGym GraphGym is a platform for designing and evaluating Graph Neural Networks (GNNs), built around a highly modularized pipeline covering data … | 23 | 1904 | maintenance |
| ronghuaiyang/arcface-pytorch A PyTorch implementation of ArcFace, a deep metric learning approach for face recognition that adds angular margin penalties to face embedd… | 32 | 1901 | maintenance |
| whai362/PVT Official PyTorch implementation of the Pyramid Vision Transformer (PVTv1 and PVTv2) model series, providing pretrained backbone networks fo… | 23 | 1900 | maintenance |
| jiesutd/NCRFpp NCRF++ is a PyTorch-based toolkit for neural sequence labeling tasks such as NER, POS tagging, and word segmentation. Model architectures c… | 32 | 1899 | maintenance |
| facebookresearch/denoiser A PyTorch library implementing a causal, real-time speech enhancement model that operates on raw waveforms to remove background noise and r… | 10 | 1899 | maintenance |
| zjhuang22/maskscoring_rcnn Official PyTorch implementation of Mask Scoring R-CNN (CVPR 2019), built on maskrcnn-benchmark. It adds a network block that learns the qua… | 32 | 1895 | maintenance |
| imgcook/imgcook imgcook is a design-to-code tool from Alibaba that intelligently generates front-end code (views, data bindings, components, and some busin… | 23 | 1892 | maintenance |
| uber/petastorm Petastorm is a Python data access library from Uber that enables single-machine or distributed training and evaluation of deep learning mod… | 58 | 1891 | maintenance |
| microsoft/BlingFire Bling Fire is a fast finite state machine and regular expression manipulation library from Microsoft, primarily used for high-performance n… | 23 | 1890 | maintenance |