domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| s3prl/s3prl S3PRL is a PyTorch toolkit for self-supervised speech pre-training and representation learning, bundling many upstream models like wav2vec … | 55 | 2561 | maintenance |
| jcjohnson/torch-rnn torch-rnn provides high-performance, reusable RNN and LSTM modules for the Torch7 deep learning framework, used for character-level languag… | 32 | 2559 | maintenance |
| OFA-Sys/OFA OFA is a unified sequence-to-sequence pretrained model supporting English and Chinese that unifies cross-modality, vision, and language tas… | 32 | 2557 | maintenance |
| onnx/tensorflow-onnx tf2onnx is a tool that converts TensorFlow, Keras, TensorFlow.js, and TFLite models into the ONNX format via a command-line interface or Py… | 77 | 2553 | maintenance |
| LiyuanLucasLiu/RAdam RAdam is a Python implementation of Rectified Adam, a variant of the Adam optimizer that analytically reduces the large variance of adaptiv… | 32 | 2549 | maintenance |
| automl/Auto-PyTorch Auto-PyTorch is an AutoML framework that jointly performs neural architecture search and hyperparameter optimization for PyTorch models, us… | 23 | 2541 | maintenance |
| dmlc/gluon-nlp GluonNLP is a Python toolkit built on Apache MXNet that provides text data loading, processing tools, pretrained model zoos, and training u… | 10 | 2540 | maintenance |
| google-deepmind/dnc A TensorFlow implementation of the Differentiable Neural Computer (DNC), a recurrent neural network architecture with an external memory pu… | 32 | 2531 | maintenance |
| dblalock/bolt Bolt is a C++ library (with a Python wrapper) for lossy compression of dense real-valued vectors that supports mathematical operations dire… | 32 | 2514 | maintenance |
| young-geng/EasyLM EasyLM is a JAX/Flax-based framework for pre-training, finetuning, evaluating, and serving large language models like LLaMA. It scales trai… | 32 | 2514 | maintenance |
| dmlc/decord Decord is a C++ library with Python bindings that provides efficient video and audio decoding for deep learning pipelines. It wraps hardwar… | 23 | 2514 | maintenance |
| zzh8829/yolov3-tf2 A clean implementation of YOLOv3 and YOLOv3-tiny object detection in TensorFlow 2.0, with pre-trained Darknet weight conversion, inference,… | 32 | 2513 | maintenance |
| unslothai/hyperlearn HyperLearn is a Python library providing optimized reimplementations of classical machine learning and statistical algorithms (SVD, eigende… | 32 | 2505 | maintenance |
| galeone/tfgo tfgo is a Go library that wraps TensorFlow's Go bindings with a friendlier, method-chaining API for building and executing computation grap… | 32 | 2491 | maintenance |
| dragen1860/MAML-Pytorch A PyTorch implementation of Model-Agnostic Meta-Learning (MAML) for few-shot supervised learning, supporting MiniImagenet and Omniglot benc… | 32 | 2483 | maintenance |
| xingyizhou/CenterTrack CenterTrack is a deep learning model and research codebase that performs simultaneous multi-object detection and tracking using center poin… | 32 | 2478 | maintenance |
| huggingface/evaluate Hugging Face's library for easily evaluating machine learning models and datasets with dozens of standardized metrics, comparisons, and mea… | 74 | 2477 | maintenance |
| microsoft/Graphormer Graphormer is a deep learning library from Microsoft providing a graph transformer backbone for molecular modeling tasks. It ships pre-trai… | 62 | 2474 | maintenance |
| CASIA-IVA-Lab/DANet DANet is the official PyTorch implementation of 'Dual Attention Network for Scene Segmentation' (CVPR 2019), which uses position and channe… | 32 | 2463 | maintenance |
| XifengGuo/CapsNet-Keras A Keras implementation of CapsNet, the capsule network architecture from the NIPS 2017 paper 'Dynamic Routing Between Capsules', achieving … | 23 | 2456 | maintenance |
| Zhongdao/Towards-Realtime-MOT A PyTorch codebase for the Joint Detection and Embedding (JDE) model, a fast multiple-object tracker that learns object detection and appea… | 32 | 2444 | maintenance |
| RameenAbdal/StyleFlow StyleFlow is a research implementation of attribute-conditioned exploration of StyleGAN-generated images using conditional continuous norma… | 32 | 2442 | maintenance |
| leggedrobotics/darknet_ros A ROS package wrapping the YOLO (Darknet) real-time object detector for use in robotic systems. It subscribes to camera image topics and pu… | 23 | 2439 | maintenance |
| HobbitLong/RepDistiller RepDistiller is a PyTorch research codebase implementing Contrastive Representation Distillation (CRD) from ICLR 2020. It also benchmarks 1… | 32 | 2437 | maintenance |
| notAI-tech/NudeNet NudeNet is a lightweight Python library for nudity detection in images, built on YOLOv8 models running via ONNX Runtime. It detects exposed… | 62 | 2433 | maintenance |
| reiinakano/scikit-plot Scikit-plot is a Python library that adds one-line plotting functions for common machine learning visualizations to scikit-learn objects, b… | 23 | 2433 | maintenance |
| jameslyons/python_speech_features A Python library for extracting common speech features used in automatic speech recognition, including MFCCs, filterbank energies, log filt… | 23 | 2423 | maintenance |
| CjangCjengh/MoeGoe MoeGoe is an executable command-line tool for running inference with VITS text-to-speech models, supporting TTS, voice conversion, HuBERT-V… | 23 | 2423 | maintenance |
| microsoft/DialoGPT DialoGPT is a large-scale pretrained dialogue response generation model from Microsoft, based on GPT-2 and trained on 147M multi-turn Reddi… | 10 | 2420 | maintenance |
| CyberZHG/keras-bert A Keras implementation of BERT that can load official Google pre-trained checkpoints for feature extraction and prediction. It includes a t… | 10 | 2419 | maintenance |
| igul222/improved_wgan_training Reference implementation of the 'Improved Training of Wasserstein GANs' (WGAN-GP) paper, providing TensorFlow code to reproduce its experim… | 32 | 2411 | maintenance |
| mravanelli/pytorch-kaldi PyTorch-Kaldi is a toolkit for developing state-of-the-art DNN/HMM hybrid speech recognition systems, combining PyTorch-managed neural netw… | 32 | 2404 | maintenance |
| danielegrattarola/spektral Spektral is a Python library for graph deep learning built on the Keras API and TensorFlow 2. It provides popular graph neural network laye… | 23 | 2396 | maintenance |
| allenai/RL4LMs RL4LMs is a modular Python library from AllenAI for fine-tuning language models with reinforcement learning to align with human preferences… | 32 | 2394 | maintenance |
| greerviau/SnakeAI A Processing application that trains neural networks to play the game Snake using a genetic algorithm with crossover and mutation across ge… | 32 | 2390 | maintenance |
| asyml/texar Texar is a modularized Python toolkit for machine learning, especially natural language processing and text generation, built on TensorFlow… | 65 | 2389 | maintenance |
| BrikerMan/Kashgari Kashgari is a Keras/TensorFlow 2-based NLP transfer learning framework for building text labeling (NER, PoS) and text classification models… | 23 | 2381 | maintenance |
| r9y9/wavenet_vocoder A PyTorch implementation of the WaveNet vocoder that generates high-quality raw speech waveforms conditioned on acoustic features like mel-… | 23 | 2376 | maintenance |
| nbei/Deep-Flow-Guided-Video-Inpainting A PyTorch implementation of the CVPR 2019 paper 'Deep Flow-Guided Video Inpainting', which fills missing regions in videos by completing op… | 32 | 2375 | maintenance |
| flann-lib/flann FLANN is a C++ library for fast approximate nearest neighbor searches in high-dimensional spaces. It bundles multiple search algorithms wit… | 32 | 2373 | maintenance |
| princeton-vl/CornerNet Official research code for CornerNet, an object detection model that detects objects as paired keypoints, reproducing results from the ECCV… | 32 | 2369 | maintenance |
| google-research/electra ELECTRA is a research library from Google for self-supervised pre-training of transformer text encoders using a discriminator-based objecti… | 10 | 2367 | maintenance |
| modAL-python/modAL modAL is a modular active learning framework for Python 3, built on top of scikit-learn. It lets users rapidly assemble active learning wor… | 23 | 2362 | maintenance |
| aboSamoor/polyglot Polyglot is a Python natural language processing toolkit designed for massive multilingual applications, offering language detection, token… | 32 | 2361 | maintenance |
| IST-DASLab/gptq Reference implementation of GPTQ, a one-shot post-training weight quantization method for large generative transformer models based on appr… | 32 | 2360 | maintenance |
| michuanhaohao/reid-strong-baseline A PyTorch implementation of the 'Bag of Tricks and A Strong Baseline for Deep Person Re-identification' paper (CVPRW 2019), providing end-t… | 32 | 2355 | maintenance |
| fxia22/pointnet.pytorch A PyTorch implementation of PointNet, the deep learning model for classification and segmentation of 3D point sets. It includes training sc… | 32 | 2354 | maintenance |
| smallcorgi/Faster-RCNN_TF A TensorFlow implementation of Faster R-CNN, a convolutional neural network for object detection with a region proposal network. It include… | 32 | 2342 | maintenance |
| n00mkrad/cupscale Cupscale is a Windows GUI application for AI-based image upscaling built around ESRGAN and Real-ESRGAN models. It supports CUDA, Vulkan/NCN… | 23 | 2341 | maintenance |
| NVIDIA/waveglow WaveGlow is a PyTorch implementation of a flow-based generative network that synthesizes high-quality speech audio from mel-spectrograms, c… | 32 | 2339 | maintenance |
| salesforce/decaNLP decaNLP is a PyTorch research framework and benchmark for the Natural Language Decathlon, a multitask NLP challenge spanning ten tasks from… | 10 | 2337 | maintenance |
| Determined22/zh-NER-TF A simple character-based BiLSTM-CRF sequence labeling model for Chinese Named Entity Recognition built with TensorFlow 1.x. It recognizes P… | 32 | 2333 | maintenance |
| Rayhane-mamah/Tacotron-2 A TensorFlow implementation of DeepMind's Tacotron-2 neural text-to-speech architecture, including both the Tacotron spectrogram predictor … | 32 | 2322 | maintenance |
| Hzzone/pytorch-openpose A PyTorch reimplementation of OpenPose for body and hand pose estimation, with models converted directly from the original OpenPose caffemo… | 32 | 2321 | maintenance |
| nanohop/sketch-to-react-native A CLI tool that converts Sketch design files (exported as SVG) into React Native components using a deep neural network and headless Chrome… | 32 | 2313 | maintenance |
| facebookresearch/pyrobot PyRobot is a lightweight, high-level Python library providing hardware-independent APIs for robotic manipulation and navigation, built on t… | 10 | 2307 | maintenance |
| microsoft/ELL ELL (Embedded Learning Library) is a C++ library and toolset from Microsoft Research for designing and deploying machine-learned models ont… | 10 | 2302 | maintenance |
| fudan-zvg/Semantic-Segment-Anything Semantic Segment Anything (SSA) is a Python framework that adds semantic category prediction to the Segment Anything Model (SAM) by combini… | 30 | 2301 | maintenance |
| facebookresearch/frankmocap FrankMocap is a single-view 3D motion capture system from Facebook AI Research that estimates 3D pose for body, hands, and whole body (body… | 10 | 2294 | maintenance |
| AIStream-Peelout/flow-forecast Flow Forecast is a PyTorch-based deep learning framework for time series forecasting, classification, and anomaly detection. It offers stat… | 66 | 2292 | maintenance |
| jsbroks/coco-annotator COCO Annotator is a web-based image annotation tool for labeling images with segments, bounding boxes, keypoints, and object tracking to cr… | 25 | 2279 | maintenance |
| astooke/rlpyt A PyTorch library providing modular, high-throughput implementations of deep reinforcement learning algorithms, covering policy gradient, d… | 32 | 2278 | maintenance |
| salesforce/TransmogrifAI TransmogrifAI is an AutoML library written in Scala that runs on Apache Spark for building modular, reusable, strongly typed machine learni… | 61 | 2277 | maintenance |
| zju3dv/NeuralRecon NeuralRecon is a deep learning framework for real-time 3D scene reconstruction from monocular video with known camera poses. It reconstruct… | 32 | 2274 | maintenance |
| rsennrich/subword-nmt A Python library and CLI toolset for unsupervised word segmentation into subword units, best known for byte pair encoding (BPE) used in neu… | 23 | 2274 | maintenance |
| qianqianwang68/omnimotion OmniMotion is a PyTorch implementation of the ICCV 2023 paper 'Tracking Everything Everywhere All at Once', which tracks every point in a v… | 29 | 2268 | maintenance |
| ndrplz/ConvLSTM_pytorch A PyTorch implementation of the Convolutional LSTM (ConvLSTM) neural network module, provided as a reusable nn.Module supporting arbitrary … | 32 | 2267 | maintenance |
| mrharicot/monodepth A TensorFlow implementation of unsupervised monocular depth estimation from single images using convolutional neural networks, based on the… | 32 | 2265 | maintenance |
| crownpku/Information-Extraction-Chinese A collection of deep learning models for Chinese information extraction, including named entity recognition with IDCNN or biLSTM+CRF and re… | 32 | 2264 | maintenance |
| MhLiao/DB A PyTorch implementation of DBNet and DBNet++, real-time arbitrary-shape scene text detection models based on differentiable binarization. … | 32 | 2260 | maintenance |
| namisan/mt-dnn A PyTorch library implementing Multi-Task Deep Neural Networks (MT-DNN) for natural language understanding, based on Microsoft research pap… | 32 | 2257 | maintenance |
| ShoufaChen/DiffusionDet PyTorch implementation of DiffusionDet, the first diffusion-model-based object detection framework (ICCV 2023 Best Paper Finalist). It prov… | 22 | 2257 | maintenance |
| Tencent/MedicalNet MedicalNet provides a series of 3D-ResNet pre-trained models trained on 23 diverse medical imaging datasets, with PyTorch transfer-learning… | 57 | 2255 | maintenance |
| hunglc007/tensorflow-yolov4-tflite A TensorFlow 2.x implementation of YOLOv4, YOLOv4-tiny, YOLOv3, and YOLOv3-tiny object detection models, with scripts that convert original… | 32 | 2254 | maintenance |
| donnyyou/torchcv TorchCV is a PyTorch-based framework providing reimplementations of deep learning models for major computer vision tasks. It covers image c… | 32 | 2251 | maintenance |
| h2oai/h2o-2 H2O-2 is a Java-based distributed machine learning and math engine that scales statistics and predictive modeling over Big Data platforms l… | 10 | 2249 | maintenance |
| idealo/image-quality-assessment A Python implementation of Google's NIMA (Neural Image Assessment) models that predict the aesthetic and technical quality of images using … | 10 | 2243 | maintenance |
| Daniil-Osokin/lightweight-human-pose-estimation.pytorch A PyTorch implementation of Lightweight OpenPose for real-time 2D multi-person human pose estimation on CPU. It detects up to 18 body keypo… | 32 | 2241 | maintenance |
| microsoft/DeBERTa Official implementation of DeBERTa and DeBERTa V3, transformer-based language models with disentangled attention and decoding-enhanced BERT… | 23 | 2239 | maintenance |
| lonePatient/BERT-NER-Pytorch A PyTorch library for Chinese Named Entity Recognition using BERT-based models with Softmax, CRF, and Span decoding heads. It also supports… | 32 | 2235 | maintenance |
| x-flux XLabs AI's training scripts for fine-tuning the FLUX.1 diffusion model with LoRA, ControlNet, and IP-Adapter adapters, using DeepSpeed and … | 23 | 2229 | maintenance |
| WillKoehrsen/feature-selector A Python library for feature selection that identifies features to remove from machine learning datasets using five methods: missing values… | 32 | 2228 | maintenance |
| KaiyangZhou/CoOp A research codebase implementing prompt learning methods (CoOp and CoCoOp) for adapting vision-language models like CLIP to downstream clas… | 32 | 2221 | maintenance |
| mit-han-lab/temporal-shift-module PyTorch implementation of the Temporal Shift Module (TSM), an ICCV 2019 technique that adds temporal modeling to 2D CNNs at zero extra comp… | 32 | 2221 | maintenance |
| thu-ml/zhusuan ZhuSuan is a Python probabilistic programming library for Bayesian deep learning built on TensorFlow. It provides primitives for building p… | 32 | 2219 | maintenance |
| ProsusAI/finBERT FinBERT is a pre-trained BERT language model further trained on financial text and fine-tuned for financial sentiment classification. It is… | 32 | 2217 | maintenance |
| oxwhirl/pymarl PyMARL is WhiRL's PyTorch framework for deep multi-agent reinforcement learning, implementing algorithms such as QMIX, COMA, VDN, IQL, and … | 32 | 2216 | maintenance |
| openai/consistencydecoder A Python library from OpenAI implementing a consistency-distilled diffusion VAE decoder that improves image decoding quality for Stable Dif… | 10 | 2212 | maintenance |
| ashawkey/torch-ngp A PyTorch implementation of instant-ngp's NeRF and SDF components with CUDA extensions and an interactive GUI for training and visualizatio… | 32 | 2208 | maintenance |
| allenai/longformer Longformer is a pretrained transformer model family (including the LongformerEncoderDecoder/LED variant) that processes long documents up t… | 23 | 2205 | maintenance |
| google-research/uda Google Research's reference implementation of Unsupervised Data Augmentation (UDA), a semi-supervised learning method that uses advanced da… | 10 | 2205 | maintenance |
| yhenon/pytorch-retinanet A PyTorch implementation of the RetinaNet object detection model with focal loss, designed for readability and easy modification. It includ… | 10 | 2205 | maintenance |
| lucidrains/reformer-pytorch A PyTorch implementation of the Reformer, an efficient Transformer architecture using LSH attention, reversible networks, and chunking to h… | 23 | 2191 | maintenance |
| fatchord/WaveRNN A PyTorch implementation of DeepMind's WaveRNN neural vocoder plus a Tacotron text-to-speech system, trained on LJSpeech. It supports train… | 32 | 2190 | maintenance |
| magicleap/SuperPointPretrainedNetwork A PyTorch pre-trained implementation of the SuperPoint fully convolutional neural network for real-time interest point detection and descri… | 32 | 2185 | maintenance |
| alibaba/EasyNLP EasyNLP is a comprehensive PyTorch-based NLP toolkit from Alibaba that provides training, inference, and deployment for pre-trained languag… | 23 | 2184 | maintenance |
| githubharald/SimpleHTR A Handwritten Text Recognition (HTR) system implemented in TensorFlow that recognizes text from images of single words or text lines, train… | 72 | 2183 | maintenance |
| tensorchord/pgvecto.rs pgvecto.rs is a Postgres extension written in Rust (via pgrx) that adds vector similarity search with high dimension support, quantization,… | 24 | 2183 | maintenance |
| maxpumperla/hyperas Hyperas is a Python library that wraps Hyperopt to provide convenient hyperparameter optimization for Keras models. Users annotate model pa… | 32 | 2173 | maintenance |
| vchoutas/smplify-x SMPLify-X is the official PyTorch implementation of the CVPR 2019 paper 'Expressive Body Capture: 3D Hands, Face, and Body from a Single Im… | 32 | 2163 | maintenance |