domain: machine-learning

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

Product	Health v2	Stars	Maturity
meta-pytorch/opacus Opacus is a PyTorch library for training neural networks with differential privacy via DP-SGD, requiring minimal code changes through its P…	79	1952	active
autodiff/autodiff autodiff is a C++17 library for automatic differentiation that computes derivatives of functions efficiently using forward mode (dual numbe…	24	1952	active
haoheliu/versatile_audio_super_resolution AudioSR is a Python library and CLI tool that performs versatile audio super-resolution, upsampling any audio (music, speech, sound effects…	45	1951	active
openai/guided-diffusion OpenAI's codebase for guided diffusion models from the paper 'Diffusion Models Beat GANs on Image Synthesis', including classifier conditio…	32	7419	maintenance
kyegomez/BitNet A PyTorch implementation of the BitNet architecture from the paper 'BitNet: Scaling 1-bit Transformers for Large Language Models', providin…	72	1945	active
Graph Convolutional Networks (GCN) A TensorFlow implementation of Graph Convolutional Networks (GCN) for semi-supervised node classification on graphs, accompanying the ICLR …	32	7400	maintenance
tensorlayer/TensorLayer TensorLayer is a TensorFlow-based deep learning and reinforcement learning library offering customizable neural layers for researchers and …	23	7381	maintenance
starik222/BooruDatasetTagManager A desktop tag editor for managing booru-style tagged image and video datasets used to train Stable Diffusion models such as LoRAs, embeddin…	76	1938	active
JuliaAI/MLJ.jl MLJ.jl is a machine learning framework for Julia providing a common interface to over 200 models, with meta-algorithms for model selection,…	93	1935	stable
NVlabs/RADIO Official PyTorch implementation of AM-RADIO and its successors (RADIOv2.5, C-RADIOv4), agglomerative vision foundation models distilled fro…	64	1933	active
PAIR-code/facets Facets is a pair of web-component visualizations (Overview and Dive) for understanding and analyzing machine learning datasets, embeddable …	10	7336	maintenance
tum-pbs/PhiFlow PhiFlow is an open-source Python simulation toolkit for solving partial differential equations with support for optimization and machine le…	72	1929	active
Audio-AGI/AudioSep AudioSep is the official implementation of the 'Separate Anything You Describe' foundation model for open-domain, language-queried audio so…	28	1929	active
GPflow/GPflow GPflow is a Python library for building Gaussian process models on top of TensorFlow 2 and TensorFlow Probability. It implements modern Gau…	95	1916	active
NVIDIA/aistore AIStore (AIS) is a lightweight, distributed object storage stack built by NVIDIA specifically for AI workloads. It provides linear scalabil…	95	1915	active
microsoft/DiskANN DiskANN3 is a composable Rust library for scalable, accurate vector indexing and nearest-neighbor search, designed to be embedded into data…	99	1913	active
FACEGOOD/FACEGOOD-Audio2Face FACEGOOD Audio2Face is an open-source deep learning framework that converts audio into facial blendshape weights for driving digital humans…	64	1909	active
diffgram/diffgram Diffgram is a self-hosted AI datastore for managing schemas, BLOBs, and predictions, with built-in human supervision (data labeling), data …	62	1909	active
mlco2/codecarbon CodeCarbon is a lightweight Python library and CLI that estimates and tracks CO2 emissions from local computing hardware (CPU, GPU, RAM) by…	98	1905	active
visual-layer/fastdup fastdup is a free Python tool for rapidly analyzing image and video datasets to surface duplicates, outliers, broken, dark, bright, blurry,…	67	1904	active
lucidrains/byol-pytorch A PyTorch library implementing the Bootstrap Your Own Latent (BYOL) self-supervised learning method from DeepMind. It wraps any image-based…	58	1903	active
google-deepmind/penzai Penzai is a JAX research toolkit for building, editing, and visualizing neural networks as legible, functional pytree data structures. It i…	10	1901	active
TheSpaghettiDetective/obico-server Obico Server is the self-hostable backend of the Obico smart 3D printing platform, providing AI-based print failure detection, webcam strea…	76	1899	active
sapientinc/HRM-Text HRM-Text is a 1B-parameter text generation model based on the hierarchical recurrent HRM architecture, released with a complete pretraining…	53	1899	active
flexflow/flexflow-train FlexFlow Train is a deep learning framework that accelerates distributed DNN training by automatically searching for efficient parallelizat…	67	1898	active
AutoViML/AutoViz AutoViz is a Python library that automatically generates visualizations for any dataset, of any size, with a single line of code. It also i…	32	1898	active
dynamicslab/pysindy PySINDy is a Python library for discovering governing equations of dynamical systems from measurement data using the Sparse Identification …	73	1893	active
LeapLabTHU/Absolute-Zero-Reasoner Official implementation of Absolute Zero Reasoner (AZR), a system that trains LLM reasoning via reinforced self-play with zero external dat…	36	1893	active
llvm/torch-mlir Torch-MLIR is a compiler project providing first-class translation of PyTorch programs into the MLIR compiler ecosystem. It lets hardware v…	67	1892	active
stanfordmlgroup/ngboost NGBoost is a Python library implementing Natural Gradient Boosting for probabilistic prediction, built on top of scikit-learn. It outputs f…	88	1887	active
ContextLab/hypertools HyperTools is a Python library for visualizing and exploring high-dimensional data via dimensionality reduction, built on matplotlib, sciki…	84	1887	active
d8ahazard/sd_dreambooth_extension A Stable Diffusion WebUI extension that adds DreamBooth fine-tuning capabilities, ported from Shivam Shrirao's optimized Diffusers implemen…	42	1887	active
netease-youdao/BCEmbedding BCEmbedding is NetEase Youdao's open-source library of bilingual and crosslingual embedding and reranker models for English and Chinese, bu…	44	1883	active
zju3dv/GVHMR GVHMR is a research codebase implementing the SIGGRAPH Asia 2024 paper 'World-Grounded Human Motion Recovery via Gravity-View Coordinates'.…	60	1882	active
guess-js/guess Guess.js is a collection of libraries and tools that use analytics data and machine learning to predict user navigation patterns on website…	77	7119	maintenance
trevorstephens/gplearn gplearn is a Python library implementing genetic programming with a scikit-learn compatible API, focused on symbolic regression. It provide…	75	1881	stable
qqwweee/keras-yolo3 A Keras (TensorFlow backend) implementation of YOLOv3 for object detection, including Darknet weight conversion, image/video detection scri…	32	7114	maintenance
laugh12321/TensorRT-YOLO A C++/Python deployment toolkit for running YOLO-family models (YOLOv3 through YOLO26) on NVIDIA GPUs using TensorRT, with custom plugins, …	63	1880	active
rednote-machine-learning/RedKnot RedKnot is a long-context LLM inference acceleration library built on SGLang, using head-classified KV reuse, offline KV storage with RoPE …	58	1880	active
SqueezeAILab/LLMCompiler LLMCompiler is a Python framework from an ICML 2024 paper that orchestrates parallel function calling with LLMs. It uses a planner, task fe…	26	1880	active
nitrain/nitrain Nitrain is a framework-agnostic Python library for sampling, augmenting, and training AI models on medical imaging datasets, with support f…	23	1880	active
nomic-ai/nomic The official Python client for Nomic Atlas, a platform for exploring, labeling, searching, and sharing massive unstructured datasets (text,…	63	1878	active
425776024/nlpcda nlpcda is a Python library for one-click Chinese NLP data augmentation. It provides techniques like random entity replacement, synonym subs…	28	1878	active
ashvardanian/NumKong NumKong is a mixed-precision numeric and linear-algebra library with ~2,000 SIMD-accelerated kernels for dot products, distances, matrix mu…	93	1877	active
INESCTEC/yake YAKE! is a lightweight unsupervised keyword extraction library that ranks the most important keywords of a single document using text stati…	63	1877	stable
johannfaouzi/pyts pyts is a Python package for time series classification that provides preprocessing, transformation, and utility tools along with implement…	35	1876	stable
robertknight/ocrs Ocrs is a Rust library and CLI tool for optical character recognition that extracts text from images such as scanned documents, photos, and…	69	1875	active
timsainb/noisereduce A Python library for noise reduction in time-domain signals using spectral gating, supporting stationary and non-stationary noise reduction…	39	1874	active
c2g-dev/city2graph City2Graph is a Python library that converts geospatial data such as buildings, streets, GTFS transit feeds, and origin-destination matrice…	86	1872	active
PRIME-RL/PRIME PRIME (Process Reinforcement through Implicit Rewards) is an open-source Python framework for online reinforcement learning with process re…	26	1871	active
nndeploy/nndeploy nndeploy is an easy-to-use, high-performance AI deployment framework written in C++ with Python bindings. It provides a visual drag-and-dro…	87	1868	active
e-p-armstrong/augmentoolkit Augmentoolkit is a Python application that generates domain-expert fine-tuning datasets from uploaded documents and trains custom LLMs on t…	60	1864	active
asavinov/intelligent-trading-bot A Python-based automated cryptocurrency trading bot that generates trade signals using machine learning and feature engineering. It separat…	76	1860	active
patrick-kidger/jaxtyping A Python library providing type annotations and runtime type-checking for the shape and dtype of arrays and tensors in JAX, PyTorch, NumPy,…	95	1859	active
laekov/fastmoe FastMoE is a PyTorch library providing efficient Mixture of Experts (MoE) layers with custom C/CUDA operators. It supports distributed expe…	26	1859	active
OpenNMT OpenNMT is an open-source ecosystem for neural machine translation and sequence learning, with PyTorch (OpenNMT-py) and TensorFlow (OpenNMT…	44	7012	maintenance
facebookresearch/MetaCLIP Meta's research code and models for Meta CLIP, a reimplementation and scaling recipe for CLIP-style contrastive vision-language models, inc…	82	1854	active
SerpentAI/SerpentAI Serpent.AI is a Python framework for building game agents—AIs and bots that learn to play any video game you own—turning games into machine…	10	6992	maintenance
Tele-AI/Telechat TeleChat is a family of open-source bilingual (Chinese-English) large language models (1B, 7B, 12B) developed by China Telecom's AI team, r…	67	1853	active
openreasoner/openr OpenR is an open-source Python framework that integrates search, reinforcement learning, and process supervision to improve chain-of-though…	23	1853	active
LuChengTHU/dpm-solver Official PyTorch implementation of DPM-Solver and DPM-Solver++, fast high-order ODE solvers for diffusion probabilistic model sampling that…	32	1852	stable
dotnet/TorchSharp TorchSharp is a .NET library providing bindings to LibTorch, the library that powers PyTorch, with a focus on tensors and a PyTorch-like AP…	73	1850	active
probcomp/Gen.jl Gen.jl is a general-purpose probabilistic programming system embedded in Julia that lets users write generative models as probabilistic pro…	62	1850	active
bmabey/pyLDAvis pyLDAvis is a Python library for interactive visualization of fitted LDA topic models, ported from the R LDAvis package. It renders web-bas…	48	1850	active
descriptinc/descript-audio-codec Descript Audio Codec (.dac) is a high-fidelity neural audio codec based on improved RVQGAN that compresses audio at roughly 90x compression…	61	1846	stable
dphnAI/sonar Sonar is a large-scale LLM inference engine for Hugging Face-compatible language and multimodal models, based on vLLM (formerly the Aphrodi…	94	1843	active
mlc-ai/xgrammar XGrammar is an open-source library for efficient, flexible, and portable structured generation in LLM inference via constrained decoding. I…	88	1843	active
ytongbai/LVM LVM is a large vision model trained with sequential next-token prediction over 'visual sentences', using no linguistic data. It builds on O…	30	1838	active
clovaai/donut Donut is an OCR-free end-to-end Transformer model for visual document understanding tasks such as document classification and information e…	23	6919	maintenance
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, …	81	1836	active
cazala/synaptic Synaptic is an architecture-free neural network library for JavaScript that runs in both Node.js and the browser. It supports building and …	66	6912	maintenance
PRIME-RL/SimpleVLA-RL SimpleVLA-RL is an open-source reinforcement learning framework for training Vision-Language-Action (VLA) models for robotic manipulation, …	46	1834	active
AgiBot X1 The inference module of the open-source AgiBot X1 humanoid robot software stack, providing model inference, platform drivers, and simulatio…	28	1834	active
dauparas/ProteinMPNN ProteinMPNN is a PyTorch-based tool that designs amino acid sequences for given protein backbone structures using a message-passing neural …	23	1834	stable
ZFTurbo/Weighted-Boxes-Fusion A Python library implementing several methods for ensembling bounding boxes from multiple object detection models, including Non-maximum Su…	65	1827	stable
microsoft/responsible-ai-toolbox A Microsoft suite of Python libraries and Jupyter widgets providing dashboards for model assessment, error analysis, interpretability, fair…	67	1824	active
boundless-large-model/boundless-world-model Boundless-World-Model (BWM) is a physically consistent, action-conditioned video world model built on Wan2.2-TI2V-5B that acts as a low-cos…	59	1824	active
potamides/DeTikZify DeTikZify is a Python library and research tool that uses multimodal large language models to synthesize TikZ/LaTeX graphics programs from …	49	1817	active
GestaltCogTeam/BasicTS BasicTS is a Python benchmark library and toolkit for fair and scalable time series analysis, built on PyTorch. It supports forecasting, cl…	68	1814	active
fastino-ai/GLiNER2 GLiNER2 is a Python library for schema-based information extraction and text classification, supporting named entity recognition, structure…	85	1813	active
trevin-creator/autoresearch-mlx An Apple Silicon (MLX) port of Karpathy's autoresearch that runs autonomous AI research loops on a Mac without PyTorch or CUDA. A coding ag…	55	1813	active
avhz/RustQuant RustQuant is a Rust library for quantitative finance, offering modules for option pricing, stochastic processes, autodiff, statistics, and …	58	1812	active
RoboVerseOrg/RoboVerse RoboVerse is a unified platform, dataset, and benchmark for scalable and generalizable robot learning, providing tasks, robots, scenes, ass…	68	1810	active
ROCm/FastFlowLM FastFlowLM (FLM) is an NPU-first LLM inference runtime purpose-built and deeply optimized for AMD Ryzen AI NPUs (XDNA2), offering an Ollama…	85	1809	active
apple/ml-4m 4M is a framework from Apple and EPFL for training any-to-any multimodal foundation models using masked modeling over discrete tokens acros…	35	1808	active
Angel-ML/angel Angel is a high-performance distributed parameter server for large-scale machine learning and graph computing, developed by Tencent and Pek…	69	6787	maintenance
worldveil/dejavu Dejavu is a Python library that fingerprints audio files and recognizes songs by matching recorded microphone input or audio from disk agai…	32	6784	maintenance
Robbyant/lingbot-va LingBot-VA is an autoregressive diffusion framework that unifies video world modeling and robot policy learning in a single interleaved vid…	56	1806	active
undertheseanlp/underthesea Underthesea is an open-source Python toolkit for Vietnamese natural language processing that has evolved into an agentic AI toolkit with mu…	94	1804	active
evalplus/evalplus EvalPlus is a Python framework and benchmark suite for rigorously evaluating LLM-synthesized code, providing HumanEval+ and MBPP+ datasets …	41	1804	active
microsoft/mattergen MatterGen is Microsoft's official implementation of a generative diffusion model for designing inorganic crystalline materials across the p…	67	1801	active
Xilinx/Vitis-AI AMD Vitis AI is an integrated development environment and stack for accelerating AI inference on AMD/Xilinx adaptable platforms, including …	71	1800	active
inference-labs-inc/inference-network Inference Network is a Proof of Inference system built on EigenLayer that provides trustless verification of AI inference results via Solid…	49	1799	active
google-research/language A shared repository collecting open-sourced NLP research projects from the Google AI Language team, written in Python. It serves as a home …	71	1795	active
FL33TW00D/whisper-turbo Whisper Turbo is a fast, cross-platform, GPU-accelerated implementation of OpenAI's Whisper speech recognition model, built on the Ratchet …	19	1794	active
acids-ircam/RAVE RAVE is the official PyTorch implementation of a realtime audio variational autoencoder for fast, high-quality neural audio synthesis. It s…	55	1790	active
lancopku/pkuseg-python pkuseg is a Python toolkit for multi-domain Chinese word segmentation based on the paper by Luo et al. (2019). It provides pre-trained mode…	23	6710	maintenance
supervc-stack/VectorChord VectorChord is a PostgreSQL extension written in Rust for scalable, disk-efficient vector similarity search, serving as the successor to pg…	79	1785	active
lehaifeng/T-GCN A collection of research source code implementing Temporal Graph Convolutional Networks (T-GCN) and related variants for urban traffic flow…	50	1781	active
Simple-Efficient/RL-Factory RLFactory is a reinforcement learning post-training framework for training agentic LLM models, decoupling the environment from RL training …	34	1780	active