domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| starVLA/starVLA StarVLA is an open-source, Lego-like modular codebase for developing Vision-Language-Action (VLA) models for generalist robots. It unifies … | 69 | 3530 | active |
| fnnx-ai/scikit-llm Scikit-LLM is a Python library that integrates large language models like GPT into scikit-learn workflows as estimators. It enables tasks s… | 72 | 3529 | active |
| google-research/big_vision Google Research's official Jax/Flax codebase for training large-scale vision models such as Vision Transformer, SigLIP, MLP-Mixer, and LiT … | 42 | 3528 | active |
| truera/trulens TruLens is a Python library for evaluating and tracking LLM-based applications and AI agents. It provides OpenTelemetry-native tracing of e… | 95 | 3525 | active |
| edtechre/pybroker PyBroker is a Python framework for developing algorithmic trading strategies with a focus on machine learning. It provides a fast NumPy/Num… | 93 | 3515 | active |
| visionml/pytracking PyTracking is a PyTorch-based framework for visual object tracking and video object segmentation, providing official implementations of tra… | 23 | 3514 | active |
| langwatch/langwatch LangWatch is an open-source LLMOps platform for LLM evaluation, AI agent testing, tracing, and production observability. It combines agent … | 90 | 3511 | active |
| google-deepmind/alphafold Open-source implementation of the AlphaFold 2 inference pipeline for predicting protein structures from amino acid sequences, including Alp… | 58 | 14811 | maintenance |
| NVIDIA/TransformerEngine Transformer Engine is an NVIDIA library for accelerating Transformer model training and inference on NVIDIA GPUs using low-precision format… | 99 | 3504 | active |
| Farama-Foundation/PettingZoo PettingZoo is a Python library providing a standard API for multi-agent reinforcement learning environments, analogous to Gymnasium but for… | 89 | 3498 | active |
| aiming-lab/MetaClaw MetaClaw is a continual meta-learning framework that lets an LLM agent evolve from real conversations, combining skill synthesis from failu… | 68 | 3497 | active |
| mne-tools/mne-python MNE-Python is an open-source Python library for exploring, visualizing, and analyzing human neurophysiological data such as MEG, EEG, sEEG,… | 88 | 3495 | stable |
| Anttwo/SuGaR SuGaR is the official PyTorch implementation of a CVPR 2024 method that extracts accurate, editable meshes from 3D Gaussian Splatting recon… | 27 | 3495 | active |
| aleju/imgaug imgaug is a Python library for augmenting images in machine learning experiments, converting a small set of input images into a much larger… | 23 | 14741 | maintenance |
| borisdayma/dalle-mini DALL·E Mini is a Python library and model that generates images from a text prompt, available via pip and hosted on Hugging Face Model Hub.… | 23 | 14740 | maintenance |
| helblazer811/ManimML ManimML is a Python library built on the Manim Community Library for creating animations and visualizations of machine learning concepts li… | 23 | 3492 | active |
| facebookresearch/ijepa Official PyTorch implementation of I-JEPA, a self-supervised learning method that predicts latent representations of image regions from oth… | 10 | 3489 | active |
| NVIDIA/Model-Optimizer NVIDIA Model Optimizer (ModelOpt) is a Python library of state-of-the-art model optimization techniques including quantization, pruning, di… | 91 | 3488 | active |
| POSTECH-CVLab/PyTorch-StudioGAN PyTorch-StudioGAN is a PyTorch library providing unified implementations of representative GAN architectures (BigGAN, StyleGAN2/3, etc.) fo… | 23 | 3487 | stable |
| Docta-ai/docta Docta is a Python library for data-centric AI that diagnoses and fixes issues in datasets such as label errors, noise, and misannotations. … | 69 | 3486 | active |
| common-voice/common-voice Mozilla Common Voice is a web platform for crowdsourcing voice donations to build public-domain speech datasets for training voice recognit… | 95 | 3484 | active |
| huggingface/optimum Optimum is a Hugging Face library that extends Transformers, Diffusers, timm, and Sentence Transformers with hardware-specific optimization… | 98 | 3469 | active |
| QData/TextAttack TextAttack is a Python framework for generating adversarial examples against NLP models, as well as for data augmentation and model trainin… | 87 | 3469 | active |
| vietanhdev/anylabeling AnyLabeling is a desktop image annotation tool that combines LabelImg/Labelme-style manual labeling with AI-assisted auto-labeling. It runs… | 85 | 3463 | active |
| PaddlePaddle/PARL PARL is a flexible, high-performance reinforcement learning framework built on PaddlePaddle, providing Model/Algorithm/Agent abstractions a… | 41 | 3453 | active |
| shenweichen/DeepCTR-Torch DeepCTR-Torch is a PyTorch library providing easy-to-use, modular, and extendable implementations of deep-learning-based CTR (click-through… | 78 | 3450 | active |
| ob-f/OpenBot OpenBot is an open-source project that turns Android smartphones into the brains of low-cost robots, paired with a ~$50 electric vehicle bo… | 67 | 3442 | active |
| PennyLaneAI/pennylane PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. It lets users build an… | 92 | 3441 | stable |
| XinJingHao/DRL-Pytorch A unified PyTorch implementation collection of popular deep reinforcement learning algorithms including DQN variants, PPO, DDPG, TD3, SAC, … | 44 | 3436 | active |
| NVlabs/GR00T-WholeBodyControl NVIDIA's unified platform for developing, training, and deploying whole-body controllers for humanoid robots, including the decoupled WBC m… | 61 | 3428 | active |
| NVlabs/stylegan The official TensorFlow implementation of StyleGAN, NVIDIA's style-based generator architecture for generative adversarial networks from th… | 32 | 14416 | maintenance |
| Qwen3-ASR Qwen3-ASR is a family of open-source speech recognition models from Alibaba's Qwen team, supporting ASR and language identification across … | 55 | 3423 | active |
| aqlaboratory/openfold OpenFold is a faithful, trainable PyTorch reproduction of DeepMind's AlphaFold 2 for protein structure prediction. It is memory-efficient a… | 48 | 3420 | active |
| microsoft/nni NNI (Neural Network Intelligence) is an open-source AutoML toolkit from Microsoft that automates hyperparameter tuning, neural architecture… | 10 | 14361 | maintenance |
| davidsandberg/facenet A TensorFlow implementation of the FaceNet face recognizer that generates 128-dimensional face embeddings, including face detection via MTC… | 32 | 14343 | maintenance |
| AlexanderKoch-Koch/low_cost_robot An open-source project with files and Python control code for building a ~$250 low-cost robot arm using Dynamixel XL430 and XL330 servo mot… | 25 | 3408 | active |
| embeddings-benchmark/mteb MTEB (Massive Text Embedding Benchmark) is a Python library and CLI for benchmarking embedding models across 1000+ tasks, languages, and mo… | 95 | 3406 | active |
| NovaSky-AI/SkyThought SkyThought is the open-source repository behind Sky-T1, a family of reasoning language models trained for under $450, including training sc… | 25 | 3399 | active |
| IQA-PyTorch A pure Python/PyTorch toolbox for image quality assessment (IQA) providing GPU-accelerated reimplementations of many full-reference and no-… | 82 | 3380 | active |
| alibaba/ROLL ROLL is an open-source reinforcement learning library from Alibaba for training large language models at scale, supporting algorithms like … | 78 | 3374 | active |
| WongKinYiu/yolov7 Official PyTorch implementation of the YOLOv7 paper, a state-of-the-art real-time object detector with trainable bag-of-freebies techniques… | 23 | 14139 | maintenance |
| CompVis/latent-diffusion The official research code and pretrained model zoo for Latent Diffusion Models (LDM), the paper behind Stable Diffusion, enabling high-res… | 32 | 14133 | maintenance |
| nv-tlabs/kimodo Kimodo is NVIDIA's official implementation of a kinematic motion diffusion model trained on 700 hours of motion capture data to generate hi… | 56 | 3365 | active |
| NVIDIA/cuda-python NVIDIA's official Python package for accessing the CUDA platform, providing Pythonic APIs and low-level bindings to CUDA C APIs, parallel a… | 95 | 3359 | active |
| Lagrange-Labs/deep-prove DeepProve is a Rust framework for generating zero-knowledge proofs of neural network inference, including end-to-end proving of full LLM fo… | 60 | 3357 | active |
| libAudioFlux/audioFlux audioFlux is a C-based library with Python bindings for audio and music analysis and feature extraction. It supports dozens of time-frequen… | 53 | 3351 | active |
| VainF/Torch-Pruning Torch-Pruning is a PyTorch framework for structural neural network pruning based on the DepGraph algorithm from CVPR 2023. It automatically… | 50 | 3348 | active |
| shankarpandala/lazypredict Lazy Predict is a Python library that trains dozens of machine learning models with minimal code to quickly identify which algorithms perfo… | 78 | 3347 | active |
| magenta/ddsp DDSP is a Python library of differentiable digital signal processing components (synthesizers, filters, waveshapers) that can be embedded i… | 64 | 3344 | active |
| google-ai-edge/LiteRT LiteRT is Google's successor to TensorFlow Lite, an on-device runtime for high-performance ML and GenAI inference on edge platforms. It pro… | 85 | 3339 | active |
| HKUDS/VideoRAG VideoRAG is a retrieval-augmented generation framework for chatting with and understanding extremely long-context videos, using a dual-chan… | 53 | 3337 | active |
| lakehq/sail Sail is an open-source, Rust-native multimodal compute engine that serves as a drop-in replacement for Apache Spark, unifying batch process… | 93 | 3333 | active |
| opengeos/geoai GeoAI is a Python package that integrates artificial intelligence with geospatial data analysis, built on PyTorch, Transformers, and segmen… | 89 | 3327 | active |
| pgmpy/pgmpy pgmpy is a Python library for causal and probabilistic reasoning with graphical models such as Bayesian Networks, Dynamic Bayesian Networks… | 81 | 3318 | stable |
| RKNN-Toolkit2 RKNN-Toolkit2 is Rockchip's SDK for converting trained neural network models into RKNN format and deploying them on Rockchip NPU chips like… | 36 | 3313 | active |
| modelscope/evalscope EvalScope is a Python framework from the ModelScope community for evaluating large language models, vision-language models, embedding model… | 93 | 3312 | active |
| huggingface/datatrove DataTrove is a Python library from Hugging Face for processing, filtering, and deduplicating large-scale text data through customizable pip… | 86 | 3308 | active |
| microsoft/LoRA loralib is the official PyTorch implementation of LoRA (Low-Rank Adaptation), which fine-tunes large language models by injecting trainable… | 23 | 13767 | maintenance |
| mljar/mljar-supervised MLJAR AutoML is a Python package for automated machine learning on tabular data, providing feature engineering, hyperparameter tuning, mode… | 91 | 3287 | active |
| hbb1/2d-gaussian-splatting Official implementation of 2D Gaussian Splatting (2DGS), a SIGGRAPH 2024 method that represents scenes as 2D oriented Gaussian disks for ge… | 69 | 3279 | stable |
| vladmandic/human Human is a JavaScript/TypeScript library built on TensorFlow.js that combines multiple ML models for 3D face detection and recognition, bod… | 48 | 3264 | active |
| MAIF/shapash Shapash is a Python library that makes machine learning models interpretable and understandable through clear visualizations, a webapp, and… | 90 | 3251 | active |
| cocktailpeanut/fluxgym FluxGym is a simple web UI for training FLUX LoRA models with low VRAM support (12GB/16GB/20GB). It combines the AI-Toolkit Gradio frontend… | 65 | 3246 | active |
| determined-ai/determined Determined is an open-source deep learning platform that combines distributed training, hyperparameter tuning, experiment tracking, and GPU… | 39 | 3236 | active |
| Jittor/jittor Jittor is a high-performance deep learning framework from Tsinghua University based on just-in-time (JIT) compilation and meta-operators, w… | 67 | 3229 | active |
| onnx/onnx-tensorrt A C++ parser library and backend that converts ONNX models into TensorRT engines for high-performance GPU inference. It is maintained by NV… | 92 | 3228 | active |
| MzeroMiko/VMamba VMamba is a PyTorch implementation of a visual state space model (SSM) vision backbone based on Mamba, featuring 2D Selective Scan (SS2D) f… | 21 | 3219 | active |
| kerlomz/captcha_trainer A deep learning training tool for image CAPTCHA recognition built on TensorFlow, using CNN/ResNet/DenseNet backbones with GRU/LSTM recurren… | 55 | 3213 | active |
| NVIDIA/physicsnemo NVIDIA PhysicsNeMo is an open-source Python deep-learning framework for building, training, fine-tuning, and inferring physics AI models us… | 89 | 3198 | active |
| Pointcept Pointcept is a PyTorch-based research codebase for point cloud perception, providing implementations of state-of-the-art 3D scene understan… | 76 | 3196 | active |
| facebookresearch/dinov2 PyTorch implementation and pretrained models for DINOv2, a self-supervised vision transformer method from Meta AI that learns robust visual… | 68 | 13266 | maintenance |
| LeelaChessZero/lc0 Lc0 is an open-source, UCI-compliant chess engine that plays chess using neural networks trained via AlphaZero-style self-play reinforcemen… | 65 | 3193 | active |
| Nerogar/OneTrainer OneTrainer is a GUI and CLI application for fine-tuning diffusion image models, supporting full fine-tuning, LoRA, and embeddings across ma… | 74 | 3184 | active |
| ARM-software/ComputeLibrary Arm's Compute Library is a C++ collection of over 100 low-level machine learning and computer vision functions optimized for Arm Cortex-A/N… | 96 | 3183 | active |
| tslearn-team/tslearn tslearn is a Python machine learning library dedicated to time series analysis, built on numpy and compatible with scikit-learn. It provide… | 88 | 3174 | stable |
| facebookresearch/tribev2 TRIBE v2 is a multimodal deep learning model from Meta AI that predicts fMRI brain responses to naturalistic video, audio, and text stimuli… | 55 | 3172 | active |
| webdataset/webdataset A Python library providing a high-performance sequential I/O system based on tar-shard files for large-scale deep learning training, with s… | 52 | 3169 | stable |
| qdrant/fastembed FastEmbed is a lightweight Python library for generating text embeddings using ONNX Runtime instead of PyTorch, requiring no GPU and minima… | 83 | 3166 | active |
| tekaratzas/RustGPT A transformer-based large language model implemented entirely in pure Rust with no external ML frameworks, using only ndarray for matrix op… | 38 | 3157 | active |
| stemrollerapp/stemroller StemRoller is a free open-source desktop app that separates vocals, drums, bass, and other stems from any song with a single click, using F… | 84 | 3156 | active |
| parrt/dtreeviz dtreeviz is a Python library for visualizing decision trees and interpreting tree-based machine learning models. It supports scikit-learn, … | 69 | 3155 | active |
| Djdefrag/QualityScaler QualityScaler is a Windows GUI application that uses AI deep-learning models to upscale, enhance, and de-noise images and videos. It is wri… | 95 | 3138 | active |
| scikit-learn-contrib/hdbscan A high-performance Python implementation of the HDBSCAN hierarchical density-based clustering algorithm, part of the scikit-learn-contrib e… | 85 | 3136 | stable |
| z-x-yang/Segment-and-Track-Anything An open-source pipeline (SAM-Track) that segments and tracks arbitrary objects in videos using the Segment Anything Model for key-frame seg… | 61 | 3134 | active |
| docarray/docarray DocArray is a Python library for representing, transmitting, storing, and retrieving multimodal data such as text, images, audio, and video… | 56 | 3124 | active |
| modelscope/3D-Speaker 3D-Speaker is an open-source Python toolkit for single- and multi-modal speaker verification, speaker recognition, and speaker diarization,… | 56 | 3121 | active |
| timescale/pgvectorscale pgvectorscale is a PostgreSQL extension written in Rust that complements pgvector with a StreamingDiskANN index, statistical binary quantiz… | 75 | 3116 | active |
| jina-ai/clip-as-service CLIP-as-service is a low-latency, high-scalability server for embedding images and text into fixed-length vectors using OpenAI's CLIP model… | 23 | 12836 | maintenance |
| MakazhanAlpamys/Soup Soup is a Python CLI that fine-tunes and post-trains LLMs from a single YAML config, supporting 23 methods (SFT, DPO, ORPO, SimPO, KTO, etc… | 77 | 3102 | active |
| ddangelov/Top2Vec Top2Vec is a Python library that learns jointly embedded topic, document, and word vectors for topic modeling and semantic search. It suppo… | 23 | 3102 | stable |
| thuml/Time-Series-Library TSLib is an open-source Python library providing a unified codebase of advanced deep learning models for general time series analysis. It s… | 66 | 12785 | maintenance |
| ikawrakow/ik_llama.cpp A fork of llama.cpp offering state-of-the-art quantization types and improved CPU/CUDA inference performance for large language models. It … | 59 | 3091 | active |
| ridgerchu/matmulfreellm A Python implementation of MatMul-Free LM, a language model architecture that eliminates matrix multiplication operations using ternary wei… | 49 | 3089 | active |
| naver/mast3r MASt3R is the official PyTorch implementation of 'Grounding Image Matching in 3D with MASt3R' (ECCV 2024), a model that performs dense 3D r… | 37 | 3088 | active |
| guillaume-be/rust-bert A Rust-native library providing ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT-2, RoBERTa, BART, etc.), por… | 60 | 3076 | active |
| imbue-bit/AlphaGPT AlphaGPT is an open-source automated factor factory based on deep reinforcement learning for quantitative finance. It mines and generates a… | 55 | 3073 | active |
| tensorflow/tflite-micro TensorFlow Lite for Microcontrollers (TFLM) is a C++ port of TensorFlow Lite for running ML models on microcontrollers, DSPs, and other mem… | 77 | 3059 | active |
| hegelai/prompttools PromptTools is an open-source Python library for testing, experimenting with, and evaluating prompts, LLMs (OpenAI, Anthropic, LLaMA, etc.)… | 50 | 3051 | active |
| TradeMaster-NTU/TradeMaster TradeMaster is an open-source Python platform for quantitative trading powered by reinforcement learning, covering the full pipeline of alg… | 34 | 3050 | active |
| sonos/tract Tract is Sonos' tiny, self-contained neural-network inference engine written in Rust. It loads ONNX, TensorFlow/TFLite, and NNEF models, op… | 99 | 3045 | active |