Open-source machine-learning software

4007 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.

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

Product	Health v2	Stars	Maturity
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
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
unrealcv/unrealcv UnrealCV is an open-source Unreal Engine plugin that connects computer vision research to virtual worlds by exposing a command API and Pyth…	94	2219	active
V-Modal AI SDKs VModal for Flutter is a Dart/Flutter SDK that adds multimodal video search (semantic, ASR, and OCR based) and streamed video uploads to And…	57	2212	active
MVIG-SJTU/AlphaPose AlphaPose is an open-source real-time multi-person full-body pose estimation and tracking system built on PyTorch. It detects human keypoin…	32	8605	maintenance
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
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
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
TensorFlow Extended (TFX) TensorFlow Extended (TFX) is an end-to-end, Google-production-scale platform for building and deploying production machine learning pipelin…	87	2191	stable
MoonshotAI/MoBA MoBA (Mixture of Block Attention) is a PyTorch implementation of a trainable block-sparse attention mechanism for long-context large langua…	24	2191	active
yyfz/Pi3 Pi3 (π³) is a feed-forward neural network for visual geometry reconstruction that eliminates the need for a fixed reference view, using a p…	56	2188	active
withcatai/node-llama-cpp node-llama-cpp is a Node.js library providing bindings to llama.cpp for running LLMs locally, with pre-built binaries and automatic GPU sup…	95	2187	active
yatengLG/ISAT_with_segment_anything ISAT_with_segment_anything is an interactive semi-automatic image annotation tool built on the Segment Anything Model family (SAM, SAM2, SA…	81	2186	active
tensorflow/quantum TensorFlow Quantum is a Python framework for hybrid quantum-classical machine learning that integrates Cirq quantum circuits with TensorFlo…	80	2186	active
hls4ml hls4ml is a Python package that converts machine learning models from Keras, PyTorch, and ONNX into high-level synthesis (C++) code for FPG…	80	2185	active
Harry24k/adversarial-attacks-pytorch Torchattacks is a PyTorch library providing implementations of adversarial attacks to generate adversarial examples against deep learning m…	23	2185	active
ViTAE-Transformer/ViTPose Official PyTorch implementation of ViTPose and ViTPose++, Vision Transformer models for human and generic body pose estimation from NeurIPS…	55	2173	stable
intel/intel-extension-for-transformers Intel's toolkit for accelerating transformer-based GenAI/LLM workloads on Intel platforms, offering state-of-the-art compression (e.g., INT…	10	2169	active
TsingZ0/PFLlib PFLlib is a beginner-friendly Python library and benchmark for traditional and personalized federated learning, offering 39 algorithms, 24 …	55	2166	active
PrimeIntellect-ai/prime-rl prime-rl is a Python framework for large-scale, fully asynchronous reinforcement learning training of language models, built on FSDP2 for t…	88	2164	active
hexo-ai/sia SIA is a Python framework implementing a self-improving AI loop in which a Meta-Agent generates a task-specific Target Agent and a Feedback…	56	2161	active
SpikingJelly SpikingJelly is an open-source deep learning framework for Spiking Neural Networks (SNNs) built on PyTorch. It provides a beginner-friendly…	77	2159	active
rupeshs/fastsdcpu FastSD CPU is a Python application that runs Stable Diffusion image generation quickly on CPUs and Intel AI PCs using Latent Consistency Mo…	75	2158	active
facebookresearch/mae A PyTorch/GPU re-implementation of the Masked Autoencoders (MAE) paper for self-supervised vision learning. It includes pre-training code, …	10	8362	maintenance
google/gin-config Gin-config is a lightweight Python configuration framework based on dependency injection. Functions or classes decorated with @gin.configur…	67	2157	stable
NannyML/nannyml NannyML is an open-source Python library for post-deployment machine learning monitoring. It estimates model performance without access to …	36	2155	active
google-deepmind/pysc2 PySC2 is DeepMind's Python component of the StarCraft II Learning Environment, exposing Blizzard's StarCraft II Machine Learning API as a P…	23	8329	maintenance
google/trax Trax is an end-to-end deep learning library built on JAX and TensorFlow that focuses on clear code and speed, developed and maintained by t…	10	8306	maintenance
tumuyan/RealSR-NCNN-Android An Android application for image super-resolution and upscaling built on NCNN and MNN inference engines, bundling models like RealSR, Real-…	77	2144	active
intel/openvino-plugins-ai-audacity A set of AI-enabled effects, generators, and analyzers for Audacity, powered by Intel OpenVINO running fully locally on CPU, GPU, or NPU. I…	75	2142	active
cvxpy/cvxpylayers CVXPYlayers is a Python library for constructing differentiable convex optimization layers in PyTorch, JAX, and MLX using CVXPY. It solves …	90	2141	active
utkuozbulak/pytorch-cnn-visualizations A PyTorch library implementing a wide range of convolutional neural network visualization and interpretability techniques, including Grad-C…	32	8236	maintenance
autonomousvision/sdfstudio SDFStudio is a unified and modular framework for neural implicit surface reconstruction built on top of nerfstudio. It provides unified imp…	32	2124	active
marlbenchmark/on-policy The official PyTorch implementation of Multi-Agent PPO (MAPPO), an on-policy reinforcement learning algorithm for cooperative multi-agent s…	32	2123	active
sunnypilot/sunnypilot sunnypilot is an open-source driver assistance system forked from comma.ai's openpilot, providing adaptive cruise control, automated lane c…	95	2121	active
deepspeedai/DeepSpeed-MII DeepSpeed-MII is a Python library for high-throughput, low-latency large language model inference, built on DeepSpeed. It provides blocked …	39	2114	active
facebookresearch/ai4animationpy AI4AnimationPy is a Python framework for AI-driven character animation using neural networks, providing motion capture processing, training…	56	2111	active
automl/auto-sklearn auto-sklearn is an automated machine learning (AutoML) toolkit that acts as a drop-in replacement for scikit-learn estimators, automaticall…	65	8135	maintenance
patrick-kidger/diffrax Diffrax is a JAX-based library providing numerical differential equation solvers for ODEs, SDEs, and CDEs. It is fully autodifferentiable a…	80	2106	active
Avalanche Avalanche is an end-to-end continual learning library built on PyTorch, developed by ContinualAI. It provides modules for benchmarks, train…	24	2104	active
kohya-ss/musubi-tuner Musubi Tuner is a set of Python scripts for training LoRA (Low-Rank Adaptation) adapters for video and image generation model architectures…	88	2103	active
HRM-Text HRM-Text is a 1B-parameter text generation model based on the hierarchical recurrent HRM architecture, released with a complete pretraining…	57	2093	active
roboterax/humanoid-gym Humanoid-Gym is a reinforcement learning framework built on Nvidia Isaac Gym for training locomotion skills in humanoid robots, with emphas…	26	2093	active
vitoplantamura/OnnxStream A lightweight C++ inference library for ONNX models that streams weights to run large models in very little memory, accelerated by XNNPACK.…	56	2089	active
NanoNets/docext docext is an on-premises document intelligence toolkit powered by vision-language models, offering OCR-free structured data extraction, PDF…	47	2086	active
OpenSeeFace OpenSeeFace is a robust realtime face and facial landmark tracking library that runs on CPU at 30-60 fps using ONNX-converted MobileNetV3 m…	85	2084	active
uber/orbit Orbit is a Python package for Bayesian time series forecasting and inference with an initialize-fit-predict interface. It uses probabilisti…	76	2079	stable
NVIDIA-NeMo/RL NeMo RL is NVIDIA's open-source post-training library for scaling reinforcement learning methods (GRPO, PPO, DPO, SFT, distillation) on LLM…	79	2077	active
jeshraghian/snntorch snnTorch is a Python library for gradient-based deep learning with spiking neural networks, built as an extension of PyTorch. It provides s…	68	2075	active
marcoslucianops/DeepStream-Yolo A collection of configuration files, parsers, and conversion utilities for running YOLO-family object detection models on NVIDIA DeepStream…	57	2074	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…	56	2074	active
WenjieDu/PyPOTS PyPOTS is a Python toolbox for machine learning and data mining on partially-observed time series with missing values. It integrates 50+ st…	89	2072	active
handy-computer/transcribe.cpp A C/C++ speech-to-text inference library built on the ggml runtime that runs 16+ ASR model families (Whisper, Parakeet, Canary, Moonshine, …	82	2071	active
bytedance/Protenix Protenix is an open-source PyTorch reproduction of AlphaFold 3 for high-accuracy biomolecular structure prediction, covering proteins, nucl…	78	2070	active
facebookresearch/ConvNeXt-V2 Official PyTorch implementation of ConvNeXt V2, a family of pure convolutional neural network models co-designed with a fully convolutional…	10	2065	stable
brightmart/text_classification A collection of deep learning baseline models for text classification in NLP, implemented in TensorFlow. It covers classic architectures li…	32	7938	maintenance
OpenMOSS/MOSS-Transcribe-Diarize MOSS-Transcribe-Diarize 0.9B is an open-source end-to-end audio understanding model that jointly performs multi-speaker speech transcriptio…	57	2064	active
facebookresearch/theseus Theseus is a PyTorch-based library for building custom differentiable nonlinear optimization layers, supporting problems in robotics and vi…	23	2060	active
ricky0123/vad A JavaScript/TypeScript library that runs Silero VAD via ONNX Runtime Web to detect voice activity directly in the browser. It provides a s…	96	2059	active
zju3dv/GVHMR GVHMR is a research codebase implementing the SIGGRAPH Asia 2024 paper 'World-Grounded Human Motion Recovery via Gravity-View Coordinates'.…	66	2059	active
DeePMD-kit DeePMD-kit is a deep learning package for building many-body potential energy representations and running molecular dynamics simulations. I…	92	2055	stable
n00mkrad/flowframes Flowframes is a Windows GUI application for AI-based video frame interpolation, supporting RIFE (Pytorch & NCNN), DAIN (NCNN), and FLAVR (P…	66	2048	active
EvalAI EvalAI is an open-source platform for evaluating and comparing machine learning and AI algorithms at scale. It provides a central leaderboa…	67	2045	active
juanmc2005/diart Diart is a Python framework for building AI-powered real-time audio applications, best known for state-of-the-art streaming speaker diariza…	59	2041	active
noamgat/lm-format-enforcer A Python library that enforces output formats like JSON Schema or regex on language models by filtering allowed tokens at each generation s…	60	2040	active
pykeen/pykeen PyKEEN is a Python library for training and evaluating multimodal knowledge graph embedding models built on PyTorch. It provides a high-lev…	69	2039	active
tensorflow/privacy TensorFlow Privacy is a Python library providing TensorFlow optimizers for training machine learning models with differential privacy. It i…	64	2036	active
SakanaAI/continuous-thought-machines The Continuous Thought Machine (CTM) is a neural network architecture from Sakana AI that uses neuron-level temporal dynamics and neural sy…	43	2036	active
tensorflow/recommenders TensorFlow Recommenders is a Python library for building recommender system models on top of TensorFlow and Keras. It covers the full workf…	84	2034	active
bytetriper/RAE Official PyTorch implementation of 'Diffusion Transformers with Representation Autoencoders' (RAE), a two-stage image generation pipeline u…	45	2032	active
tdrussell/diffusion-pipe A Python training script for fine-tuning diffusion models (image and video generation) using DeepSpeed pipeline parallelism across multiple…	67	2028	active
NUS-HPC-AI-Lab/VideoSys VideoSys is an open-source Python library providing easy and efficient infrastructure for video generation, supporting training, inference,…	30	2025	active
tatsu-lab/alpaca_eval AlpacaEval is an LLM-based automatic evaluation framework for instruction-following language models, producing win rates against a GPT-4 ba…	33	2018	active
uncertainty-toolbox/uncertainty-toolbox A Python library for predictive uncertainty quantification, providing metrics, visualizations, and recalibration procedures for regression …	23	2015	active
meta-recsys/generative-recommenders Meta's research library implementing HSTU and M-FALCON from the ICML'24 paper 'Actions Speak Louder than Words: Trillion-Parameter Sequenti…	70	2012	active
philipperemy/keras-tcn A Keras/TensorFlow implementation of Temporal Convolutional Networks (TCN) with dilated causal convolutions, usable as a drop-in layer alte…	58	2011	active
cambrian-mllm/cambrian Cambrian-1 is a fully open family of vision-centric multimodal large language models (MLLMs) from NYU's VISIONx group, with training and ev…	43	2011	active
lyhue1991/torchkeras torchkeras is a lightweight PyTorch model training template library that brings Keras-style compile/fit/evaluate APIs to PyTorch. Its core …	52	2010	active
AntixK/PyTorch-VAE A collection of Variational Autoencoder (VAE) model implementations in PyTorch, including Beta-VAE, VQ-VAE, IWAE, WAE, and others, with a f…	34	7678	maintenance
FireRedASR2S FireRedASR is a family of open-source industrial-grade automatic speech recognition models supporting Mandarin, Chinese dialects, and Engli…	49	2005	active
mil-tokyo/webdnn WebDNN is a framework for running deep neural network inference directly in the web browser, accepting ONNX models without Python preproces…	68	2003	active
chaidiscovery/chai-lab Chai-1 is a state-of-the-art multi-modal foundation model for biomolecular structure prediction, handling proteins, small molecules, DNA, R…	62	2001	active
clementchadebec/benchmark_VAE Pythae is a PyTorch library that unifies implementations of many Variational Autoencoder (VAE) variants under a common interface, enabling …	23	2000	active
elyra-ai/elyra Elyra is a set of AI-centric extensions for JupyterLab, including a visual pipeline editor for building and executing notebook-based pipeli…	67	1998	active
antirez/iris.c Iris is a pure C inference pipeline that generates images from text prompts using open-weights diffusion transformer models like FLUX.2 Kle…	43	1994	active
hkchengrex/XMem XMem is a PyTorch model for semi-supervised video object segmentation that tracks objects through long videos using an Atkinson-Shiffrin-in…	23	1994	stable
facebookresearch/dino PyTorch implementation of DINO, a self-supervised learning method for training Vision Transformers, with pretrained model weights. It is th…	10	7609	maintenance
kwuking/TimeMixer Official PyTorch implementation of TimeMixer, an ICLR 2024 model for time series forecasting using decomposable multiscale mixing. It has s…	42	1991	active
featureform/featureform Featureform is a virtual feature store that sits atop your existing data infrastructure and orchestrates it to define, manage, and serve ML…	32	1991	active
jianchang512/vocal-separate A minimal local web-based tool for separating vocals from background music in audio or video files, using Spleeter 2stems/4stems/5stems mod…	10	1990	active
haykgrigo3/TimeCapsuleLLM TimeCapsuleLLM is a research project training language models from scratch (and fine-tuning small base models) exclusively on text from spe…	70	1989	active
Hyperopt Hyperopt is a Python library for distributed asynchronous hyperparameter optimization over search spaces with real-valued, discrete, and co…	84	7595	maintenance
GPT2-Chinese A Python library providing GPT-2 training and text generation code tailored for Chinese, built on HuggingFace Transformers with BERT or BPE…	32	7594	maintenance