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
xbpeng/MimicKit MimicKit is a lightweight Python framework providing a suite of motion imitation methods (DeepMimic, AMP, ASE, AWR, and others) for trainin…	50	2360	active
NVIDIA-NeMo/DataDesigner NeMo Data Designer is an open-source Python orchestration framework from NVIDIA for generating high-quality synthetic datasets from scratch…	85	2357	active
tencent-ailab/V-Express V-Express is a Python research project from Tencent AI Lab that generates talking head portrait videos from a reference image, audio, and V…	25	2357	active
GalTransl/GalTransl GalTransl is an automated translation tool for visual novels (galgames) that leverages large language models like GPT-4, Claude, Deepseek, …	97	2353	active
Neural Circuit Policies (ncps) A Python package providing PyTorch and TensorFlow/Keras implementations of Neural Circuit Policies (NCPs), including liquid time-constant (…	23	2351	active
google-deepmind/optax Optax is a gradient processing and optimization library for JAX, offering composable building blocks like optimizers and loss functions. It…	80	2350	stable
maitrix-org/llm-reasoners LLM Reasoners is a Python library that provides a standardized modular framework for advanced LLM reasoning, implementing algorithms like M…	28	2345	active
ASLP-lab/DiffRhythm DiffRhythm is an open-source latent diffusion model for end-to-end full-length song generation, supporting lyrics-to-song, text style promp…	42	2344	active
SkyworkAI/Matrix-Game Matrix-Game is Skywork AI's open-source series of interactive world foundation models that generate real-time, streaming video in response …	50	2342	active
Vortex GPGPU Vortex is a full-stack open-source RISC-V GPGPU supporting OpenCL, with configurable cores, warps, and threads, plus graphics and tensor co…	89	2337	active
ALOHA ALOHA is an open-source hardware and software system for low-cost bimanual robot teleoperation and demonstration data collection, built on …	31	2336	active
refuel-ai/autolabel Autolabel is a Python library for labeling, cleaning, and enriching text datasets using large language models like GPT-4, Claude, and open-…	22	2330	active
facebookresearch/schedule_free A PyTorch library implementing schedule-free optimizers (SGD, AdamW, RAdam variants) that remove the need for learning rate schedules or sp…	64	2329	active
LabML A curated collection of over 60 PyTorch implementations of deep learning papers, complete with side‑by‑side explanatory notes. It serves as…	26	2327	active
LabML A Python library for tracking and monitoring deep learning experiments, with a self-hostable server app for viewing metrics and hardware us…	26	2327	active
ufoym/imbalanced-dataset-sampler A PyTorch dataset sampler (ImbalancedDatasetSampler, pip package torchsampler) that rebalances class distributions during training by overs…	60	2326	stable
YvanYin/Metric3D Metric3D is the official PyTorch implementation of Metric3Dv1 and Metric3Dv2, monocular geometric foundation models that predict metric dep…	31	2326	active
Picotron Picotron is a minimalist, hackable distributed training framework for pre-training Llama-like large language models using 4D parallelism (d…	37	2325	active
PKU-YuanGroup/MoE-LLaVA MoE-LLaVA is an open-source Mixture-of-Experts based sparse large vision-language model, released with the MoE-Tuning training strategy fro…	28	2324	active
UZ-SLAMLab/ORB_SLAM3 ORB-SLAM3 is a real-time SLAM library supporting Visual, Visual-Inertial, and Multi-Map SLAM with monocular, stereo, and RGB-D cameras usin…	23	9125	maintenance
iver56/audiomentations Audiomentations is a Python library for audio data augmentation with an API inspired by albumentations. It provides fast CPU-based waveform…	61	2323	stable
Cadene/pretrained-models.pytorch A Python library providing pretrained ConvNet models (ResNet, ResNeXt, InceptionV4, Xception, NASNet, SENet, DPN, etc.) for PyTorch behind …	32	9101	maintenance
nv-tlabs/lyra Project Lyra is NVIDIA's open series of generative 3D world models, including Lyra 1.0 for feed-forward 3D/4D scene generation from a singl…	56	2317	active
facebookresearch/ImageBind A PyTorch library from Meta AI implementing ImageBind, a model that learns a joint embedding space across six modalities: images, text, aud…	51	9076	maintenance
OpenHelix-Team/VLA-Adapter VLA-Adapter is the official implementation of a tiny-scale vision-language-action model that bridges vision-language representations to rob…	47	2313	active
chrisworsey55/atlas-gic ATLAS is a Python framework of self-improving AI trading agents that optimize their own prompts against market outcomes, using multi-agent …	59	2312	active
LAION-AI/CLAP CLAP (Contrastive Language-Audio Pretraining) is a Python library from LAION that learns joint audio and text embeddings using contrastive …	38	2312	stable
IDEA-Research/detrex detrex is an open-source PyTorch-based research platform and toolbox for DETR-style Transformer detection algorithms, built on top of Detec…	38	2311	active
apache/mahout Apache Mahout is an Apache Software Foundation project providing Qumat, a high-level Python library for quantum computing with a unified AP…	84	2310	active
frgfm/torch-cam TorchCAM is a Python library that extracts class activation maps (CAMs) from PyTorch CNN classifiers, supporting many CAM variants such as …	88	2307	active
levy-street/world-of-claudecraft World of ClaudeCraft is a free, open-source classic-style browser MMO built in TypeScript with Three.js, playable online with persistent ch…	77	2302	active
emgucv/emgucv Emgu CV is a cross-platform .NET wrapper for the OpenCV image processing library, allowing OpenCV functions to be called from .NET-compatib…	90	2300	active
traveller59/spconv SpConv is a spatially sparse convolution library for deep learning on 3D point clouds and sparse tensors, distributed as PyPI packages with…	32	2299	active
NVlabs/nvdiffrec nvdiffrec is NVIDIA's official implementation of a CVPR 2022 oral paper that jointly optimizes triangular 3D meshes, PBR materials, and lig…	72	2298	stable
THU-MIG/yoloe YOLOE is the official PyTorch implementation of an open-vocabulary object detection and segmentation model presented at ICCV 2025. It unifi…	30	2297	active
codeplea/genann Genann is a minimal, well-tested C99 library for training and running feedforward artificial neural networks. It is contained in a single s…	91	2293	stable
fairlearn/fairlearn Fairlearn is a Python package for assessing and mitigating fairness issues in machine learning models. It provides fairness metrics for eva…	91	2293	active
aigc-apps/VideoX-Fun VideoX-Fun is a Python-based video generation pipeline built on Diffusion Transformer models (CogVideoX-Fun, Wan-Fun) that generates videos…	68	2292	active
feature-engine/feature_engine Feature-engine is a Python library for feature engineering and feature selection with transformers that follow scikit-learn's fit/transform…	82	2290	stable
benedekrozemberczki/karateclub Karate Club is a Python library of unsupervised machine learning algorithms for graph-structured data, built as an extension of NetworkX. I…	23	2290	stable
facebookresearch/fairchem FAIR Chemistry's centralized Python library of machine learning models, datasets, and applications for materials science and quantum chemis…	98	2289	active
hkchengrex/MMAudio MMAudio is a PyTorch-based model for generating synchronized audio from video and/or text inputs, using multimodal joint training across au…	39	2287	active
shuyu-labs/BigBanana-AI-Director BigBanana AI Director is a self-hosted, industrial-grade AI platform for generating short dramas and motion comics end-to-end, from script …	79	2281	active
Liuziyu77/Visual-RFT Official research code for Visual-RFT and Visual-ARFT, applying GRPO-based reinforcement fine-tuning with rule-based verifiable rewards to …	39	2280	active
TEN-framework/ten-vad TEN VAD is a lightweight, low-latency voice activity detection library with an ONNX model for real-time speech detection. It supports C, Py…	47	2277	active
opendatalab/DocLayout-YOLO DocLayout-YOLO is a real-time YOLO-v10-based model for detecting document layout elements (text blocks, tables, figures, etc.) in diverse d…	26	2277	active
stepfun-ai/gelab-zero GELab-Zero is an open-source GUI agent framework for mobile devices, combining a 4B vision-language model (GELab-Zero-4B) with engineering …	50	2276	active
OlafenwaMoses/ImageAI ImageAI is a Python library that lets developers add computer vision capabilities like image classification, object detection, and video ob…	23	8888	maintenance
ermig1979/Simd Simd Library is a free open-source C++ image processing and machine learning library with a C API and Python wrapper. Its algorithms are ha…	98	2272	active
aixcoder-plugin/aiXcoder-7B Official repository for aiXcoder-7B, an open-weights 7B-parameter code large language model trained on 1.2T tokens for code completion, gen…	26	2271	active
RosettaCommons/RoseTTAFold RoseTTAFold is the official implementation of a deep learning system for predicting protein structures and interactions using a three-track…	23	2269	stable
Amazon SageMaker Containers & SDK The SageMaker Python SDK is an open-source Python library for training and deploying machine learning models on Amazon SageMaker. It suppor…	94	2268	active
ashawkey/stable-dreamfusion A PyTorch implementation of Dreamfusion that generates 3D models from text prompts or images using NeRF combined with Stable Diffusion guid…	23	8865	maintenance
stepfun-ai/Step1X-Edit Step1X-Edit is an open-source state-of-the-art instruction-based image editing model from StepFun, designed to rival closed-source editors …	52	2266	active
qhjqhj00/MemoRAG MemoRAG is a Python RAG framework that uses a super-long memory model to build a global understanding of an entire corpus (up to ~1M tokens…	29	2266	active
aigc-apps/EasyAnimate EasyAnimate is an end-to-end Python pipeline for high-resolution, long video and image generation based on transformer diffusion (DiT) mode…	16	2266	active
666DZY666/micronet micronet is a Python library for deep neural network model compression and deployment built on PyTorch. It provides quantization (QAT, PTQ,…	38	2265	active
yemount/pose-animator Pose Animator is a browser-based tool that animates 2D SVG vector characters in real time using pose and face keypoints detected by PoseNet…	32	8851	maintenance
galilai-group/stable-worldmodel A Python library providing a unified platform for reproducible world model research, covering data collection, training, and evaluation via…	78	2260	active
TuringLang/Turing.jl Turing.jl is a probabilistic programming library in Julia for Bayesian inference. Models are defined with an @model macro resembling mathem…	93	2258	active
fishaudio/Bert-VITS2 Bert-VITS2 is a text-to-speech model implementation combining the VITS2 architecture with multilingual BERT embeddings, written in Python. …	64	8805	maintenance
facebookresearch/fvcore fvcore is a lightweight Python core library providing common functionality shared across FAIR's computer vision frameworks such as Detectro…	73	2254	stable
MinishLab/model2vec Model2Vec is a Python library that distills any sentence transformer into a tiny, fast static embedding model by computing one fixed vector…	91	2251	active
CoinCheung/pytorch-loss A PyTorch library providing a collection of loss functions (focal loss, triplet loss, AMSoftmax, label-smooth CE, dice loss, lovasz-softmax…	32	2251	active
tphakala/birdnet-go BirdNET-Go is a self-hosted, 24/7 realtime soundscape analyser that classifies birds, bats, and other wildlife using local AI models like B…	92	2250	active
google-deepmind/mujoco_playground MuJoCo Playground is an open-source Python library of GPU-accelerated robot learning environments built on MuJoCo MJX and MuJoCo Warp. It s…	74	2250	active
Alpha-VLLM/Lumina-T2X Lumina-T2X is a unified framework for text-to-any-modality generation built on flow-based large diffusion transformers. It supports generat…	25	2250	active
kubeflow/trainer Kubeflow Trainer is a Kubernetes-native platform for distributed AI model training and LLM fine-tuning across frameworks like PyTorch, JAX,…	92	2249	active
DepthAnything/Video-Depth-Anything Video Depth Anything is a transformer-based monocular depth estimation model for arbitrarily long videos, built on Depth Anything V2. It pr…	35	2247	active
ByteDance-Seed/VeOmni VeOmni is a PyTorch-native framework for single- and multi-modal model pre-training and post-training, with a modular, trainer-free design …	87	2246	active
Accenture/AmpliGraph AmpliGraph is a Python library built on TensorFlow 2 for representation learning on knowledge graphs. It provides neural models (TransE, Di…	78	2242	active
Raster Vision Raster Vision is an open source Python library and low-code framework for building computer vision models on satellite, aerial, and other l…	57	2241	active
rapidsai/cugraph cuGraph is NVIDIA's RAPIDS collection of GPU-accelerated graph analytics libraries, offering Python, C, and C++ APIs for building graphs an…	90	2237	active
microsoft/LLaVA-Med LLaVA-Med is a large language-and-vision assistant fine-tuned for the biomedicine domain, built on the LLaVA multimodal architecture. It su…	37	2237	active
NVlabs/MambaVision MambaVision is NVIDIA's official PyTorch implementation of a hybrid Mamba-Transformer vision backbone, published at CVPR 2025. It provides …	46	2235	active
ellisdg/3DUnetCNN A PyTorch library for building, training, and applying 3D U-Net convolutional neural networks for medical image segmentation. It provides c…	42	2235	active
openpnp/openpnp OpenPnP is open source software (with accompanying hardware designs) for controlling SMT pick and place machines used in PCB assembly. It c…	77	2233	active
PyGAD PyGAD is an open-source Python 3 library for implementing the genetic algorithm to optimize single- and multi-objective problems. It can al…	90	2230	active
NVIDIA/vid2vid A PyTorch implementation of NVIDIA's video-to-video synthesis method for generating high-resolution (e.g., 2048x1024) photorealistic videos…	32	8692	maintenance
floneum/kalosm Kalosm is a Rust ecosystem of crates providing simple interfaces for running pre-trained language, audio, and image models locally or remot…	65	2227	active
facebookresearch/DiT Official PyTorch implementation of Diffusion Transformers (DiT) from the paper 'Scalable Diffusion Models with Transformers', including mod…	10	8676	maintenance
thuml/iTransformer Official PyTorch implementation of iTransformer, an ICLR 2024 Spotlight paper that inverts the Transformer architecture for multivariate ti…	39	2225	stable
DerwenAI/pytextrank PyTextRank is a Python implementation of TextRank and related textgraph algorithms (PositionRank, Biased TextRank, TopicRank) as a spaCy pi…	59	2222	active
apple/ml-ferret Apple's Ferret, an end-to-end multimodal large language model (MLLM) that accepts any-form referring and grounds anything in its responses,…	70	8655	maintenance
lifeiteng/vall-e An unofficial PyTorch implementation of VALL-E, a zero-shot text-to-speech model that treats TTS as a conditional language modeling task ov…	37	2221	active
Esri/arcgis-python-api The ArcGIS API for Python is a Pythonic library for working with maps, geospatial data, and web GIS, supporting spatial analysis, deep lear…	89	2217	active
google-deepmind/tapnet Google DeepMind's official repository for Tracking Any Point (TAP), containing the TAP-Vid and TAPVid-3D benchmarks, the TAPIR and TAPNext …	75	2217	active
kijai/ComfyUI-LivePortraitKJ ComfyUI custom nodes that integrate the LivePortrait face animation and retargeting model, supporting image-to-video, video-to-video, and n…	24	2211	active
DigitalPhonetics/IMS-Toucan IMS Toucan is a PyTorch-based toolkit for training and running state-of-the-art, controllable text-to-speech synthesis, home of the massive…	48	2209	active
RubixML/ML Rubix ML is a high-level open-source machine learning and deep learning library for PHP, offering 40+ supervised and unsupervised algorithm…	98	2208	active
Core AI Apple's repository of model export recipes, Python primitives, and Swift runtime utilities for building on-device AI with the Core AI frame…	70	2206	active
WebODM/OpenSplat OpenSplat is a free, open-source C++ implementation of 3D gaussian splatting that turns camera poses and sparse points into splat scene fil…	91	2205	active
NX-AI/xlstm Official PyTorch implementation of xLSTM, an extended Long Short-Term Memory recurrent architecture with exponential gating and matrix memo…	76	2205	active
lucidrains/lion-pytorch A PyTorch implementation of the Lion optimizer (Evolved Sign Momentum), discovered by Google Brain via genetic algorithms and claimed to ou…	59	2205	active
kleinlee/DH_live DH_live (mini) is an open-source 2D talking-head digital human toolkit that generates real-time lip-synced avatar video from a single refer…	64	2202	active
RightNow-AI/picolm PicoLM is a pure C11 LLM inference engine that runs 1-billion parameter models in GGUF format on extremely constrained hardware like $10 bo…	43	2202	active
OSQP OSQP is a C library for solving convex quadratic programs using an ADMM-based operator splitting method, with interfaces to Python, Julia, …	47	2201	stable
spla-tam/SplaTAM SplaTAM is a dense RGB-D SLAM system that uses 3D Gaussian splatting for high-fidelity scene reconstruction and precise camera tracking fro…	18	2200	active
ZJU-FAST-Lab/ego-planner-swarm EGO-Swarm is a decentralized, asynchronous trajectory planner for single and multi-agent quadrotor navigation in unknown, obstacle-rich env…	33	2197	stable
AlphaGenome A Python SDK providing programmatic access to Google DeepMind's AlphaGenome model, which predicts genomic regulatory outputs such as gene e…	89	2192	active