Open-source machine-learning software

5374 open-source libraries, tools and applications tagged machine-learning, ranked by how closely machine-learning is their primary purpose, then by adoption and maintenance health. Each entry links to a page with sources, use cases and when to choose or avoid it.

5374 products, primary matches first, then adoption-weighted; health v2 shown.

Product	Health v2	Stars	Maturity
zai-org/GLM-4.5 GLM-4.5 (and successors GLM-4.6/4.7) is a family of open-weight Mixture-of-Experts foundation models from Z.ai focused on agentic tasks, re…	44	4424	active
huawei-noah/Efficient-AI-Backbones A collection of efficient neural network backbone architectures (GhostNet, TNT, ViG, WaveMLP, TinyNet, etc.) from Huawei Noah's Ark Lab, wi…	24	4423	active
onnxsim/onnxsim ONNX Simplifier (onnxsim) is a tool that simplifies ONNX models by inferring the whole computation graph and replacing redundant operators …	100	4415	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…	61	4412	active
MedSAM MedSAM is a fine-tuned Segment Anything Model (SAM) foundation model for universal medical image segmentation, trained on over 1.5 million …	26	4411	active
microsoft/FLAML FLAML is a lightweight Python library for automated machine learning (AutoML) and hyperparameter tuning. It efficiently finds quality model…	94	4409	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…	23	4405	active
meta-llama/PurpleLlama Purple Llama is Meta's umbrella project of tools and benchmarks for assessing and improving the security of large language models. It inclu…	68	4401	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	19274	maintenance
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…	61	4387	active
AI4Finance-Foundation/ElegantRL ElegantRL is a lightweight, modular deep reinforcement learning library built on PyTorch that implements core model-free RL algorithms (PPO…	50	4368	active
dreamgaussian/dreamgaussian DreamGaussian is the official PyTorch implementation of an ICLR 2024 Oral paper for efficient 3D content creation using generative Gaussian…	19	4360	active
reBot Arm An open-source robotic arm project (hardware and software) from Seeed, with CERN-OHL-W-2.0 licensed hardware designs and Apache-2.0 softwar…	61	4346	active
VectorSpaceLab/OmniGen OmniGen is a unified diffusion-based image generation model that produces and edits images from multi-modal prompts without auxiliary modul…	44	4346	active
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	4330	active
tloen/alpaca-lora A repository of scripts for reproducing Stanford Alpaca-style instruction tuning of LLaMA models using low-rank adaptation (LoRA) on consum…	31	18899	maintenance
zvtvz/zvt ZVT is a unified, modular quantitative trading framework written in Python for stocks and cryptocurrencies. It provides data ingestion, fac…	73	4311	active
NeuralProphet NeuralProphet is a Python library for interpretable time series forecasting built on PyTorch, combining neural networks with traditional ti…	23	4301	active
Tencent-Hunyuan/HunyuanDiT Hunyuan-DiT is Tencent's open-source diffusion transformer model for text-to-image generation with fine-grained Chinese language understand…	46	4288	active
isaac-sim/IsaacSim NVIDIA Isaac Sim is an open-source robotics simulation application built on NVIDIA Omniverse for developing, simulating, and testing AI-dri…	83	4283	active
richzhang/PerceptualSimilarity A PyTorch library implementing the LPIPS (Learned Perceptual Image Patch Similarity) metric, which measures perceptual distance between ima…	23	4280	stable
Nixtla/neuralforecast NeuralForecast is a Python library offering a large collection of state-of-the-art neural forecasting models (NBEATS, NHITS, TFT, PatchTST,…	97	4277	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…	45	4262	active
jwohlwend/boltz Boltz is a family of open-source biomolecular interaction models (Boltz-1 and Boltz-2) that predict complex structures and binding affiniti…	58	4257	active
ThilinaRajapakse/simpletransformers Simple Transformers is a Python library built on Hugging Face Transformers that lets users train, fine-tune, and evaluate Transformer model…	57	4256	active
kubeflow/pipelines Kubeflow Pipelines is a Kubernetes-native platform for building, deploying, and orchestrating reusable end-to-end machine learning workflow…	98	4248	stable
ZhengPeng7/BiRefNet BiRefNet is a PyTorch implementation of the CAAI AIR 2024 paper 'Bilateral Reference for High-Resolution Dichotomous Image Segmentation'. I…	64	4247	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 …	56	4247	active
ali-vilab/AnyDoor AnyDoor is the official implementation of a diffusion-based model that teleports target objects into new scenes at user-specified locations…	29	4241	active
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…	44	4238	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…	44	4232	active
RobotLocomotion/drake Drake is a C++ toolbox (with Python bindings) for model-based design and verification of robotics systems, developed by the Robot Locomotio…	100	4230	stable
MaartenGr/KeyBERT KeyBERT is a minimal Python library that extracts keywords and keyphrases from documents using BERT embeddings and cosine similarity. It re…	64	4223	stable
Fannovel16/comfyui_controlnet_aux A collection of ComfyUI custom nodes providing ControlNet auxiliary preprocessors that generate hint images (Canny edges, lineart, depth ma…	73	4221	active
facebookresearch/nevergrad Nevergrad is a Python toolbox for gradient-free (derivative-free) optimization, providing many optimization algorithms such as NGOpt and di…	79	4215	active
Open Duck Mini Open Duck Mini is an open-source project to build a miniature (~42cm) bipedal robot inspired by Disney's BDX droid, with a bill of material…	51	4208	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…	84	4199	active
torchgeo/torchgeo TorchGeo is a PyTorch domain library, similar to torchvision, providing datasets, samplers, transforms, and pre-trained models specific to …	93	4192	active
JEPA Official PyTorch implementation of V-JEPA, a self-supervised method for learning visual representations from video using a joint-embedding …	26	4179	active
lllyasviel/style2paints Style2Paints is an AI-driven tool that colorizes lineart sketches, optionally guided by human hints, style reference images, and lighting. …	32	18158	maintenance
stemdeckapp/stemdeck StemDeck is a free, local stem separation application that splits songs into up to six stems (vocals, drums, bass, guitar, piano, other) us…	78	4166	active
Spark NLP Spark NLP is an open-source natural language processing library built natively on Apache Spark, providing scalable NLP annotations and tran…	98	4160	stable
shitagaki-lab/see-through A research framework from a SIGGRAPH 2026 paper that decomposes a single anime character illustration into up to 23 fully inpainted, semant…	58	4153	active
SamGeo SamGeo (segment-geospatial) is a Python package that applies Meta AI's Segment Anything Model (SAM, SAM2, SAM3, HQ-SAM) to geospatial data …	94	4147	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…	48	4138	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	17981	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…	69	4136	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…	67	4123	active
openai/transformer-debugger Transformer Debugger (TDB) is an OpenAI tool for investigating specific behaviors of small language models, combining automated interpretab…	57	4123	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…	28	4120	active
ericciarla/trendFinder A self-hosted Node.js application that monitors influencer posts on Twitter/X and website changes via Firecrawl, then uses LLMs (Together A…	22	4120	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…	48	4119	active
VectorSpaceLab/OmniGen2 OmniGen2 is an open-source unified multimodal generation model supporting text-to-image generation, instruction-guided image editing, and i…	48	4113	active
lllyasviel/sd-forge-layerdiffuse A Stable Diffusion WebUI (Forge) extension implementing Layer Diffusion to generate transparent images and separate foreground/background l…	26	4111	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…	46	4103	stable
Tencent-Hunyuan/Hunyuan3D-2.1 Tencent's open-source 3D asset generation model that creates high-fidelity 3D meshes with production-ready PBR materials from single images…	37	4096	active
PaddlePaddle/PaddleRec PaddleRec is a large-scale recommendation algorithm library built on PaddlePaddle, containing classic and state-of-the-art recommendation m…	26	4086	active
google-deepmind/acme Acme is a Python library of reinforcement learning building blocks and agent implementations from DeepMind. It provides readable reference …	67	4080	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…	74	4074	active
DLRM A PyTorch-based implementation of the Deep Learning Recommendation Model (DLRM) from Facebook Research, which processes dense and sparse fe…	57	4074	stable
FedML FedML (TensorOpera) is a unified Python library for large-scale distributed training, model serving, and federated learning across GPU clou…	41	4070	active
uxlfoundation/oneDNN oneDNN is an open-source cross-platform performance library providing optimized building blocks (primitives) for deep learning applications…	99	4067	stable
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…	37	4063	active
StarsfieldAI/R1-V R1-V is an open-source research codebase for training vision-language models with reinforcement learning (RLVR/GRPO), demonstrating strong …	18	4063	active
deepchecks/deepchecks Deepchecks is an open-source Python library for continuous validation of machine learning models and data, covering testing, CI integration…	57	4060	active
ginobefun/BestBlogs BestBlogs.dev is an AI-powered personal reading assistant and content curation platform that aggregates articles, podcasts, videos, tweets,…	64	4053	active
KieronQuinn/TapTap Tap, Tap is an Android app that ports the Pixel's double-tap-on-back gesture to any Android 7.0+ device with an accelerometer and gyroscope…	23	4053	active
defog-ai/sqlcoder SQLCoder is a family of fine-tuned open-weight LLMs that convert natural language questions into SQL queries, outperforming GPT-4 on text-t…	29	4048	active
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	17473	maintenance
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…	57	4021	active
Nixtla/nixtla Nixtla's TimeGPT is a production-ready pre-trained foundation model for time series forecasting and anomaly detection, accessed via a Pytho…	98	4019	active
lucidrains/vector-quantize-pytorch A PyTorch library implementing vector and scalar quantization, including Residual VQ and techniques like DiVeQ codebook updates. It origina…	81	4019	active
microsoft/PromptWizard PromptWizard is a discrete prompt optimization framework from Microsoft that uses a self-evolving mechanism where an LLM generates, critiqu…	42	4010	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	3997	active
Marin Marin is an open-source Python framework and research program for training foundation models, covering the full pipeline from data curation…	79	3988	active
LLaDA Official PyTorch implementation of LLaDA, a family of large language diffusion models (8B base/instruct, MoE, and iLLaDA variants) with pre…	59	3986	active
IREE IREE is an MLIR-based end-to-end machine learning compiler and runtime that lowers models from frameworks like PyTorch, TensorFlow, JAX, an…	95	3980	active
hustvl/4DGaussians An official PyTorch implementation of 4D Gaussian Splatting (4D-GS) for real-time rendering of dynamic scenes, published at CVPR 2024. It c…	28	3978	active
dpilger26/NumCpp NumCpp is a templatized, header-only C++ library that reimplements the Python NumPy API, centered on an NdArray container with NumPy-style …	93	3966	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 …	24	3964	active
ali-vilab/VACE VACE is the official implementation of an all-in-one video creation and editing model from Tongyi Lab, built on Wan2.1 diffusion models. It…	38	3962	active
aurelio-labs/semantic-router Semantic Router is a Python library that provides a fast decision-making layer for LLMs and agents, routing requests based on semantic vect…	95	3959	active
yuanzhongqiao/printfilm Printfilm is a self-hostable AI short-drama (short film / motion comic) creation SaaS platform built on Next.js and Spring Boot. It provide…	67	3958	active
thu-ml/SageAttention SageAttention is a family of quantized attention kernels (INT8/FP8/FP4) that accelerate transformer inference 2-5x over FlashAttention with…	38	3956	active
NeuralOperator A PyTorch library for learning neural operators, which map between function spaces rather than finite-dimensional vectors. It provides the …	81	3955	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…	73	3953	active
TransformerLensOrg/TransformerLens TransformerLens is a Python library for mechanistic interpretability of GPT-style transformer language models. It lets researchers load tho…	98	3949	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…	40	3941	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	3940	active
huggingface/smollm Hugging Face's repository for the SmolLM and SmolVLM families of compact, fully open language and vision-language models, including trainin…	67	3926	active
GPyTorch GPyTorch is a Gaussian process library built on PyTorch, designed for scalable, flexible, and modular GP models. It uses linear algebra tec…	82	3925	stable
hustvl/Vim Vision Mamba (Vim) is a PyTorch implementation of a generic vision backbone built on bidirectional Mamba state space models, published at I…	27	3913	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…	87	3901	stable
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
PySR PySR is a high-performance symbolic regression library that searches for symbolic mathematical expressions optimizing a given objective, im…	99	3893	stable
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	3892	active
vllm-project/llm-compressor LLM Compressor is a Python library for applying quantization and pruning algorithms to large language models, producing compressed-tensors …	92	3886	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…	44	3877	active
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 …	47	3871	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	3868	active