domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| timesler/facenet-pytorch A PyTorch library providing pretrained face detection (MTCNN) and facial recognition (Inception ResNet V1) models, ported from the TensorFl… | 42 | 5162 | stable |
| open-mmlab/mmaction2 MMAction2 is OpenMMLab's PyTorch-based toolbox and benchmark for video understanding, covering action recognition, temporal action localiza… | 55 | 5142 | active |
| hiyouga/EasyR1 EasyR1 is an efficient, scalable reinforcement learning training framework for large language models and vision-language models, built as a… | 65 | 5129 | active |
| chatopera/Synonyms A Python library for Chinese synonym lookup and semantic similarity computation, built on word vector models for NLP and chatbot use cases.… | 52 | 5111 | active |
| grobidOrg/grobid GROBID is a Java machine learning library that extracts, parses, and re-structures raw documents such as PDFs into structured XML/TEI, focu… | 87 | 5105 | active |
| facebookresearch/AugLy AugLy is a Python data augmentation library from Meta AI supporting audio, image, text, and video with over 100 augmentations. It focuses o… | 67 | 5089 | stable |
| facebookresearch/co-tracker CoTracker is a transformer-based model from Meta AI and Oxford VGG that jointly tracks any point (pixel) across a video, handling occlusion… | 60 | 5080 | active |
| Deci-AI/super-gradients SuperGradients is an open-source PyTorch-based training library for building, training, and fine-tuning state-of-the-art computer vision mo… | 54 | 5052 | active |
| lightvector/KataGo KataGo is an open-source Go (baduk) engine trained via AlphaZero-like self-play, one of the strongest Go bots available. It runs as a GTP e… | 99 | 5036 | active |
| Kiln-AI/Kiln Kiln is a free desktop app and MIT-licensed Python library for the full AI development loop: evals, prompt optimization, RAG, agents, fine-… | 90 | 5034 | active |
| huggingface/text-embeddings-inference Text Embeddings Inference (TEI) is a Rust-based toolkit from Hugging Face for deploying and serving open-source text embedding, re-ranker, … | 84 | 5028 | active |
| microsoft/SPTAG SPTAG is a C++ library from Microsoft for large-scale approximate nearest neighbor (ANN) vector search, offering kd-tree/balanced k-means t… | 76 | 5012 | active |
| Plachtaa/VITS-fast-fine-tuning A Python pipeline for fast fine-tuning of VITS text-to-speech models, enabling speaker adaptation in under an hour from short audio, long a… | 10 | 5012 | active |
| ArthurBrussee/brush Brush is a 3D reconstruction engine using Gaussian splatting, built in Rust on the Burn ML framework and WebGPU. It trains and renders spla… | 65 | 4992 | active |
| sktime/pytorch-forecasting A PyTorch-based Python library for time series forecasting with state-of-the-art deep learning architectures. It provides a high-level API … | 92 | 4977 | active |
| shibing624/text2vec A Python library that converts text (words, sentences, paragraphs) into vector embeddings, implementing Word2Vec, RankBM25, BERT, Sentence-… | 53 | 4974 | active |
| pytorch/executorch ExecuTorch is PyTorch's framework for exporting and running AI models on-device across mobile, embedded, and edge hardware, with a tiny (~5… | 95 | 4953 | active |
| microsoft/muzic Muzic is a Microsoft Research project providing deep learning models for music understanding and generation, including MusicBERT, SongMASS,… | 66 | 4952 | active |
| going-doer/Paper2Code PaperCoder is a multi-agent LLM system that automatically generates a code repository from a machine learning research paper via a planning… | 52 | 4925 | active |
| Picovoice/porcupine Porcupine is a highly-accurate, lightweight on-device wake word detection engine powered by deep neural networks. It enables always-listeni… | 73 | 4922 | stable |
| apple/embedding-atlas Embedding Atlas is an open-source tool from Apple for interactive visualization of large embeddings and their metadata, supporting clusteri… | 86 | 4921 | active |
| arrayfire/arrayfire ArrayFire is a general-purpose tensor/numerical computing library for C, C++, and Python that accelerates array operations on GPUs (CUDA, O… | 57 | 4902 | stable |
| KaiyangZhou/deep-person-reid Torchreid is a PyTorch library for deep-learning person re-identification, supporting both image and video reid with end-to-end training an… | 50 | 4900 | stable |
| Nixtla/statsforecast StatsForecast is a Python library providing fast, scalable implementations of statistical and econometric univariate time series forecastin… | 93 | 4887 | stable |
| PaddlePaddle/VisualDL VisualDL is a deep learning visualization toolkit for PaddlePaddle that provides charts for tracking training metrics, visualizing model st… | 24 | 4884 | stable |
| google-deepmind/alphageometry Google DeepMind's implementation of AlphaGeometry and DDAR, AI systems that solve Olympiad-level geometry theorem proving problems. It comb… | 55 | 4879 | active |
| deepjavalibrary/djl Deep Java Library (DJL) is an engine-agnostic, high-level deep learning framework for Java that supports backends like PyTorch, TensorFlow,… | 80 | 4842 | active |
| zju3dv/EasyMocap EasyMocap is an open-source Python toolbox for markerless human motion capture and novel view synthesis from RGB videos. It fits parametric… | 54 | 4783 | active |
| dimforge/nalgebra nalgebra is a general-purpose linear algebra library for Rust providing statically-sized and dynamically-sized vectors and matrices, along … | 73 | 4779 | stable |
| pytorch/ignite PyTorch-Ignite is a high-level library for training and evaluating neural networks in PyTorch flexibly and transparently. It provides an En… | 89 | 4778 | stable |
| SeldonIO/seldon-core Seldon Core 2 is an MLOps and LLMOps framework for deploying, managing, and scaling machine learning models and modular AI applications on … | 72 | 4778 | active |
| Rasa Open Source Rasa Open Source is a Python framework for building text- and voice-based conversational AI assistants, combining NLU (intent/entity recogn… | 65 | 21306 | maintenance |
| facebookresearch/lingua Meta Lingua is a minimal, fast LLM training and inference library built on easy-to-modify PyTorch components for research purposes. It supp… | 36 | 4766 | active |
| py-why/EconML EconML is a Python package from Microsoft Research's ALICE project for estimating heterogeneous treatment effects from observational data u… | 87 | 4763 | active |
| FluxML/Flux.jl Flux.jl is a machine learning library written entirely in Julia, providing lightweight abstractions over Julia's native GPU support and aut… | 98 | 4739 | stable |
| rust-ml/linfa Linfa is a Rust machine learning framework offering classical ML algorithms and preprocessing, inspired by Python's scikit-learn. It is org… | 82 | 4736 | active |
| cvg/LightGlue LightGlue is a deep neural network library that matches sparse local features across image pairs with high accuracy and fast inference. It … | 50 | 4728 | stable |
| esimov/pigo Pigo is a pure Go library for fast face detection, pupil/eye localization, and facial landmark detection based on the Pixel Intensity Compa… | 31 | 4728 | stable |
| cjlin1/libsvm LIBSVM is a widely used library and command-line toolset for support vector machine classification, regression, and distribution estimation… | 59 | 4708 | stable |
| facebookresearch/flow_matching A PyTorch library for implementing flow matching algorithms with continuous, discrete, and Riemannian flow matching implementations. It acc… | 48 | 4705 | active |
| mindspore-ai/mindspore MindSpore is an open-source deep learning framework for training and inference across mobile, edge, and cloud scenarios. It provides automa… | 32 | 4700 | active |
| google-deepmind/dm_control Google DeepMind's Python stack for physics-based simulation and reinforcement learning environments built on the MuJoCo physics engine. It … | 94 | 4672 | stable |
| PKU-Alignment/align-anything Align-Anything is a modular Python framework for aligning any-to-any (all-modality) large models with human intentions and values using fee… | 48 | 4667 | active |
| makcedward/nlpaug nlpaug is a Python library for augmenting textual, audio, and spectrogram data to improve machine learning model performance. It provides A… | 66 | 4665 | active |
| RLinf/RLinf RLinf is an open-source, flexible and scalable reinforcement learning training infrastructure for embodied AI (vision-language-action model… | 74 | 4655 | active |
| Blealtan/efficient-kan An efficient pure-PyTorch implementation of Kolmogorov-Arnold Networks (KAN) that reformulates the computation as matrix multiplications ov… | 24 | 4655 | active |
| Tencent/TNN TNN is a high-performance, lightweight deep learning inference framework developed by Tencent Youtu Lab, supporting mobile, desktop, and se… | 32 | 4648 | active |
| OpenNMT/CTranslate2 CTranslate2 is a C++ and Python library for fast, memory-efficient inference of Transformer models on CPU and GPU. It uses quantization, la… | 96 | 4644 | stable |
| promptslab/Promptify Promptify is a Python library that turns LLMs into task-based NLP engines, providing high-level APIs for NER, classification, QA, and custo… | 65 | 4635 | active |
| Rikorose/DeepFilterNet DeepFilterNet is a low-complexity speech enhancement framework that performs real-time noise suppression on full-band 48kHz audio using dee… | 23 | 4632 | active |
| ObjectBox ObjectBox is a fast, lightweight embedded NoSQL object database with built-in on-device vector search, available for Java/Kotlin (Android/J… | 88 | 4622 | active |
| deepseek-ai/Engram Official implementation of Engram, a conditional memory module from DeepSeek that modernizes N-gram embeddings for O(1) lookup as a new spa… | 43 | 4614 | active |
| joanrod/star-vector StarVector is a foundation model that generates scalable vector graphics (SVG) code from images and text by treating vectorization as a cod… | 49 | 4560 | active |
| RecBole RecBole is a unified, comprehensive and efficient recommendation library built on Python and PyTorch for reproducing and developing recomme… | 26 | 4541 | stable |
| OAID/Tengine Tengine is a lightweight, high-performance, modular deep learning inference engine developed by OPEN AI LAB for embedded and edge devices. … | 27 | 4531 | active |
| NVlabs/tiny-cuda-nn A small, self-contained C++/CUDA framework for training and querying neural networks, featuring a lightning-fast fully fused MLP and a vers… | 58 | 4528 | active |
| facebookresearch/vjepa2 Official PyTorch codebase and pretrained models for V-JEPA 2, a self-supervised video encoder trained on internet-scale video, plus V-JEPA … | 52 | 4527 | active |
| dedupeio/dedupe Dedupe is a Python library that uses machine learning to perform fuzzy matching, deduplication, and entity resolution on structured data. I… | 48 | 4507 | stable |
| openxla/xla XLA (Accelerated Linear Algebra) is an open-source machine learning compiler that optimizes models from frameworks like PyTorch, TensorFlow… | 77 | 4499 | active |
| mcmonkeyprojects/SwarmUI SwarmUI (formerly StableSwarmUI) is a modular, self-hosted web user interface for AI image generation, supporting models like Stable Diffus… | 76 | 4499 | active |
| real-stanford/diffusion_policy Official PyTorch implementation of Diffusion Policy, a visuomotor robot policy learning method that represents robot behavior as a conditio… | 30 | 4491 | stable |
| salesforce/Merlion Merlion is a Python library from Salesforce providing an end-to-end machine learning framework for time series intelligence, covering forec… | 10 | 4491 | active |
| zjunlp/DeepKE DeepKE is an open-source PyTorch-based knowledge extraction toolkit for knowledge graph construction, supporting named entity recognition, … | 64 | 4472 | active |
| Mobile ALOHA Mobile ALOHA is an open-source system for low-cost whole-body teleoperation and data collection with a bimanual mobile manipulation robot, … | 29 | 4465 | active |
| layumi/Person_reID_baseline_pytorch A small, friendly PyTorch baseline implementation for person and vehicle re-identification (ReID). It reproduces strong top-conference resu… | 65 | 4446 | stable |
| modelscope/ClearerVoice-Studio ClearerVoice-Studio is an open-source, AI-powered speech processing toolkit from ModelScope/Alibaba offering state-of-the-art pretrained mo… | 38 | 4445 | active |
| rom1504/img2dataset A Python tool that downloads large sets of image URLs and packages them into machine learning datasets, with resizing and caption support. … | 56 | 4443 | active |
| unionai-oss/pandera Pandera is a Python library for statistical data validation of dataframe-like objects, supporting pandas, polars, PySpark, and others. It l… | 96 | 4442 | stable |
| mosaicml/llm-foundry LLM Foundry is a PyTorch-based codebase for training, finetuning, evaluating, and deploying large language models from 125M to 70B+ paramet… | 65 | 4441 | active |
| ssrajadh/sentrysearch SentrySearch is a Python CLI tool that provides semantic search over video footage by embedding video chunks with Gemini Embedding, Alibaba… | 65 | 4433 | active |
| xlite-dev/lite.ai.toolkit A lightweight C++ toolkit providing unified APIs for 100+ pre-trained AI models across inference backends like ONNX Runtime, MNN, TensorRT,… | 74 | 4427 | active |
| tensorflow/probability TensorFlow Probability is a Python library for probabilistic reasoning and statistical analysis built on TensorFlow, with a JAX substrate. … | 66 | 4427 | active |
| DistrictDataLabs/yellowbrick Yellowbrick is a Python library of visual diagnostic tools called Visualizers that extend the scikit-learn API using matplotlib. It helps u… | 26 | 4400 | active |
| ROCm/hip HIP is a C++ runtime API and kernel programming language for AMD GPUs that mirrors the NVIDIA CUDA programming interface. It enables develo… | 95 | 4392 | active |
| iperov/DeepFaceLab DeepFaceLab is the leading open-source Windows application for creating deepfakes, allowing users to swap, de-age, or replace faces and hea… | 10 | 19292 | maintenance |
| onnxsim/onnxsim ONNX Simplifier (onnxsim) is a tool that simplifies ONNX models by inferring the whole computation graph and replacing redundant operators … | 98 | 4390 | active |
| microsoft/FLAML FLAML is a lightweight Python library for automated machine learning (AutoML) and hyperparameter tuning. It efficiently finds quality model… | 89 | 4390 | active |
| lululxvi/deepxde DeepXDE is a Python library for scientific machine learning and physics-informed learning, built on top of PyTorch, TensorFlow, JAX, and Pa… | 81 | 4386 | stable |
| EvolvingLMMs-Lab/lmms-eval lmms-eval is a unified Python framework for evaluating large multimodal models across text, image, video, and audio tasks, with 100+ benchm… | 85 | 4377 | active |
| open-compass/VLMEvalKit VLMEvalKit is an open-source Python toolkit for evaluating large vision-language models (LMMs/LVLMs) across 80+ benchmarks with support for… | 62 | 4359 | active |
| AI4Finance-Foundation/ElegantRL ElegantRL is a lightweight, modular deep reinforcement learning library built on PyTorch that implements core model-free RL algorithms (PPO… | 53 | 4355 | active |
| lucas-maes/le-wm LeWorldModel (LeWM) is the official PyTorch codebase for a JEPA-based world model that trains stably end-to-end from raw pixels using only … | 52 | 4344 | active |
| ymcui/Chinese-LLaMA-Alpaca A project releasing Chinese-adapted LLaMA and instruction-tuned Alpaca large language models with extended Chinese vocabulary, plus pre-tra… | 56 | 18931 | maintenance |
| tloen/alpaca-lora A repository of scripts for reproducing Stanford Alpaca-style instruction tuning of LLaMA models using low-rank adaptation (LoRA) on consum… | 30 | 18906 | maintenance |
| rust-ndarray/ndarray ndarray is a Rust crate providing an n-dimensional array container for general elements and numerics, with owned arrays, lightweight array … | 85 | 4314 | active |
| ourownstory/neural_prophet NeuralProphet is a Python library for interpretable time series forecasting built on PyTorch, combining neural networks with traditional ti… | 23 | 4295 | active |
| unum-cloud/USearch USearch is a fast, single-file similarity search and clustering engine for vectors and arbitrary objects, supporting spatial, binary, proba… | 93 | 4278 | active |
| Nixtla/neuralforecast NeuralForecast is a Python library offering a large collection of state-of-the-art neural forecasting models (NBEATS, NHITS, TFT, PatchTST,… | 98 | 4257 | active |
| SysCV/sam-hq HQ-SAM (Segment Anything in High Quality) upgrades Meta's Segment Anything Model with a learnable High-Quality Output Token for accurate ze… | 48 | 4255 | active |
| ThilinaRajapakse/simpletransformers Simple Transformers is a Python library built on Hugging Face Transformers that lets users train, fine-tune, and evaluate Transformer model… | 61 | 4254 | active |
| MaartenGr/KeyBERT KeyBERT is a minimal Python library that extracts keywords and keyphrases from documents using BERT embeddings and cosine similarity. It re… | 67 | 4215 | stable |
| SylphAI-Inc/AdalFlow AdalFlow is a PyTorch-like Python library for building and auto-optimizing LLM application workflows such as chatbots, RAG pipelines, and a… | 67 | 4211 | active |
| facebookresearch/nevergrad Nevergrad is a Python toolbox for gradient-free (derivative-free) optimization, providing many optimization algorithms such as NGOpt and di… | 76 | 4204 | active |
| cvg/Hierarchical-Localization hloc is a modular Python toolbox for state-of-the-art 6-DoF visual localization, combining image retrieval and feature matching (SuperPoint… | 48 | 4194 | active |
| kubeflow/pipelines Kubeflow Pipelines is a Kubernetes-native platform for building, deploying, and orchestrating reusable end-to-end machine learning workflow… | 97 | 4193 | stable |
| iterative/cml CML (Continuous Machine Learning) is an open-source CLI tool that brings CI/CD practices to machine learning projects on GitHub, GitLab, an… | 45 | 4187 | active |
| bbfamily/abu AbuQuant (abupy) is an open-source Python quantitative trading and investment framework supporting stocks, options, futures, and Bitcoin, w… | 52 | 18239 | maintenance |
| SwanHubX/SwanLab SwanLab is an open-source AI training tracking and visualization platform with a Python SDK, CLI, and cloud or self-hosted dashboard. It in… | 89 | 4175 | active |
| facebookresearch/vggt-omega VGGT-Omega is a research library from Oxford VGG and Meta AI providing pretrained transformer models for 3D vision tasks such as camera pos… | 58 | 4165 | active |
| llm-d/llm-d llm-d is a Kubernetes-native distributed LLM inference serving stack that orchestrates model servers like vLLM and SGLang across clusters. … | 82 | 4164 | active |