domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| mit-han-lab/torchquantum TorchQuantum is a PyTorch-based framework for quantum computing simulation, supporting statevector and pulse-level simulation on GPUs with … | 64 | 1662 | active |
| meyda/meyda Meyda is a JavaScript audio feature extraction library that computes standard signal features like RMS, spectral centroid, and spectral fla… | 23 | 1662 | stable |
| SciML/ModelingToolkit.jl ModelingToolkit.jl is a Julia-based acausal, equation-based modeling framework and computer algebra system for symbolic-numeric computation… | 95 | 1661 | active |
| thtrieu/darkflow Darkflow is a Python library that translates Darknet's YOLO neural network definitions to TensorFlow, enabling real-time object detection a… | 32 | 6139 | maintenance |
| MgArcher/Text_select_captcha A PyTorch-based deep learning system that recognizes click-based (text-select) CAPTCHAs by detecting and ordering Chinese character positio… | 69 | 1656 | active |
| Data-Centric-AI-Community/fg-data-synthetic A Python library for generating synthetic tabular and time-series data using state-of-the-art generative models such as GANs (including CTG… | 72 | 1654 | active |
| mhamilton723/FeatUp FeatUp is a model-agnostic framework that upsamples the spatial resolution of deep neural network features by 16-32x without changing their… | 16 | 1654 | active |
| maziarraissi/PINNs The original reference implementation of Physics-Informed Neural Networks (PINNs), which train neural networks to solve and discover nonlin… | 62 | 6111 | maintenance |
| cubiq/ComfyUI_IPAdapter_plus A ComfyUI custom node implementation of IPAdapter models for image-to-image conditioning in Stable Diffusion workflows. It transfers the su… | 35 | 6110 | maintenance |
| lightly-ai/lightly-train LightlyTrain is a Python framework for training computer vision models, covering pretraining of vision foundation models (DINOv2/v3) on unl… | 85 | 1652 | active |
| skrub-data/skrub skrub is a Python library that facilitates machine learning with dataframes, providing preprocessing, encoding, and wrangling tools for tab… | 93 | 1650 | active |
| sunlabuiuc/PyHealth PyHealth is an open-source Python toolkit for clinical deep learning, unifying healthcare datasets (EHRs, physiological signals, medical im… | 91 | 1650 | active |
| bytedance/DreamO DreamO is the official PyTorch implementation of a unified framework for image customization, built on FLUX diffusion models. It supports t… | 35 | 1649 | active |
| XueZeyue/DanceGRPO Official implementation of DanceGRPO, a framework applying Group Relative Policy Optimization (GRPO) to fine-tune visual generation models … | 40 | 1648 | active |
| vllm-project/vllm-metal A community-maintained hardware plugin that enables vLLM to run LLM inference on Apple Silicon Macs using MLX as the primary compute backen… | 79 | 1647 | active |
| Anemll/Anemll ANEMLL is an open-source Python library and toolchain for converting and running large language models on the Apple Neural Engine (ANE) via… | 61 | 1647 | active |
| JIA-Lab-research/ControlNeXt ControlNeXt is the official implementation of a controllable generation method for images and videos, built on Stable Diffusion XL, Stable … | 24 | 1646 | active |
| airockchip/rknn-llm RKLLM is Rockchip's software stack for converting, quantizing, and running large language models on Rockchip NPU chips such as RK3588 and R… | 79 | 1644 | active |
| huggingface/aisheets Hugging Face AI Sheets is an open-source no-code web application for building, enriching, and transforming datasets using AI models. It can… | 58 | 1642 | active |
| opendilab/LightZero LightZero is a lightweight, open-source Python toolkit that unifies Monte Carlo Tree Search (MCTS) with deep reinforcement learning, implem… | 67 | 1636 | active |
| facebookincubator/submitit Submitit is a lightweight Python library for submitting Python functions as jobs to a Slurm cluster, wrapping submission and providing acce… | 51 | 1635 | active |
| meta-llama/synthetic-data-kit A Python CLI tool from Meta for generating high-quality synthetic datasets to fine-tune LLMs. It follows a four-stage pipeline (ingest, cre… | 42 | 1632 | active |
| facebookresearch/DomainBed DomainBed is a PyTorch suite containing benchmark datasets and algorithms for domain generalization research. It implements many algorithms… | 10 | 1630 | active |
| Lucchetto/SuperImage SuperImage is an Android application that upscales low-resolution images using a Real-ESRGAN neural network running on-device via the MNN d… | 10 | 1630 | active |
| yoshitomo-matsubara/torchdistill torchdistill is a modular, configuration-driven PyTorch framework for knowledge distillation and general deep learning experiments, requiri… | 86 | 1629 | active |
| AnswerDotAI/rerankers A lightweight, dependency-free Python library providing a unified API for using all common reranking and cross-encoder models. It lets deve… | 42 | 1629 | active |
| snorkel-team/snorkel Snorkel is a Python library for programmatically building and managing training data using weak supervision, letting users write labeling f… | 62 | 6001 | maintenance |
| InterDigitalInc/CompressAI CompressAI is a PyTorch library and evaluation platform for end-to-end learned data compression research. It provides custom layers, entrop… | 73 | 1627 | active |
| ZhangGe6/onnx-modifier onnx-modifier is a web-based visual editor for ONNX neural network models, built on Netron for graph visualization and Flask as the backend… | 72 | 1627 | active |
| apple/ml-mobileclip Official implementation of Apple's MobileCLIP and MobileCLIP2, fast image-text contrastive (CLIP-style) models optimized for mobile latency… | 60 | 1627 | active |
| scikit-learn-contrib/boruta_py A Python implementation of the Boruta all-relevant feature selection method, built on scikit-learn. It identifies all features carrying pre… | 46 | 1627 | active |
| paperswithcode/ai-deadlines A web application that displays countdown timers to submission deadlines for top-tier AI, machine learning, computer vision, NLP, and robot… | 32 | 5998 | maintenance |
| gomlx/gomlx GoMLX is an accelerated machine learning and math framework for Go, comparable to PyTorch/JAX/TensorFlow. It offers differentiable operator… | 95 | 1621 | active |
| mlcommons/inference Reference implementations of the MLPerf Inference benchmark suite, the industry-standard benchmarks for measuring how fast ML systems run m… | 83 | 1621 | active |
| pavlin-policar/openTSNE openTSNE is a modular Python library implementing t-Distributed Stochastic Neighbor Embedding (t-SNE) for visualizing high-dimensional data… | 66 | 1621 | stable |
| bowang-lab/scGPT scGPT is a foundation model for single-cell multi-omics built with a generative transformer, providing pretrained checkpoints and a Python … | 67 | 1620 | active |
| jeffffffli/HybrIK HybrIK is the official PyTorch implementation of a hybrid analytical-neural inverse kinematics method for 3D human pose and shape estimatio… | 23 | 1618 | stable |
| aiwaves-cn/agents Agents 2.0 is a Python framework for data-centric, self-evolving autonomous language agents based on symbolic learning. It adapts neural-ne… | 28 | 5957 | maintenance |
| csinva/imodels A Python package providing scikit-learn-compatible implementations of interpretable machine-learning models such as rule lists, RuleFit, an… | 85 | 1617 | active |
| pixeltable/pixeltable Pixeltable is a Python library providing declarative, incremental data infrastructure for multimodal AI applications, unifying storage of i… | 92 | 1613 | active |
| PaddlePaddle/PaddleSlim PaddleSlim is an open-source library built on PaddlePaddle for deep learning model compression and architecture search. It provides low-bit… | 50 | 1612 | active |
| open-gigaai/giga-world-0 GigaWorld-0 is a unified world model framework that acts as a data engine for Vision-Language-Action (VLA) learning in embodied AI. It comb… | 41 | 1612 | active |
| PKU-Alignment/safe-rlhf Beaver is a modular open-source framework from Peking University for training large language models with SFT, RLHF, and Safe RLHF (constrai… | 53 | 1611 | active |
| gorgonia/gorgonia Gorgonia is a Go library for machine learning that lets you define and evaluate mathematical equations over multidimensional arrays using a… | 23 | 5929 | maintenance |
| Robbyant/lingbot-depth LingBot-Depth is a PyTorch-based model and toolkit for masked depth modeling that transforms incomplete, noisy depth sensor data into metri… | 56 | 1610 | active |
| chainer/chainer Chainer is a Python-based deep learning framework that pioneered the define-by-run approach with dynamic computational graphs and automatic… | 23 | 5924 | maintenance |
| NVIDIA/RULER RULER is a benchmark framework from NVIDIA that generates synthetic test examples to measure the true effective context size of long-contex… | 66 | 1609 | active |
| dmlc/gluon-cv GluonCV is a deep learning toolkit providing state-of-the-art computer vision model implementations with 170+ pre-trained models. It suppor… | 23 | 5916 | maintenance |
| pypose/pypose PyPose is a PyTorch-based Python library for differentiable robotics on manifolds, combining deep perceptual models with physics-based opti… | 91 | 1605 | active |
| pq-yang/MatAnyone MatAnyone is a CVPR 2025 human video matting framework that extracts alpha mattes of target people from video using consistent memory propa… | 54 | 1605 | active |
| meituan/YOLOv6 YOLOv6 is a single-stage object detection framework implemented in PyTorch, designed for industrial applications with a family of pretraine… | 23 | 5895 | maintenance |
| apache/opennlp Apache OpenNLP is a machine learning based Java toolkit for processing natural language text, supporting tasks like tokenization, sentence … | 97 | 1602 | stable |
| ml4a/ml4a ml4a is a Python library and collection of Jupyter notebooks for making art with machine learning. It wraps popular deep learning models li… | 32 | 1602 | active |
| Kruk2/jasna Jasna is a GPU-accelerated tool that detects and restores mosaics in JAV videos and still images, with a native GUI, CLI, and streaming sup… | 81 | 1601 | active |
| JustGlowing/minisom MiniSom is a minimalistic, NumPy-based Python implementation of Self Organizing Maps (SOM), a type of neural network for unsupervised learn… | 75 | 1601 | stable |
| time-series-foundation-models/lag-llama Lag-Llama is the first open-source foundation model for probabilistic time series forecasting, built on a transformer architecture. It prov… | 37 | 1601 | active |
| bilibili/ailab Bilibili's AI lab repository, best known for Real-CUGAN, a deep learning model for anime image super-resolution (upscaling). It provides pr… | 23 | 5881 | maintenance |
| autonomousvision/transfuser Official PyTorch implementation of TransFuser, a transformer-based multi-modal sensor fusion model for end-to-end autonomous driving, publi… | 53 | 1599 | stable |
| SonyResearch/micro_diffusion Official implementation of Sony Research's micro-budget approach to training large-scale text-to-image diffusion transformer models from sc… | 24 | 1594 | active |
| facebookresearch/fast3r Fast3R is the official PyTorch implementation of a CVPR 2025 model from Meta FAIR that reconstructs 3D scenes and estimates camera poses fr… | 10 | 1593 | active |
| google/uncertainty-baselines A library of high-quality, minimal-dependency implementations of standard and state-of-the-art uncertainty and robustness methods for deep … | 77 | 1592 | active |
| alibaba/Pai-Megatron-Patch Pai-Megatron-Patch is an open-source deep learning training toolkit from Alibaba Cloud for large-scale training and inference of LLMs and V… | 56 | 1591 | active |
| spotify/voyager Voyager is an in-memory approximate nearest-neighbor search library implementing the HNSW algorithm, with bindings for Python and Java (and… | 55 | 1591 | stable |
| Tencent-Hunyuan/HunyuanWorld-Voyager HunyuanWorld-Voyager is a video diffusion framework from Tencent Hunyuan that generates world-consistent RGBD video and 3D point-cloud sequ… | 52 | 1590 | active |
| samim23/polymath Polymath is a Python CLI tool that uses machine learning to convert any music library into a searchable music production sample-library. It… | 31 | 1589 | active |
| intel/auto-round AutoRound is Intel's advanced quantization toolkit for LLMs and vision-language models, using sign-gradient descent to achieve high accurac… | 91 | 1588 | active |
| microsoft/Semi-supervised-learning USB (Unified Semi-supervised learning Benchmark) is a PyTorch-based codebase from Microsoft for semi-supervised learning across computer vi… | 65 | 1588 | active |
| databricks/megablocks MegaBlocks is a lightweight Python library for efficient training of mixture-of-experts (MoE) models, built around its dropless-MoE (dMoE) … | 62 | 1588 | active |
| zju3dv/EasyVolcap EasyVolcap is a PyTorch-based library for accelerating neural volumetric video research, covering volumetric video capture, reconstruction,… | 27 | 1587 | active |
| DeepGraphLearning/torchdrug TorchDrug is a PyTorch-based machine learning library for drug discovery, covering graph neural networks, deep generative models, and reinf… | 23 | 1587 | active |
| pytorch/FBGEMM FBGEMM is a collection of highly optimized low-precision matrix multiplication and convolution kernels for server-side deep learning infere… | 93 | 1584 | active |
| scikit-learn-contrib/MAPIE MAPIE is a scikit-learn-compatible Python library for quantifying uncertainty in machine learning models via conformal prediction. It compu… | 93 | 1584 | active |
| microsoft/MMdnn MMdnn is a Microsoft toolkit for converting, visualizing, and diagnosing deep learning models across frameworks such as TensorFlow, PyTorch… | 39 | 5805 | maintenance |
| meta-pytorch/torchtune Torchtune is a PyTorch-native library for authoring, post-training, and experimenting with large language models. It provides hackable trai… | 69 | 5801 | maintenance |
| SalesforceAIResearch/uni2ts Uni2TS is a PyTorch library for unified pre-training, fine-tuning, inference, and evaluation of universal time series forecasting transform… | 60 | 1580 | active |
| MatrAIx-ai/MatrAIx-Persona-8B MatrAIx is a population-scale, persona-driven evaluation framework that instantiates sampled persona records as LLM agents to simulate hete… | 57 | 1580 | active |
| kotaro-kinoshita/yomitoku YomiToku is an AI-powered document image analysis engine specialized for Japanese, providing full-text OCR, layout analysis, table structur… | 87 | 1579 | active |
| RWKV/rwkv.cpp A C++ port of the RWKV language model to the ggml tensor library, providing FP32, FP16, and quantized INT4/INT5/INT8 inference focused on C… | 37 | 1579 | active |
| tensorflow/model-optimization The TensorFlow Model Optimization Toolkit (tfmot) is a Python library providing tools to optimize machine learning models for deployment, i… | 82 | 1578 | stable |
| CodeReclaimers/neat-python A pure-Python implementation of NEAT (NeuroEvolution of Augmenting Topologies), an algorithm for evolving neural network topologies and wei… | 79 | 1578 | active |
| DiffEqML/torchdyn Torchdyn is a PyTorch library dedicated to numerical deep learning, providing tools for neural differential equations, implicit models, and… | 23 | 1578 | active |
| NVlabs/sionna Sionna is an open-source, GPU-accelerated, differentiable Python library from NVIDIA for research on communication systems. It comprises Si… | 83 | 1577 | active |
| stanfordnlp/pyreft pyreft is Stanford NLP's Python library for Representation Finetuning (ReFT), which adapts frozen language models by learning task-specific… | 54 | 1577 | active |
| edbeeching/godot_rl_agents Godot RL Agents is an open-source Python package that bridges games built in the Godot Engine with reinforcement learning algorithms, enabl… | 64 | 1575 | active |
| Layout-Parser/layout-parser LayoutParser is a Python toolkit for deep learning based document image analysis, offering unified APIs for layout detection models, layout… | 23 | 5774 | maintenance |
| karpathy/arxiv-sanity-preserver A self-hostable web application for browsing, searching, and filtering recent arXiv submissions, with personalized paper recommendations ba… | 32 | 5772 | maintenance |
| BloodAxe/pytorch-toolbelt A Python library of PyTorch extensions providing building blocks for fast R&D prototyping, including encoder-decoder architectures, special… | 44 | 1574 | active |
| Tencent/DepthCrafter DepthCrafter is a diffusion-based video depth estimation model from Tencent AI Lab that generates temporally consistent long depth sequence… | 38 | 1574 | active |
| photosynthesis-team/piq PyTorch Image Quality (PIQ) is a collection of measures and metrics for image quality assessment in image-to-image tasks, written in pure P… | 23 | 1574 | stable |
| ArztSamuel/Applying_EANNs A 2D Unity simulation where cars learn to navigate courses using a feedforward neural network trained by a modified genetic algorithm. It s… | 43 | 1571 | stable |
| gcorso/DiffDock DiffDock is a deep learning implementation of a diffusion generative model for molecular docking, predicting how small molecule ligands bin… | 31 | 1569 | active |
| FluxML/Zygote.jl Zygote.jl is a source-to-source automatic differentiation library for Julia that hooks into the Julia compiler to generate gradient (backwa… | 97 | 1568 | active |
| quixio/quix-streams Quix Streams is a pure Python framework for building real-time data pipelines and event-driven applications on Apache Kafka using a Streami… | 97 | 1568 | active |
| Babyhamsta/Aimmy Aimmy is a universal AI-based aim alignment mechanism (aim assist) for gamers with impairments, built in C# using YOLOv8 models run via ONN… | 78 | 1568 | active |
| MLSysOps/MLE-agent MLE-Agent is an LLM-powered CLI agent that acts as a pairing assistant for machine learning engineers and researchers. It automates ML task… | 67 | 1568 | active |
| Xilinx/brevitas Brevitas is a PyTorch library for neural network quantization supporting both post-training quantization (PTQ) and quantization-aware train… | 91 | 1567 | active |
| Chinese-Text-Classification-Pytorch A PyTorch-based collection of ready-to-run Chinese text classification models including TextCNN, TextRNN, FastText, TextRCNN, BiLSTM with a… | 32 | 5728 | maintenance |
| sdv-dev/CTGAN CTGAN is a Python library of deep learning based synthetic data generators for single-table tabular data, implementing the CTGAN conditiona… | 80 | 1562 | active |
| evo-design/evo Evo is a 7-billion-parameter DNA foundation model built on the StripedHyena architecture, trained on ~300 billion tokens of prokaryotic who… | 71 | 1560 | active |
| real-stanford/universal_manipulation_interface Universal Manipulation Interface (UMI) is a data collection and policy learning framework that transfers in-the-wild human demonstrations i… | 63 | 1560 | active |