domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| TRI-ML/packnet-sfm Official PyTorch implementation of PackNet and related self-supervised monocular depth estimation methods from Toyota Research Institute's … | 23 | 1274 | maintenance |
| marcbelmont/cnn-watermark-removal A TensorFlow implementation of a fully convolutional neural network that removes transparent watermark overlays from images. It trains on s… | 32 | 1270 | maintenance |
| google-research/deduplicate-text-datasets A Rust implementation of ExactSubstr deduplication for language model training datasets, with Python scripts for running deduplication and … | 10 | 1270 | maintenance |
| DeepGraphLearning/graphvite GraphVite is a general, high-performance graph embedding engine written in C++ with Python bindings, supporting node embedding, knowledge g… | 32 | 1269 | maintenance |
| summanlp/textrank A Python 3 implementation of the TextRank algorithm for extractive text summarization and keyword extraction. It is the library published o… | 23 | 1269 | maintenance |
| HendrikStrobelt/LSTMVis LSTMVis is a visual analysis toolbox for inspecting hidden state dynamics of LSTM (and other RNN) networks. It provides a browser-based int… | 23 | 1266 | maintenance |
| tensorlayer/HyperPose HyperPose is a library for building high-performance custom human pose estimation applications. It combines a C++ inference engine with Ten… | 23 | 1264 | maintenance |
| kennymckormick/pyskl PYSKL is a PyTorch-based toolbox for skeleton-based human action recognition, built on MMAction2. It provides official implementations of P… | 53 | 1263 | maintenance |
| TimoBolkart/voca VOCA is a speech-driven 3D facial animation framework that synthesizes realistic character face animations from an arbitrary speech signal … | 32 | 1263 | maintenance |
| linkedin/detext DeText is a deep neural text understanding framework from LinkedIn for NLP ranking, classification, and language generation tasks. It uses … | 23 | 1263 | maintenance |
| salesforce/pytorch-qrnn A PyTorch implementation of the Quasi-Recurrent Neural Network (QRNN) from Salesforce Research, offering LSTM-like accuracy with 2-17x fast… | 10 | 1263 | maintenance |
| openpifpaf/openpifpaf OpenPifPaf is a PyTorch library implementing Composite Fields for semantic keypoint detection and spatio-temporal association, primarily fo… | 23 | 1262 | maintenance |
| UKPLab/EasyNMT EasyNMT is a Python library providing easy access to state-of-the-art neural machine translation for 100+ languages using models like Opus-… | 23 | 1261 | maintenance |
| kuleshov/audio-super-res A Python/TensorFlow implementation of neural network models for audio super resolution, upsampling low-resolution audio to higher sampling … | 32 | 1260 | maintenance |
| google-research/planet Open-source implementation of the PlaNet agent, a purely model-based reinforcement learning algorithm that solves control tasks from pixels… | 10 | 1260 | maintenance |
| uclaml/SPIN Official implementation of Self-Play Fine-Tuning (SPIN), a method that improves language models by having them iteratively generate and dis… | 26 | 1254 | maintenance |
| ycjuan/kaggle-2014-criteo The winning '3 Idiots' solution code for the 2014 Kaggle Criteo Display Advertising Challenge, built around field-aware factorization machi… | 10 | 1254 | maintenance |
| maciejkula/glove-python A toy Python/Cython implementation of the GloVe algorithm for training dense word vector embeddings by factorizing the log of the word co-o… | 32 | 1253 | maintenance |
| davidmrau/mixture-of-experts A PyTorch re-implementation of the sparsely-gated Mixture-of-Experts layer from the paper 'Outrageously Large Neural Networks' by Shazeer e… | 32 | 1253 | maintenance |
| clovaai/CutMix-PyTorch Official PyTorch implementation of CutMix, an image data augmentation regularizer that cuts and pastes patches between training images whil… | 32 | 1251 | maintenance |
| ARM-software/ML-KWS-for-MCU TensorFlow models and training scripts for keyword spotting (wake-word detection) on Arm Cortex-M microcontrollers, accompanying the 'Hello… | 32 | 1250 | maintenance |
| kamalkraj/BERT-NER A PyTorch library for training and running named entity recognition (NER) models based on Google's BERT, evaluated on the CoNLL-2003 datase… | 32 | 1249 | maintenance |
| CQFIO/PhotographicImageSynthesis A TensorFlow implementation of the ICCV 2017 paper 'Photographic Image Synthesis with Cascaded Refinement Networks', which synthesizes phot… | 32 | 1248 | maintenance |
| daijifeng001/R-FCN MATLAB implementation of R-FCN, a region-based fully convolutional object detection framework described in a NIPS 2016 paper. It builds on … | 32 | 1248 | maintenance |
| RedaOps/ann-visualizer A Python library that generates presentable graph diagrams of Keras neural network architectures using Graphviz. It supports visualizing bo… | 10 | 1248 | maintenance |
| XiangLi1999/Diffusion-LM Diffusion-LM is the official research code for the paper 'Diffusion-LM Improves Controllable Text Generation', implementing a diffusion-bas… | 32 | 1246 | maintenance |
| KMnP/vpt Official PyTorch implementation of Visual Prompt Tuning (VPT), an ECCV 2022 method for parameter-efficient fine-tuning of vision transforme… | 32 | 1244 | maintenance |
| mravanelli/SincNet SincNet is a PyTorch neural architecture that processes raw audio waveforms using parametrized sinc band-pass filters in the first convolut… | 32 | 1243 | maintenance |
| SUC-DriverOld/MSST-WebUI A WebUI application for running inference with Music-Source-Separation-Training (MSST) and UVR/VR models, with preset processing flows, mod… | 69 | 1242 | maintenance |
| tf-encrypted/tf-encrypted TF Encrypted is a Python framework for privacy-preserving machine learning in TensorFlow, enabling training and prediction on encrypted dat… | 32 | 1242 | maintenance |
| ShannonAI/service-streamer Service Streamer is a Python middleware that queues discrete web service requests into mini-batches for deep learning model inference, impr… | 32 | 1241 | maintenance |
| lfz/DSB2017 The winning solution of team 'grt123' for the 2017 Data Science Bowl (DSB2017), a deep learning pipeline for detecting lung cancer from CT … | 32 | 1241 | maintenance |
| AlgoTraders/stock-analysis-engine A distributed stock analysis and backtesting framework that ingests automated pricing data from IEX Cloud, Tradier, and FinViz and runs tho… | 32 | 1241 | maintenance |
| VeritasYin/STGCN_IJCAI-18 Reference implementation of Spatio-Temporal Graph Convolutional Networks (STGCN) from the IJCAI 2018 paper, a deep learning framework for t… | 71 | 1239 | maintenance |
| autonomousvision/giraffe Official PyTorch implementation of GIRAFFE, a CVPR 2021 (oral, best paper award) generative model that represents scenes as compositional g… | 32 | 1238 | maintenance |
| BR-IDL/PaddleViT PaddleViT is a collection of state-of-the-art Vision Transformer and MLP model implementations for PaddlePaddle 2.1+, covering image classi… | 23 | 1238 | maintenance |
| facebookresearch/simsiam A PyTorch implementation of SimSiam, a simple Siamese-based self-supervised representation learning method from Facebook AI Research (CVPR … | 10 | 1237 | maintenance |
| jfzhang95/pytorch-video-recognition A PyTorch library implementing C3D, R3D, and R2Plus1D models for video action recognition, with training scripts for UCF101 and HMDB51 data… | 32 | 1236 | maintenance |
| ultralytics/JSON2YOLO A legacy Python toolkit that converts JSON-format annotation datasets (COCO, LabelMe, Labelbox, VoTT, INFOLKS, ATH) into the YOLO format fo… | 67 | 1235 | maintenance |
| jakesnell/prototypical-networks Reference PyTorch implementation of Prototypical Networks for few-shot classification from the NeurIPS 2017 paper. It includes training and… | 32 | 1234 | maintenance |
| AGI-Edgerunners/LLM-Adapters LLM-Adapters is a Python framework extending HuggingFace's PEFT library that integrates various adapter types (LoRA, series/parallel adapte… | 30 | 1234 | maintenance |
| cmu-db/ottertune OtterTune is a tool from the CMU Database Group that automatically finds good settings for a DBMS's configuration knobs using machine learn… | 10 | 1233 | maintenance |
| pollen-robotics/dtw A small Python module implementing Dynamic Time Warping (DTW), a similarity measure between temporal sequences. It offers a basic pure-Pyth… | 23 | 1232 | maintenance |
| yuanxiaosc/Entity-Relation-Extraction A TensorFlow and BERT based pipeline for joint entity and relation extraction, built as a solution for the 2019 Language and Intelligence C… | 32 | 1230 | maintenance |
| shobrook/BitVision BitVision is a terminal-based real-time charting and trading dashboard for the Bitstamp Bitcoin exchange, built with Node.js and Python. It… | 32 | 1229 | maintenance |
| PlayVoice/vits_chinese A Chinese text-to-speech library combining VITS with BERT-based prosody embeddings and NaturalSpeech infer-loss features, supporting ONNX c… | 23 | 1226 | maintenance |
| bheinzerling/bpemb BPEmb is a collection of pre-trained subword embeddings in 275 languages based on Byte-Pair Encoding, trained on Wikipedia, distributed as … | 32 | 1224 | maintenance |
| tylin/coco-caption The official evaluation code for the Microsoft COCO image captioning benchmark, implementing metrics such as BLEU, METEOR, ROUGE-L, CIDEr, … | 32 | 1224 | maintenance |
| google-research/fixmatch Official research code for FixMatch, a semi-supervised learning method that combines consistency regularization with confidence-based pseud… | 10 | 1224 | maintenance |
| ml4a/ml4a-ofx A collection of openFrameworks applications in C++ for real-time interactive machine learning, aimed at artists and creative coders. It inc… | 23 | 1223 | maintenance |
| iqiyi/FASPell FASPell is a Python-based Chinese spell checker based on the DAE-Decoder paradigm, published at the EMNLP 2019 W-NUT workshop. It detects a… | 32 | 1222 | maintenance |
| xvjiarui/GCNet Official PyTorch implementation of GCNet (Global Context Networks), a paper combining ideas from Non-Local Networks and Squeeze-Excitation … | 32 | 1220 | maintenance |
| Jhy1993/HAN Reference implementation of the Heterogeneous Graph Attention Network (HAN) model from the WWW 2019 paper, built on top of GAT in Python. I… | 32 | 1219 | maintenance |
| getnamo/TensorFlow-Unreal An Unreal Engine plugin that wraps TensorFlow machine learning operations as Actor Components accessible from C++, Blueprints, and Python. … | 23 | 1218 | maintenance |
| ucbdrive/few-shot-object-detection FsDet is the official implementation of the ICML 2020 paper 'Frustratingly Simple Few-Shot Object Detection' (TFA), built on detectron2. It… | 23 | 1218 | maintenance |
| mit-han-lab/tinyml MIT Han Lab's TinyML research repository containing projects like TinyTL and NetAug for memory-efficient deep learning on microcontrollers … | 32 | 1216 | maintenance |
| awslabs/sockeye Sockeye is an open-source sequence-to-sequence framework for Neural Machine Translation built on PyTorch, with distributed training and opt… | 23 | 1216 | maintenance |
| The-AI-Summer/self-attention-cv A PyTorch library implementing various self-attention mechanisms and transformer building blocks for computer vision, including multi-head … | 23 | 1215 | maintenance |
| anderskm/gputil GPUtil is a Python module that queries NVIDIA GPU status (memory usage, load, availability) by wrapping the nvidia-smi command-line tool. I… | 65 | 1214 | maintenance |
| nihui/realsr-ncnn-vulkan A command-line tool implementing the RealSR real-world super-resolution model using the ncnn inference framework with Vulkan GPU accelerati… | 23 | 1214 | maintenance |
| mouna99/dien A TensorFlow implementation of the Deep Interest Evolution Network (DIEN) for click-through rate prediction, based on the Alibaba research … | 32 | 1211 | maintenance |
| Shikhargupta/Spiking-Neural-Network A pure Python implementation of a hardware-efficient spiking neural network (SNN) trained with Spike-Time Dependent Plasticity (STDP). It i… | 32 | 1210 | maintenance |
| imfing/keras-flask-deploy-webapp A minimal, customizable Flask web app template for deploying Keras/TensorFlow image classification models. It ships with a mobile-friendly … | 32 | 1209 | maintenance |
| Alexander-H-Liu/End-to-end-ASR-Pytorch A PyTorch implementation of end-to-end automatic speech recognition (ASR), formerly known as Listen, Attend and Spell. It supports seq2seq … | 32 | 1208 | maintenance |
| YuliangXiu/ECON ECON is a research tool that reconstructs high-fidelity 3D clothed human avatars from a single color image by combining implicit and explic… | 32 | 1207 | maintenance |
| facebookresearch/ToMe ToMe (Token Merging) is a PyTorch library from Meta AI that speeds up existing Vision Transformers by merging similar tokens inside the net… | 10 | 1207 | maintenance |
| lessw2020/Ranger-Deep-Learning-Optimizer Ranger is a PyTorch deep learning optimizer that synergistically combines RAdam (Rectified Adam), LookAhead, and Gradient Centralization in… | 32 | 1206 | maintenance |
| bckenstler/CLR A Keras callback implementing cyclical learning rate (CLR) policies from Leslie Smith's paper, with built-in triangular, triangular2, and e… | 32 | 1206 | maintenance |
| reiinakano/arbitrary-image-stylization-tfjs A browser-based implementation of arbitrary neural style transfer built with TensorFlow.js, letting users stylize any photo in the style of… | 32 | 1205 | maintenance |
| aigamedev/scikit-neuralnetwork A Python library providing deep neural network classifiers and regressors with a scikit-learn-compatible API, wrapping Lasagne and Theano. … | 10 | 1204 | maintenance |
| twitter-research/tgn Official PyTorch implementation of Temporal Graph Networks (TGN), a framework for deep learning on dynamic graphs represented as sequences … | 32 | 1203 | maintenance |
| google-research/tapas TAPAS is Google Research's implementation of transformer-based models for question answering over tables, including pretrained checkpoints … | 10 | 1202 | maintenance |
| epfml/sent2vec Sent2vec is a C++ library with a Cython/Python interface that trains unsupervised distributed representations of sentences and short texts,… | 23 | 1201 | maintenance |
| henrywoo/chatllama An open-source Python library implementing RLHF-based fine-tuning of Meta's LLaMA models to build ChatGPT-style chat assistants, runnable o… | 22 | 1201 | maintenance |
| astroML/astroML AstroML is a Python library for machine learning, statistics, and data mining aimed at astronomy and astrophysics, built on numpy, scipy, s… | 32 | 1199 | maintenance |
| chainer/chainerrl ChainerRL is a deep reinforcement learning library implementing state-of-the-art algorithms like DQN, Rainbow, PPO, SAC, and A3C in Python … | 23 | 1199 | maintenance |
| DeepWisdom/AutoDL AutoDL is a fully automated deep learning framework from DeepWisdom that won the NeurIPS AutoDL challenge. It performs automatic feature en… | 23 | 1197 | maintenance |
| facebookresearch/mixup-cifar10 Official PyTorch implementation of mixup, a data augmentation technique that trains neural networks on convex combinations of image pairs a… | 10 | 1197 | maintenance |
| KaihuaTang/Scene-Graph-Benchmark.pytorch A PyTorch codebase for Scene Graph Generation (SGG) built on maskrcnn-benchmark, implementing methods from the CVPR 2020 paper 'Unbiased Sc… | 59 | 1195 | maintenance |
| huggingface/hmtl HMTL is a Hierarchical Multi-Task Learning model for NLP that jointly trains Named Entity Recognition, Entity Mention Detection, Relation E… | 32 | 1195 | maintenance |
| rinongal/StyleGAN-nada Official PyTorch implementation of StyleGAN-NADA, a CLIP-guided method for adapting pre-trained StyleGAN image generators to new domains us… | 32 | 1195 | maintenance |
| philipperemy/yolo-9000 A packaging of YOLO9000 (YOLOv2), a real-time object detection model that can detect over 9,000 object classes, built around the Darknet fr… | 32 | 1194 | maintenance |
| Kagami/go-face A Go library that provides face recognition by wrapping dlib's machine learning models, including face detection, landmark prediction, and … | 32 | 1194 | maintenance |
| rockchip-linux/rknn-toolkit2 RKNN-Toolkit2 is Rockchip's software development kit for converting trained AI models to RKNN format and deploying them on Rockchip NPU chi… | 23 | 1194 | maintenance |
| ROCm/MIOpen MIOpen is AMD's open-source library providing high-performance GPU kernels for machine learning primitives (convolutions, normalization, ac… | 93 | 1193 | maintenance |
| altosaar/variational-autoencoder A reference implementation of a variational autoencoder (VAE) in PyTorch, TensorFlow, and JAX, including an inverse autoregressive flow var… | 23 | 1192 | maintenance |
| khundman/telemanom Telemanom is a Python framework that uses vanilla LSTMs (Keras/TensorFlow) to detect anomalies in multivariate time series sensor data, wit… | 33 | 1189 | maintenance |
| elliottwu/unsup3d Official PyTorch implementation of the CVPR 2020 (Oral, Best Paper Award) research paper 'Unsupervised Learning of Probably Symmetric Defor… | 32 | 1189 | maintenance |
| aitorzip/DeepGTAV DeepGTAV is a C++ plugin for Grand Theft Auto V that converts the game into a vision-based self-driving car research environment. It expose… | 32 | 1188 | maintenance |
| lightaime/deep_gcns_torch A PyTorch library implementing DeepGCNs, DeeperGCN, and GNN1000 - graph convolutional network architectures that borrow residual/dense conn… | 32 | 1187 | maintenance |
| whai362/PSENet Official PyTorch implementation of PSENet, a progressive scale expansion network for robust scene text detection. It includes training, tes… | 23 | 1186 | maintenance |
| openai/procgen A benchmark of 16 procedurally-generated, game-like Gym environments for measuring how quickly reinforcement learning agents learn generali… | 56 | 1185 | maintenance |
| zju3dv/snake Official research code for 'Deep Snake for Real-Time Instance Segmentation' (CVPR 2020 oral), implementing a deep contour-based instance se… | 32 | 1185 | maintenance |
| kexinhuang12345/DeepPurpose DeepPurpose is a PyTorch-based deep learning library for molecular modeling, supporting drug-target interaction (DTI), drug property, drug-… | 23 | 1185 | maintenance |
| business-science/ai-data-science-team A Python library of specialized LLM-powered agents for common data science workflows such as data loading, cleaning, wrangling, visualizati… | 60 | 5394 | experimental |
| HyperGAN/HyperGAN HyperGAN is a composable GAN (generative adversarial network) framework built on PyTorch, offering both a Python API and a CLI with a user … | 23 | 1183 | maintenance |
| jankrepl/deepdow deepdow is a Python library built on PyTorch that connects deep learning with portfolio optimization, allowing networks that output asset w… | 32 | 1180 | maintenance |
| Smerity/sha-rnn A PyTorch implementation of the Single Headed Attention RNN (SHA-LSTM) from Stephen Merity's paper, combining an LSTM with a single attenti… | 32 | 1180 | maintenance |
| clovaai/voxceleb_trainer A PyTorch framework for training and evaluating speaker recognition and verification models on the VoxCeleb datasets. It implements multipl… | 67 | 1176 | maintenance |
| VoltaML/voltaML VoltaML is a lightweight Python library that compiles and optimizes machine learning and deep learning models for high-performance inferenc… | 32 | 1176 | maintenance |