domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| Duankaiwen/CenterNet Official PyTorch implementation of CenterNet, a one-stage keypoint-triplet-based object detector built on CornerNet. It trains from scratch… | 32 | 1884 | maintenance |
| hexiangnan/neural_collaborative_filtering Reference implementation of the WWW'17 Neural Collaborative Filtering paper, providing GMF, MLP, and NeuMF models for implicit-feedback rec… | 32 | 1883 | maintenance |
| mpatacchiola/deepgaze Deepgaze is a Python computer vision library for human-computer interaction built on OpenCV and TensorFlow. It provides CNN-based head pose… | 32 | 1880 | maintenance |
| open-mmlab/mmaction MMAction is an open-source PyTorch toolbox for video action understanding, covering action recognition, temporal action detection, and spat… | 32 | 1876 | maintenance |
| tradytics/surpriver Surpriver is a Python application that uses machine learning and anomaly detection to find stocks with unusual price and volume patterns th… | 32 | 1874 | maintenance |
| google/TensorNetwork TensorNetwork is a Python library for building and contracting tensor networks, with backends for TensorFlow, JAX, PyTorch, and NumPy. It p… | 10 | 1870 | maintenance |
| pierluigiferrari/ssd_keras A Keras implementation of the Single Shot MultiBox Detector (SSD) object detection architecture, with ports of the original trained weights… | 23 | 1869 | maintenance |
| facebookresearch/DPR DPR is a set of tools and pretrained models for dense passage retrieval in open-domain question answering, based on the EMNLP 2020 paper fr… | 10 | 1869 | maintenance |
| waleedka/hiddenlayer A lightweight Python library for rendering readable neural network architecture graphs and plotting training metrics for PyTorch, TensorFlo… | 23 | 1867 | maintenance |
| minimaxir/automl-gs automl-gs is a Python AutoML tool that takes an input CSV and a target prediction field and automatically generates a trained machine learn… | 23 | 1866 | maintenance |
| facebookresearch/TimeSformer Official PyTorch implementation of TimeSformer, a space-time attention transformer model for video understanding from an ICML 2021 paper. I… | 10 | 1863 | maintenance |
| hanzhanggit/StackGAN TensorFlow implementation of StackGAN, a stacked generative adversarial network that synthesizes photo-realistic images from text descripti… | 32 | 1860 | maintenance |
| raminmh/liquid_time_constant_networks Official code repository for Liquid Time-Constant Networks (LTCs), a class of continuous-time recurrent neural networks with varying time-c… | 32 | 1857 | maintenance |
| magicleap/Atlas Atlas is a PyTorch-based deep learning model from Magic Leap that performs end-to-end 3D scene reconstruction from posed RGB images, produc… | 32 | 1857 | maintenance |
| MrNothing/AI-Blocks AI-Blocks is a WYSIWYG desktop application for visually building machine learning models by dragging objects with attached scripts in a sce… | 23 | 1857 | maintenance |
| nlpodyssey/spago Spago is a self-contained machine learning and NLP library written in pure Go, featuring an asynchronous computation graph with automatic d… | 29 | 1850 | maintenance |
| ZhengyaoJiang/PGPortfolio PGPortfolio is a Python library implementing a deep reinforcement learning framework for financial portfolio management, based on the paper… | 23 | 1849 | maintenance |
| WangYueFt/dgcnn Author's reference implementation of Dynamic Graph CNN (DGCNN), a neural network module called EdgeConv for learning on 3D point clouds. It… | 32 | 1846 | maintenance |
| ermongroup/ddim Official PyTorch implementation of Denoising Diffusion Implicit Models (DDIM), a faster sampling method for diffusion probabilistic models.… | 32 | 1845 | maintenance |
| Score-Based Generative Modeling through SDEs Official PyTorch implementation of the ICLR 2021 paper 'Score-Based Generative Modeling through Stochastic Differential Equations', unifyin… | 32 | 1844 | maintenance |
| google/gemmlowp gemmlowp is a small, self-contained, headers-only C++ library for low-precision (quantized) general matrix multiplication (GEMM). It provid… | 32 | 1843 | maintenance |
| AlexeyAB/Yolo_mark A Windows and Linux GUI application for drawing bounding boxes around objects in images to create labeled training data for YOLO v2/v3 obje… | 32 | 1840 | maintenance |
| microsoft/Cream A collection of Microsoft research projects on neural architecture search (NAS) and efficient Vision Transformers, including AutoFormer, Ti… | 23 | 1839 | maintenance |
| zetane/viewer Zetane Viewer is a free desktop application that visualizes machine learning models and their internal tensors in 3D, showing architecture,… | 23 | 1839 | maintenance |
| google-deepmind/kinetics-i3d A repository of pre-trained Inflated 3D Convnet (I3D) models for video action classification, trained on the Kinetics dataset, released alo… | 32 | 1838 | maintenance |
| minimaxir/aitextgen aitextgen is a Python library for training and generating text with GPT-2 and GPT Neo models, built on PyTorch, Hugging Face Transformers, … | 23 | 1838 | maintenance |
| anthropics/jacobian-lens A Python library implementing the Jacobian lens, an interpretability technique that linearly transports residual-stream activations at any … | 56 | 1836 | maintenance |
| byzer-org/byzer-lang Byzer (formerly MLSQL) is a low-code, SQL-like distributed programming language and engine for data pipelines, analytics, and AI, built aro… | 23 | 1835 | maintenance |
| bubbliiiing/faster-rcnn-pytorch A PyTorch implementation of the Faster R-CNN two-stage object detection model, supporting training on VOC-format datasets with ResNet or VG… | 23 | 1833 | maintenance |
| Kyubyong/tacotron A heavily documented TensorFlow implementation of Tacotron, a fully end-to-end text-to-speech synthesis model. It includes training, prepro… | 32 | 1832 | maintenance |
| NVlabs/stylegan2-ada NVIDIA's official TensorFlow implementation of StyleGAN2 with adaptive discriminator augmentation (ADA), enabling high-quality GAN training… | 32 | 1829 | maintenance |
| NVIDIA/semantic-segmentation NVIDIA's PyTorch monorepo implementing the paper 'Hierarchical Multi-Scale Attention for Semantic Segmentation', with pretrained weights an… | 32 | 1828 | maintenance |
| THUDM/CogDL CogDL is a Python/PyTorch toolkit for graph deep learning that lets researchers train and compare GNN and graph embedding models on tasks l… | 23 | 1821 | maintenance |
| yassouali/pytorch-segmentation A PyTorch library implementing multiple semantic segmentation models (DeepLab V3+, PSPNet, U-Net, SegNet, FCN, ENet, and others) with datas… | 26 | 1818 | maintenance |
| shunsukesaito/PIFu PyTorch implementation of PIFu (Pixel-Aligned Implicit Function), a deep learning method that reconstructs high-resolution 3D surfaces and … | 32 | 1817 | maintenance |
| OpenPPL/ppq PPQ (PPL Quantization Tool) is an extensible, high-performance offline neural network quantization framework written in Python. It parses a… | 23 | 1816 | maintenance |
| google/compare_gan A TensorFlow library from Google Research providing configurable implementations of GAN components: losses, penalties, normalization techni… | 10 | 1815 | maintenance |
| yongzhuo/Keras-TextClassification A Keras-based library for Chinese text classification supporting long/short text, multi-label classification, and sentence pair similarity.… | 23 | 1808 | maintenance |
| rosinality/vq-vae-2-pytorch A PyTorch implementation of VQ-VAE-2 for generating diverse high-fidelity images, including training scripts for the VQ-VAE and PixelSNAIL … | 32 | 1803 | maintenance |
| xmartlabs/Bender Bender is a Swift library that provides an abstraction layer over Apple's MetalPerformanceShaders for defining and running neural networks … | 23 | 1802 | maintenance |
| symisc/sod SOD is an embedded, cross-platform computer vision and machine learning library written in C, distributed as a single dependency-free amalg… | 23 | 1798 | maintenance |
| Trusted-AI/AIX360 AI Explainability 360 is an open-source Python library from IBM Research offering a comprehensive set of algorithms for interpreting and ex… | 66 | 1796 | maintenance |
| MCG-NJU/VideoMAE Official PyTorch implementation of VideoMAE, a masked autoencoder method for data-efficient self-supervised video pre-training with video t… | 32 | 1784 | maintenance |
| scikit-learn-contrib/lightning A Python library for large-scale linear classification, regression, and ranking that follows scikit-learn API conventions. It supports dens… | 10 | 1782 | maintenance |
| OryxProject/oryx Oryx 2 is a Java framework implementing the lambda architecture on Apache Spark and Apache Kafka, specialized for real-time large-scale mac… | 10 | 1782 | maintenance |
| bazingagin/npc_gzip A Python library and research codebase implementing the parameter-free 'kNN-gzip' text classification method from the ACL 2023 paper, which… | 20 | 1780 | maintenance |
| idiap/fast-transformers A PyTorch library providing fast transformer implementations with efficient attention mechanisms such as linear attention, designed to scal… | 32 | 1779 | maintenance |
| jquesnelle/yarn Reference implementation of YaRN, an efficient method for extending the context window of large language models, published as an ICLR 2024 … | 29 | 1777 | maintenance |
| wkentaro/pytorch-fcn A PyTorch implementation of Fully Convolutional Networks (FCN) for semantic segmentation, including training code that reproduces the origi… | 10 | 1777 | maintenance |
| impira/docquery DocQuery is a Python library and CLI tool that uses large language models to answer questions about semi-structured and unstructured docume… | 32 | 1775 | maintenance |
| qiuqiangkong/audioset_tagging_cnn PANNs is a collection of large-scale pretrained CNN audio neural networks trained on AudioSet (5000 hours, 527 classes) for audio pattern r… | 32 | 1772 | maintenance |
| princeton-vl/CornerNet-Lite CornerNet-Lite is the official PyTorch implementation of the paper 'CornerNet-Lite: Efficient Keypoint Based Object Detection', providing t… | 32 | 1771 | maintenance |
| facebookresearch/TensorComprehensions Tensor Comprehensions is a C++ library with a Python API that provides a domain-specific language for expressing machine learning workloads… | 10 | 1767 | maintenance |
| nywang16/Pixel2Mesh Pixel2Mesh is the official TensorFlow implementation of the ECCV 2018 paper that generates 3D mesh models from a single RGB image using a g… | 32 | 1765 | maintenance |
| cmu-db/noisepage NoisePage is a relational database management system from Carnegie Mellon University designed for autonomous, self-driving operation, with … | 10 | 1765 | maintenance |
| starry-sky6688/MARL-Algorithms PyTorch implementations of classic multi-agent reinforcement learning algorithms (IQL, QMIX, VDN, COMA, QTRAN, MAVEN, CommNet, DyMA-CL, G2A… | 32 | 1760 | maintenance |
| WXinlong/SOLO Official PyTorch implementation of SOLO and SOLOv2, box-free fully convolutional methods for instance segmentation published at ECCV 2020 a… | 32 | 1758 | maintenance |
| OpenCodeInterpreter/OpenCodeInterpreter OpenCodeInterpreter is a suite of open-source code generation models and deployment code that integrates code execution and iterative refin… | 25 | 1757 | maintenance |
| Tencent/Metis Metis is a learnware platform for AIOps from Tencent, currently open-sourcing a time-series anomaly detection learnware. It combines statis… | 10 | 1757 | maintenance |
| salesforce/ALBEF Official PyTorch implementation of ALBEF, a vision-and-language pre-training method that aligns image and text representations before fusin… | 10 | 1755 | maintenance |
| enzoampil/fastquant fastquant is a Python library that makes backtesting and optimizing trading strategies accessible with as few as 3 lines of code. It also p… | 32 | 1754 | maintenance |
| deepset-ai/FARM FARM is a Python framework for fine-tuning and evaluating transformer-based language models for NLP tasks, with a focus on question answeri… | 10 | 1752 | maintenance |
| Lightning-Universe/lightning-bolts Lightning Bolts is a toolbox of pre-built models, callbacks, and datasets that extend PyTorch Lightning for AI/ML research and production. … | 10 | 1751 | maintenance |
| libfann/fann Fast Artificial Neural Network (FANN) is a free open-source C library implementing multilayer neural networks with fully and sparsely conne… | 73 | 1750 | maintenance |
| VertaAI/modeldb ModelDB is an open-source system for machine learning model versioning, metadata tracking, and experiment management. It versions models al… | 23 | 1749 | maintenance |
| ScottfreeLLC/AlphaPy AlphaPy is a Python machine learning framework built on scikit-learn, pandas, Keras, XGBoost, LightGBM, and CatBoost for building classific… | 40 | 1745 | maintenance |
| magenta/mt3 MT3 is a multi-instrument automatic music transcription model built on the T5X framework, converting audio recordings into multitrack MIDI.… | 73 | 1744 | maintenance |
| facebookresearch/deepcluster DeepCluster is a research codebase from Facebook AI Research implementing the 'Deep Clustering for Unsupervised Learning of Visual Features… | 10 | 1743 | maintenance |
| google-research/robotics_transformer A Python library from Google Research implementing RT-1 (Robotics Transformer), a transformer model for end-to-end robotic control with an … | 10 | 1743 | maintenance |
| tkipf/gae A TensorFlow implementation of (Variational) Graph Auto-Encoders from Kipf & Welling's 2016 paper, supporting unsupervised learning, cluste… | 32 | 1742 | maintenance |
| dav/word2vec The original Google word2vec tool, hosted on GitHub with community patches for Mac OS X compilation and memory fixes. It efficiently trains… | 32 | 1741 | maintenance |
| bearpaw/pytorch-classification A PyTorch library providing a unified interface for training image classification models on CIFAR-10/100 and ImageNet. It includes implemen… | 32 | 1738 | maintenance |
| openai/Video-Pre-Training OpenAI's Video PreTraining (VPT) codebase for learning Minecraft agents by watching unlabeled online videos, including behavioral cloning a… | 41 | 1737 | maintenance |
| zijundeng/pytorch-semantic-segmentation A PyTorch library providing implementations of popular semantic segmentation models such as FCN, U-Net, SegNet, PSPNet, GCN, and DUC/HDC. I… | 32 | 1735 | maintenance |
| zhougr1993/DeepInterestNetwork A TensorFlow demo implementation of the Deep Interest Network (DIN) paper for click-through rate prediction, run on the Amazon electronics … | 32 | 1733 | maintenance |
| experiencor/keras-yolo2 A Keras/TensorFlow implementation of the YOLOv2 real-time object detection model with support for training on custom datasets. It offers mu… | 23 | 1733 | maintenance |
| Teachable Machine Teachable Machine is a web-based tool by Google Creative Lab that lets anyone train and use simple machine learning models (image, sound, p… | 72 | 1731 | maintenance |
| NVIDIA/Cosmos-Tokenizer NVIDIA Cosmos Tokenizer is a suite of neural tokenizers for images and videos that convert visual data into continuous latents or discrete … | 10 | 1731 | maintenance |
| Qihoo360/hbox Hbox (formerly XLearning) is a scheduling platform that runs machine learning and deep learning frameworks like TensorFlow, MXNet, Caffe, a… | 47 | 1727 | maintenance |
| google-research/torchsde A PyTorch library providing differentiable stochastic differential equation (SDE) solvers with GPU support and efficient backpropagation vi… | 10 | 1726 | maintenance |
| Franck-Dernoncourt/NeuroNER NeuroNER is a Python/TensorFlow program that performs named-entity recognition (NER) using neural networks, with pretrained models and an e… | 23 | 1723 | maintenance |
| DeepMotionEditing/deep-motion-editing A PyTorch-based end-to-end library for editing and rendering 3D character motion, built on SIGGRAPH 2020 research. It provides deep learnin… | 32 | 1722 | maintenance |
| Lightning-Universe/lightning-flash Lightning Flash is a high-level PyTorch library built on PyTorch Lightning that provides ready-made 'recipes' for over 15 AI tasks across 7… | 10 | 1722 | maintenance |
| dunbar12138/pix2pix3D pix2pix3D is the official PyTorch implementation of a CVPR 2023 paper on 3D-aware conditional image synthesis. It generates 3D objects (neu… | 31 | 1716 | maintenance |
| allenai/scibert SciBERT is a BERT-based pretrained language model trained on a large multi-domain corpus of scientific publications, released by the Allen … | 32 | 1712 | maintenance |
| sipeed/MaixPy-v1 MaixPy-v1 is a MicroPython port for the Kendryte K210 RISC-V AI chip, letting users program Sipeed Maix boards in Python. It provides APIs … | 23 | 1712 | maintenance |
| Maratyszcza/NNPACK NNPACK is a C99 acceleration package providing high-performance SIMD and multi-core CPU implementations of neural network layers, especiall… | 32 | 1710 | maintenance |
| airaria/TextBrewer TextBrewer is a PyTorch-based knowledge distillation toolkit for NLP models. It provides an easy-to-use framework implementing state-of-the… | 23 | 1708 | maintenance |
| thunlp/OpenNE OpenNE is an open-source Python toolkit for network embedding (graph representation learning) that unifies training and testing interfaces … | 32 | 1706 | maintenance |
| microsoft/i-Code Microsoft's i-Code is a collection of research models and frameworks for integrative, composable multimodal AI spanning vision, language, a… | 32 | 1703 | maintenance |
| imcaspar/gpt2-ml GPT2-ML is a TensorFlow-based GPT-2 training and inference project supporting multiple languages, with a focus on Chinese. It provides 1.5B… | 23 | 1701 | maintenance |
| NVIDIA/DeepRecommender A research library implementing deep autoencoders for collaborative filtering in recommender systems, based on the paper 'Training Deep Aut… | 10 | 1701 | maintenance |
| CPJKU/madmom Madmom is a Python audio signal processing library with a strong focus on music information retrieval (MIR) tasks. It provides reference im… | 56 | 1699 | maintenance |
| tensorflow/addons TensorFlow Addons is a collection of extra layers, optimizers, losses, metrics, activations, callbacks, and image ops that conform to Tenso… | 10 | 1699 | maintenance |
| hanleyweng/CoreML-in-ARKit A Swift demo/template iOS app that runs CoreML object detection (Inception V3) on live ARKit camera frames and renders 3D text labels above… | 32 | 1697 | maintenance |
| VITA-Group/TransGAN Official PyTorch implementation of TransGAN, a NeurIPS 2021 paper that builds a GAN whose generator and discriminator are both pure transfo… | 32 | 1695 | maintenance |
| YoongiKim/AutoCrawler A Python multiprocess image web crawler that downloads images from Google and Naver image search using Selenium and ChromeDriver. It is des… | 32 | 1691 | maintenance |
| graph4ai/graph4nlp Graph4NLP is a Python library for applying Graph Neural Networks to Natural Language Processing tasks, built on DGL and PyTorch. It provide… | 23 | 1690 | maintenance |
| ELS-RD/transformer-deploy A Python tool that optimizes Hugging Face transformer models for production inference using ONNX Runtime and NVIDIA TensorRT, deploying the… | 23 | 1689 | maintenance |
| adobe/antialiased-cnns A PyTorch library providing antialiased CNN models and a BlurPool layer from the ICML 2019 paper 'Making Convolutional Networks Shift-Invar… | 23 | 1683 | maintenance |