function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| cure-lab/LTSF-Linear Official PyTorch implementation of LTSF-Linear (Linear, DLinear, NLinear) from the AAAI-23 paper 'Are Transformers Effective for Time Serie… | 32 | 2509 | maintenance |
| filipedeschamps/video-maker An open-source Node.js project that automates video production by chaining robots that fetch content, analyze text with IBM Watson NLU, and… | 32 | 2499 | maintenance |
| meijieru/crnn.pytorch A PyTorch implementation of the Convolutional Recurrent Neural Network (CRNN) for scene text recognition, based on the 2016 paper by Shi et… | 32 | 2494 | 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 |
| huggingface/evaluate Hugging Face's library for easily evaluating machine learning models and datasets with dozens of standardized metrics, comparisons, and mea… | 74 | 2478 | maintenance |
| xingyizhou/CenterTrack CenterTrack is a deep learning model and research codebase that performs simultaneous multi-object detection and tracking using center poin… | 32 | 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 | 2462 | 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 |
| 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 | 2441 | maintenance |
| RameenAbdal/StyleFlow StyleFlow is a research implementation of attribute-conditioned exploration of StyleGAN-generated images using conditional continuous norma… | 32 | 2441 | maintenance |
| HobbitLong/RepDistiller RepDistiller is a PyTorch research codebase implementing Contrastive Representation Distillation (CRD) from ICLR 2020. It also benchmarks 1… | 32 | 2439 | 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 | 2437 | maintenance |
| zai-org/CogVLM2 CogVLM2 is an open-source multi-modal vision-language model family built on Meta-Llama-3-8B-Instruct, offering image and video understandin… | 28 | 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 |
| AkariAsai/self-rag Original implementation of Self-RAG, a framework that trains a language model to adaptively retrieve passages on demand and critique its ow… | 27 | 2421 | 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 |
| deepcam-cn/yolov5-face YOLOv5-Face is a real-time, high-accuracy face detector built on the YOLOv5 object detection framework in PyTorch, with TensorRT deployment… | 32 | 2406 | maintenance |
| OpenBMB/CPM-Bee CPM-Bee is a fully open-source, commercially usable 10-billion-parameter bilingual (Chinese-English) foundation large language model traine… | 71 | 2405 | 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 | 2405 | 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 | 2395 | 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 | 2395 | maintenance |
| iPERDance/iPERCore Impersonator++ (iPERCore) is a PyTorch implementation of Liquid Warping GAN with Attention, a unified framework for human image synthesis. … | 32 | 2393 | maintenance |
| asyml/texar Texar is a modularized Python toolkit for machine learning, especially natural language processing and text generation, built on TensorFlow… | 65 | 2388 | 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 | 2388 | 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 | 2380 | 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 | 2376 | 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 |
| 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 | 2374 | 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 | 2366 | 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 | 2363 | 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 | 2363 | 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 | 2356 | 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 |
| 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 | 2344 | maintenance |
| moskomule/senet.pytorch A PyTorch implementation of Squeeze-and-Excitation Networks (SENet), the ILSVRC 2017 classification winner, including SE-ResNet and SE-Ince… | 23 | 2343 | 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 | 2341 | 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 |
| OAID/TengineKit TengineKit is a mobile AI SDK by OPEN AI LAB providing real-time face detection, face 2D/3D landmarks, face attributes, iris, hand, and bod… | 23 | 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 |
| kijai/ComfyUI-SUPIR A ComfyUI custom node wrapper for the SUPIR image upscaler, an SDXL img2img pipeline with a custom ControlNet. It offers more efficient mod… | 61 | 2308 | maintenance |
| hzwer/ICCV2019-LearningToPaint A PyTorch research implementation of the ICCV 2019 paper 'Learning to Paint With Model-based Deep Reinforcement Learning'. It trains agents… | 41 | 2308 | 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 |
| 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 | 2295 | 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 | 2295 | maintenance |
| JonathonLuiten/Dynamic3DGaussians Official PyTorch implementation of 'Dynamic 3D Gaussians: Tracking by Persistent Dynamic View Synthesis' (3DV 2024), which models dynamic 3… | 28 | 2293 | maintenance |
| titu1994/Neural-Style-Transfer A Keras 2.0+ implementation of Neural Style Transfer from the paper 'A Neural Algorithm of Artistic Style', with improvements like INetwork… | 23 | 2287 | maintenance |
| approximatelabs/sketch Sketch is an AI code-writing assistant for pandas users that understands the context of your dataframe content to generate relevant suggest… | 32 | 2284 | 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 | 2276 | maintenance |
| andreas128/RePaint Official PyTorch implementation of RePaint, a CVPR 2022 paper on image inpainting using denoising diffusion probabilistic models. It provid… | 32 | 2269 | maintenance |
| ndrplz/ConvLSTM_pytorch A PyTorch implementation of the Convolutional LSTM (ConvLSTM) neural network module, provided as a reusable nn.Module supporting arbitrary … | 32 | 2268 | 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 |
| mrharicot/monodepth A TensorFlow implementation of unsupervised monocular depth estimation from single images using convolutional neural networks, based on the… | 32 | 2264 | 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 | 2261 | 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 | 2257 | 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 | 2256 | 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 | 2253 | maintenance |
| MineDojo/MineDojo MineDojo is an AI research framework built on Minecraft for training open-ended, generally capable embodied agents. It provides a simulatio… | 23 | 2253 | 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 |
| bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets A PyTorch implementation of several U-Net variants for image segmentation, including UNet, R2U-Net, Attention U-Net, Attention R2U-Net, and… | 32 | 2249 | 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 | 2241 | maintenance |
| chartbeat-labs/textacy textacy is a Python NLP library built on spaCy that handles tasks before and after spaCy's core pipeline, such as text cleaning, informatio… | 23 | 2239 | maintenance |
| hustvl/YOLOP YOLOP is a multi-task deep learning network that jointly performs traffic object detection, drivable area segmentation, and lane detection … | 32 | 2238 | maintenance |
| machrisaa/tensorflow-vgg A Python library implementing the VGG16 and VGG19 convolutional neural networks in TensorFlow, loading pretrained weights from npy files fo… | 32 | 2237 | 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 | 2236 | maintenance |
| Jongchan/attention-module Official PyTorch implementation of two attention modules for convolutional neural networks: BAM (Bottleneck Attention Module) and CBAM (Con… | 32 | 2229 | 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 |
| anopara/genetic-drawing A Python toy project that uses a genetic algorithm to progressively draw a target image with brushstrokes, producing animated drawing seque… | 32 | 2225 | maintenance |
| gligen/GLIGEN GLIGEN is a research library for open-set grounded text-to-image generation, extending frozen diffusion models with spatial grounding via b… | 31 | 2223 | 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 | 2222 | 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 |
| 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 | 2219 | 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 | 2218 | 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 | 2218 | 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 | 2210 | maintenance |
| epfLLM/meditron Meditron is a suite of open-source medical large language models (7B and 70B) adapted from Llama-2 via continued pretraining on a curated m… | 27 | 2210 | maintenance |
| dmis-lab/biobert BioBERT is a pre-trained biomedical language representation model (based on BERT) with code for fine-tuning it on biomedical text mining ta… | 32 | 2206 | 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 | 2204 | 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 |