domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| Spark NLP Spark NLP is an open-source natural language processing library built natively on Apache Spark, providing scalable NLP annotations and tran… | 96 | 4159 | stable |
| torchgeo/torchgeo TorchGeo is a PyTorch domain library, similar to torchvision, providing datasets, samplers, transforms, and pre-trained models specific to … | 95 | 4159 | active |
| ArcInstitute/evo2 Evo 2 is a DNA foundation language model (1B-40B parameters) that models genomes at single-nucleotide resolution with up to 1 million base … | 62 | 4157 | active |
| OpenManus/OpenManus-RL OpenManus-RL is an open-source project for reinforcement-learning-based tuning of LLM agents, built on the verl training framework. It prov… | 46 | 4155 | active |
| FareedKhan-dev/all-agentic-architectures A Python library packaging 35 production-grade agentic AI architectures (Reflexion, LATS, GraphRAG, MemGPT, Voyager, etc.) as runnable Arch… | 63 | 4154 | active |
| justadudewhohacks/face-api.js A JavaScript face detection and face recognition library built on top of tensorflow.js, usable in the browser and Node.js. It provides mode… | 23 | 17945 | maintenance |
| openai/transformer-debugger Transformer Debugger (TDB) is an OpenAI tool for investigating specific behaviors of small language models, combining automated interpretab… | 60 | 4124 | active |
| IDEA-CCNL/Fengshenbang-LM Fengshenbang-LM is an open-source suite of Chinese large language models and a PyTorch training framework from IDEA Research's CCNL lab, ai… | 71 | 4123 | active |
| opengeos/segment-geospatial SamGeo (segment-geospatial) is a Python package that applies Meta AI's Segment Anything Model (SAM, SAM2, SAM3, HQ-SAM) to geospatial data … | 97 | 4122 | active |
| huggingface/distil-whisper Distil-Whisper is a distilled version of OpenAI's Whisper model for English speech recognition, offering 6x faster inference, 49% fewer par… | 27 | 4112 | active |
| facebookresearch/jepa Official PyTorch implementation of V-JEPA, a self-supervised method for learning visual representations from video using a joint-embedding … | 29 | 4105 | active |
| ZhengPeng7/BiRefNet BiRefNet is a PyTorch implementation of the CAAI AIR 2024 paper 'Bilateral Reference for High-Resolution Dichotomous Image Segmentation'. I… | 65 | 4098 | active |
| CSGHub CSGHub is an open-source, self-hostable platform for managing the full lifecycle of LLM assets including models, datasets, spaces, and agen… | 91 | 4097 | active |
| GuyTevet/motion-diffusion-model Official PyTorch implementation of the Human Motion Diffusion Model (MDM) paper, generating 3D human motion sequences from text prompts usi… | 52 | 4092 | active |
| princeton-vl/RAFT Official PyTorch implementation of RAFT (Recurrent All Pairs Field Transforms for Optical Flow), an ECCV 2020 model for estimating dense op… | 49 | 4091 | stable |
| danveloper/flash-moe An application for running large language models locally on small laptops, written in Objective-C. It enables on-device inference of big mo… | 48 | 4086 | active |
| PaddlePaddle/PaddleRec PaddleRec is a large-scale recommendation algorithm library built on PaddlePaddle, containing classic and state-of-the-art recommendation m… | 29 | 4085 | active |
| hao-ai-lab/FastVideo FastVideo is a unified Python framework for post-training and real-time inference of video diffusion models, covering data preprocessing, f… | 80 | 4076 | active |
| lllyasviel/Paints-UNDO Paints-UNDO is a family of deep learning models that take an image as input and generate the step-by-step drawing sequence (sketching, inki… | 40 | 4067 | active |
| facebookresearch/dlrm A PyTorch-based implementation of the Deep Learning Recommendation Model (DLRM) from Facebook Research, which processes dense and sparse fe… | 60 | 4065 | stable |
| StarsfieldAI/R1-V R1-V is an open-source research codebase for training vision-language models with reinforcement learning (RLVR/GRPO), demonstrating strong … | 21 | 4063 | active |
| FedML-AI/FedML FedML (TensorOpera) is a unified Python library for large-scale distributed training, model serving, and federated learning across GPU clou… | 45 | 4062 | active |
| marcelroed/gigatoken Gigatoken is a Rust-based tokenizer library for language modeling that tokenizes text at GB/s throughput, claiming ~1000x speedups over Hug… | 60 | 4061 | active |
| thinking-machines-lab/tinker-cookbook Tinker Cookbook is a Python library of realistic examples and abstractions for post-training (fine-tuning) language models via the Tinker A… | 85 | 4058 | active |
| open-spaced-repetition/fsrs4anki FSRS4Anki is a modern spaced-repetition scheduler for Anki based on the Free Spaced Repetition Scheduler (FSRS) algorithm. It replaces Anki… | 80 | 4053 | active |
| google-deepmind/acme Acme is a Python library of reinforcement learning building blocks and agent implementations from DeepMind. It provides readable reference … | 67 | 4050 | active |
| umlx5h/LLPlayer LLPlayer is a Windows media player built for language learning, featuring dual subtitles, AI-generated subtitles via Whisper ASR, real-time… | 78 | 4049 | active |
| deepchecks/deepchecks Deepchecks is an open-source Python library for continuous validation of machine learning models and data, covering testing, CI integration… | 61 | 4047 | active |
| limix-ldm-ai/LimiX LimiX is the first large structured-data foundation model (LDM), a transformer-based model for tabular data that handles classification, re… | 60 | 4045 | active |
| uxlfoundation/oneDNN oneDNN is an open-source cross-platform performance library providing optimized building blocks (primitives) for deep learning applications… | 99 | 4042 | stable |
| tensorflow/tensor2tensor Tensor2Tensor (T2T) is a Python library of deep learning models and datasets built on TensorFlow, developed by the Google Brain team to mak… | 10 | 17464 | maintenance |
| zml/zml ZML is a production LLM inference stack written in Zig, built on MLIR and OpenXLA, that compiles models to run at peak performance across N… | 72 | 4003 | active |
| lucidrains/vector-quantize-pytorch A PyTorch library implementing vector and scalar quantization, including Residual VQ and techniques like DiVeQ codebook updates. It origina… | 83 | 3996 | active |
| OSU-NLP-Group/HippoRAG HippoRAG is a Python RAG framework inspired by human long-term memory that combines LLMs, knowledge graphs, and Personalized PageRank to in… | 59 | 3967 | active |
| city96/ComfyUI-GGUF A ComfyUI custom node pack that adds GGUF quantization support for loading diffusion models like FLUX and Stable Diffusion 3.5 in low-bitra… | 51 | 3953 | active |
| AnswerDotAI/RAGatouille RAGatouille is a Python library that makes it easy to use and train state-of-the-art late-interaction retrieval models like ColBERT in any … | 26 | 3953 | active |
| Nunchaku Nunchaku is a high-performance inference engine for 4-bit quantized diffusion models (and LLMs) based on the SVDQuant technique from an ICL… | 65 | 3937 | active |
| thuml/Transfer-Learning-Library TLlib is a PyTorch-based open-source library for transfer learning, covering domain adaptation, task adaptation (finetuning), and domain ge… | 23 | 3931 | active |
| Lightning-AI/LitServe LitServe is a Python framework for building custom AI inference servers with full control over batching, routing, streaming, and scaling lo… | 92 | 3930 | active |
| stanford-futuredata/ColBERT ColBERT is a fast and accurate neural retrieval model that encodes passages and queries into token-level embedding matrices and scores them… | 45 | 3924 | active |
| cornellius-gp/gpytorch GPyTorch is a Gaussian process library built on PyTorch, designed for scalable, flexible, and modular GP models. It uses linear algebra tec… | 82 | 3906 | stable |
| iree-org/iree IREE is an MLIR-based end-to-end machine learning compiler and runtime that lowers models from frameworks like PyTorch, TensorFlow, JAX, an… | 87 | 3901 | active |
| dora-rs/dora Dora is a Rust-based dataflow middleware framework for building real-time robotics and AI applications, where applications are modeled as d… | 87 | 3900 | active |
| hustvl/Vim Vision Mamba (Vim) is a PyTorch implementation of a generic vision backbone built on bidirectional Mamba state space models, published at I… | 29 | 3899 | active |
| openai/baselines OpenAI Baselines is a collection of high-quality reference implementations of reinforcement learning algorithms such as DQN, PPO, and ACKTR… | 32 | 16764 | maintenance |
| Plachtaa/seed-vc Seed-VC is a Python tool and model for zero-shot voice conversion, real-time voice conversion, and singing voice conversion, cloning a voic… | 10 | 3888 | active |
| huggingface/smollm Hugging Face's repository for the SmolLM and SmolVLM families of compact, fully open language and vision-language models, including trainin… | 59 | 3884 | active |
| safetensors/safetensors Safetensors is a simple, secure file format and library for storing and distributing tensors, designed as a fast zero-copy alternative to p… | 91 | 3876 | stable |
| LazyAGI/LazyLLM LazyLLM is a low-code Python framework for building multi-agent LLM applications, covering prototype assembly, data feedback, and iterative… | 91 | 3875 | active |
| hkust-nlp/simpleRL-reason A research codebase from HKUST-NLP implementing a simple reinforcement learning recipe (rule-based rewards on GSM8K/Math data) to train LLM… | 47 | 3874 | active |
| NVIDIA/k8s-device-plugin NVIDIA's official Kubernetes device plugin that exposes NVIDIA GPUs as schedulable resources in Kubernetes clusters. It supports GPU sharin… | 95 | 3859 | stable |
| mseitzer/pytorch-fid A PyTorch port of the official TensorFlow implementation of the Fréchet Inception Distance (FID), a metric for measuring similarity between… | 23 | 3851 | stable |
| MrForExample/ComfyUI-3D-Pack An extensive ComfyUI custom node suite for processing 3D inputs like meshes and UV textures using algorithms such as 3D Gaussian Splatting … | 51 | 3850 | active |
| open-mmlab/mmpretrain MMPretrain is OpenMMLab's PyTorch-based toolbox and benchmark for image classification model pre-training, covering supervised, self-superv… | 23 | 3850 | active |
| Stability-AI/stable-audio-tools Stability AI's training and inference toolkit for conditional audio generation models, including Stable Audio Open. It supports training cu… | 73 | 3849 | active |
| huggingface/huggingface_hub The official Python client and CLI for the Hugging Face Hub, a platform for sharing machine learning models, datasets, and apps. It lets us… | 95 | 3845 | stable |
| shenweichen/GraphEmbedding A Python library providing implementations of classic graph embedding algorithms including DeepWalk, LINE, Node2Vec, SDNE, and Struc2Vec. I… | 68 | 3845 | active |
| piskvorky/gensim Gensim is a Python library for topic modelling, document indexing, and similarity retrieval over large text corpora. It provides memory-ind… | 50 | 16481 | maintenance |
| neuraloperator/neuraloperator A PyTorch library for learning neural operators, which map between function spaces rather than finite-dimensional vectors. It provides the … | 82 | 3836 | active |
| Netflix/maestro Netflix Maestro is a general-purpose workflow orchestrator providing workflow-as-a-service for data, ML, and software pipelines. It schedul… | 69 | 3829 | active |
| predibase/lorax LoRAX is a multi-LoRA inference server that serves thousands of fine-tuned LLM adapters on a single GPU by sharing a common base model. It … | 62 | 3826 | active |
| TransformerLensOrg/TransformerLens TransformerLens is a Python library for mechanistic interpretability of GPT-style transformer language models. It lets researchers load tho… | 99 | 3825 | active |
| google-research/scenic Scenic is a JAX-based library from Google Research focused on attention-based models for computer vision, providing shared lightweight libr… | 76 | 3821 | active |
| benfred/implicit A fast Python library implementing collaborative filtering recommendation algorithms for implicit feedback datasets, including ALS, BPR, lo… | 74 | 3815 | active |
| ddbourgin/numpy-ml numpy-ml is a collection of machine learning models and algorithms implemented exclusively in NumPy and the Python standard library, coveri… | 32 | 16330 | maintenance |
| lightly-ai/lightly LightlySSL is a Python library built on PyTorch for self-supervised learning on images, offering modular implementations of methods like Si… | 93 | 3797 | active |
| google/deepvariant DeepVariant is a deep learning-based genomic variant caller that converts aligned DNA sequencing reads (BAM/CRAM) into pileup image tensors… | 70 | 3791 | stable |
| SciSharp/LLamaSharp LLamaSharp is a cross-platform C#/.NET library that wraps llama.cpp to run LLaMA-family and other LLMs (including multimodal LLaVA) efficie… | 92 | 3787 | active |
| aubio/aubio aubio is a C library (with Python bindings and CLI tools) for labeling music and sounds, detecting events such as note onsets, pitch, and t… | 57 | 3754 | stable |
| astroautomata/PySR PySR is a high-performance symbolic regression library that searches for symbolic mathematical expressions optimizing a given objective, im… | 99 | 3739 | stable |
| vllm-project/llm-compressor LLM Compressor is a Python library for applying quantization and pruning algorithms to large language models, producing compressed-tensors … | 90 | 3726 | active |
| polyaxon/polyaxon Polyaxon is an open-source AI engineering control plane that teams self-host on their own Kubernetes clusters to schedule, track, and gover… | 77 | 3723 | active |
| roboflow/trackers A Python library of clean-room, Apache 2.0 implementations of multi-object tracking algorithms including SORT, ByteTrack, OC-SORT, BoT-SORT… | 85 | 3717 | active |
| genmoai/mochi Mochi 1 is Genmo's open-source, state-of-the-art text-to-video generation model released under Apache 2.0, with a Python API, CLI, and Grad… | 46 | 3713 | active |
| MTG/essentia Essentia is an open-source C++ library for audio analysis and music information retrieval, offering a large collection of reusable DSP algo… | 65 | 3705 | stable |
| danijar/dreamerv3 A Python/JAX reimplementation of DreamerV3, a model-based reinforcement learning algorithm that learns a world model and trains an actor-cr… | 69 | 3704 | active |
| Belval/TextRecognitionDataGenerator A Python library and CLI tool (trdg) that generates synthetic text images for training OCR and text recognition models. It supports multipl… | 23 | 3691 | stable |
| DLR-RM/BlenderProc BlenderProc is a procedural Python pipeline built on Blender for generating photorealistic synthetic training images with ground-truth anno… | 62 | 3684 | active |
| thu-ml/SageAttention SageAttention is a family of quantized attention kernels (INT8/FP8/FP4) that accelerate transformer inference 2-5x over FlashAttention with… | 40 | 3684 | active |
| facebookresearch/map-anything MapAnything is an open-source research framework from Meta and CMU for universal feed-forward metric 3D reconstruction using an end-to-end … | 77 | 3682 | active |
| ferdous-alam/GenCAD GenCAD is a research codebase for image-conditioned CAD model generation using transformer-based contrastive representations (CCIP) and dif… | 36 | 3669 | active |
| PAIR-code/lit The Learning Interpretability Tool (LIT) is a visual, interactive web-based tool for understanding and debugging ML model behavior across t… | 73 | 3660 | active |
| HazyResearch/ThunderKittens ThunderKittens is a C++/CUDA framework of tile-based primitives for writing fast deep learning GPU kernels. It embeds natively into CUDA so… | 70 | 3659 | active |
| kaldi-asr/kaldi Kaldi is a C++ toolkit for speech recognition research and development, including acoustic modeling, feature extraction, decoding, and spea… | 52 | 15469 | maintenance |
| opendilab/DI-engine DI-engine is an open-source reinforcement learning framework from OpenDILab that provides comprehensive implementations of deep RL algorith… | 48 | 3638 | active |
| ploomber/ploomber Ploomber is a Python framework for building maintainable data pipelines from scripts and Jupyter notebooks, with iterative local developmen… | 10 | 3622 | active |
| mit-han-lab/llm-awq AWQ is a Python library implementing Activation-aware Weight Quantization for compressing large language models to low-bit (INT3/4) weights… | 44 | 3621 | active |
| facebookresearch/detr DETR is Facebook Research's PyTorch implementation of Detection Transformer, an end-to-end object detection model that replaces hand-crafte… | 10 | 15354 | maintenance |
| facebookresearch/sam-audio SAM-Audio is Meta's foundation model for isolating any sound in audio using text, visual, or temporal prompts. This repository provides inf… | 55 | 3612 | active |
| albumentations-team/albumentations Albumentations is a fast, flexible Python image augmentation library for computer vision, supporting images, masks, bounding boxes, keypoin… | 10 | 15315 | maintenance |
| MrNeRF/LichtFeld-Studio LichtFeld Studio is a native open-source desktop application for 3D Gaussian Splatting that combines training, real-time inspection, splat … | 92 | 3594 | active |
| AI4Finance-Foundation/FinRL-Trading FinRL-X is an open-source, AI-native modular infrastructure for quantitative trading that unifies data processing, strategy composition, ba… | 71 | 3592 | active |
| meta-pytorch/botorch BoTorch is a Python library for Bayesian Optimization research built on top of PyTorch, part of the PyTorch ecosystem. It provides modular,… | 91 | 3589 | active |
| urchade/GLiNER GLiNER is a Python framework for training and deploying lightweight, zero-shot Named Entity Recognition models that can extract arbitrary e… | 87 | 3574 | active |
| rdkit/rdkit RDKit is an open-source cheminformatics toolkit written in C++ with Python, Java, C#, and JavaScript wrappers. It provides core data struct… | 98 | 3565 | stable |
| RUC-NLPIR/FlashRAG FlashRAG is a modular Python toolkit for reproducing and developing Retrieval-Augmented Generation (RAG) research, bundling 36 pre-processe… | 68 | 3557 | active |
| sdv-dev/SDV SDV (Synthetic Data Vault) is a Python library for generating synthetic tabular data using machine learning models ranging from GaussianCop… | 99 | 3549 | stable |
| microsoft/hummingbird Hummingbird is a Microsoft library that compiles trained traditional ML models (decision trees, random forests, XGBoost, LightGBM) into ten… | 37 | 3544 | active |
| jmschrei/pomegranate Pomegranate is a Python library for fast and flexible probabilistic modeling, built on PyTorch. It supports distributions, mixture models, … | 27 | 3542 | active |
| pytorch/rl TorchRL is a PyTorch-native library of composable primitives for building reinforcement learning systems, including environments, policies,… | 97 | 3538 | active |