domain: machine-learning

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

Product	Health v2	Stars	Maturity
ace-agent/ace ACE is a Python framework for Agentic Context Engineering that lets language models self-improve by treating contexts as evolving playbooks…	61	1275	active
Ma-Lab-Berkeley/CRATE CRATE is the official PyTorch implementation of the Coding RAte reduction TransformEr, a family of 'white-box' transformer architectures de…	29	1275	active
google-research/simclr Google Research's official implementation of SimCLR and SimCLRv2, a framework for contrastive learning of visual representations, with 65 p…	10	4502	maintenance
turboderp-org/exllamav3 ExLlamaV3 is a Python library for fast quantization and inference of large language models on consumer-class GPUs, featuring the EXL3 quant…	87	1274	active
flutter-ml/google_ml_kit_flutter A set of Flutter plugins wrapping Google's ML Kit on-device machine learning APIs for Android and iOS. It provides vision APIs (barcode, fa…	76	1274	active
DreamTechAI/Direct3D-S2 Direct3D-S2 is a research framework for high-resolution 3D shape generation from images, built on sparse volumetric representations and a n…	29	1274	active
dcharatan/pixelsplat pixelSplat is a PyTorch implementation of a feed-forward model that reconstructs 3D radiance fields parameterized by 3D Gaussian primitives…	27	1274	stable
nianticlabs/monodepth2 Monodepth2 is the reference PyTorch implementation of the ICCV 2019 paper 'Digging into Self-Supervised Monocular Depth Prediction'. It tra…	32	4497	maintenance
SimonBlanke/Gradient-Free-Optimizers A Python library for gradient-free optimization of black-box functions, offering 23 algorithms (hill climbing, Bayesian optimization, parti…	94	1273	active
Renumics/spotlight Renumics Spotlight is an open-source tool for interactively exploring unstructured datasets (images, audio, text, video, time-series, meshe…	93	1272	active
NVlabs/stylegan2-ada-pytorch Official PyTorch implementation of StyleGAN2-ADA, a generative adversarial network with adaptive discriminator augmentation for training wi…	32	4487	maintenance
Visual-Agent/DeepEyes DeepEyes is a research project that trains multimodal vision-language models to 'think with images' using end-to-end reinforcement learning…	44	1271	active
TheRock TheRock is AMD's open-source build and release system for the ROCm software stack, replacing the legacy monolithic ROCm release process wit…	89	1270	active
leoxiaobin/deep-high-resolution-net.pytorch Official PyTorch implementation of HRNet (Deep High-Resolution Representation Learning for Human Pose Estimation, CVPR 2019), which maintai…	32	4480	maintenance
Nixtla/mlforecast mlforecast is a Python framework for time series forecasting using any machine learning model with fit/predict methods, providing efficient…	96	1269	active
MilaNLProc/contextualized-topic-models A Python library implementing Contextualized Topic Models (CTM), which combine pre-trained contextual embeddings like BERT with neural topi…	47	1269	active
luo3300612/Visualizer A lightweight Python library that extracts attention maps and other local variables from deep inside PyTorch models for visualization. It w…	32	1269	stable
BachiLi/diffvg diffvg is a differentiable rasterizer for 2D vector graphics that bridges the raster and vector domains via backpropagation. It computes pi…	42	1268	active
amaiya/ktrain ktrain is a lightweight Python wrapper around TensorFlow Keras that provides pre-canned, low-code models for text, vision, graph, and tabul…	25	1268	active
thunlp/OpenNRE OpenNRE is an open-source Python toolkit for neural relation extraction, extracting relation triples between entities from plain text. It u…	32	4467	maintenance
magenta/magenta-studio Magenta Studio is a collection of five MIDI music generation plugins (Continue, Generate, Interpolate, Groove, Drumify) for Ableton Live, b…	62	1266	active
Bjarten/early-stopping-pytorch A small PyTorch utility package providing an EarlyStopping class that monitors validation loss during training and stops when it stops impr…	34	1266	stable
tensorflow/model-analysis TensorFlow Model Analysis (TFMA) is a Python library for evaluating TensorFlow models on large datasets in a distributed manner. It compute…	83	1265	active
muscriptor/muscriptor MuScriptor is a multi-instrument music transcription model by Kyutai and Mirelo that converts audio recordings into MIDI and sheet music. I…	79	1263	active
meta-pytorch/data TorchData is a PyTorch library providing scalable, performant data loading utilities, including StatefulDataLoader, a drop-in replacement f…	67	1263	active
bryandlee/animegan2-pytorch A PyTorch implementation of AnimeGANv2, a GAN-based image-to-image style transfer model that converts photos into anime-style images. It pr…	32	4452	maintenance
stair-lab/kg-gen kg-gen is a Python library that extracts knowledge graphs from arbitrary plain text or conversation messages using LLMs, with model routing…	63	1260	active
mjpost/sacrebleu SacreBLEU is a Python library and CLI tool for computing shareable, comparable, and reproducible BLEU, chrF, and TER scores for machine tra…	76	1258	active
0xSero/ai-data-extraction A Python toolkit of extraction scripts that pulls complete chat, agent, and code-context history from AI coding assistants like Cursor, Cla…	60	1257	active
kubeai-project/kubeai KubeAI is a Kubernetes operator for serving machine learning models in production, supporting LLMs via vLLM and Ollama, vector embeddings, …	89	1256	active
aiming-lab/Agent0 Agent0 Series is a research framework for training self-evolving LLM agents from zero external data via tool-integrated reasoning and co-ev…	58	1256	active
BytedTsinghua-SIA/CUDA-Agent CUDA-Agent is a large-scale agentic reinforcement learning system from ByteDance Seed and Tsinghua that trains LLMs to generate high-perfor…	56	1256	active
stardist/stardist StarDist is a Python library for object detection and instance segmentation in 2D and 3D microscopy images using star-convex shapes, built …	60	1255	stable
JonathonLuiten/TrackEval TrackEval is a Python library for evaluating multi-object tracking (MOT) algorithms, implementing metrics such as HOTA, CLEARMOT, IDF1, VAC…	32	1255	stable
facebookresearch/llm-transparency-tool An interactive toolkit from Meta Research for analyzing the internal workings of Transformer-based language models. It visualizes contribut…	10	1254	active
TencentCloudADP/youtu-graphrag Youtu-GraphRAG is a Python framework for graph-based retrieval-augmented generation that unifies graph schema construction, community detec…	57	1253	active
VainF/pytorch-msssim A PyTorch library providing fast, differentiable SSIM and MS-SSIM image quality metrics using separable Gaussian filtering for speed. It ca…	23	1253	stable
Neroued/ninfer NInfer is a from-scratch C++/CUDA inference engine optimized for maximum single-GPU performance on a narrow, explicitly registered set of Q…	58	1252	active
tatuylonen/wiktextract A Python package and CLI tool that parses Wiktionary XML dump files and extracts structured dictionary data (glosses, translations, pronunc…	76	1251	active
PyDMD/PyDMD PyDMD is a Python package implementing Dynamic Mode Decomposition (DMD) and its many variants (mrDMD, HODMD, etc.) for extracting spatiotem…	60	1251	active
ModelCloud/GPTQModel GPTQModel is a production-ready Python toolkit for quantizing (compressing) large language models using GPTQ, AWQ, and related methods, wit…	91	1248	active
sentinel-hub/eo-learn eo-learn is a collection of open-source Python packages for accessing and processing spatio-temporal satellite imagery, built around modula…	51	1247	active
withoutbg/withoutbg-python A Python SDK (pip install withoutbg) for removing image backgrounds, offering a free local open-weights ONNX model and an optional paid clo…	80	1246	active
thu-ml/Motus Motus is the official implementation of a unified latent action world model for robotics, combining a video generation model, a vision-lang…	43	1246	active
lpiccinelli-eth/UniDepth UniDepth is a Python library and research codebase for universal monocular metric depth estimation from single images, based on CVPR 2024 a…	35	1246	active
automl/SMAC3 SMAC3 is a Python library for Bayesian Optimization used to tune hyperparameters of machine learning algorithms and configure arbitrary alg…	81	1244	active
unum-cloud/UForm UForm is a compact multimodal AI library providing tiny image-text embedding models (64-768 dimensions, Matryoshka-style) and small generat…	55	1244	active
fpgaminer/joycaption JoyCaption is an open, free, and uncensored image captioning Visual Language Model (VLM) with released weights and training scripts. It gen…	53	1244	active
LTH14/fractalgen A PyTorch implementation of Fractal Generative Models (FractalGen), enabling pixel-by-pixel high-resolution image generation. It includes p…	24	1244	active
IDSIA/sacred Sacred is a Python library for configuring, organizing, logging, and reproducing machine learning and scientific experiments. It provides c…	44	4374	maintenance
abewley/sort SORT is a barebones Python implementation of a simple online and realtime multiple object tracking algorithm for 2D video sequences, based …	32	4373	maintenance
HJYao00/Mulberry Mulberry is a research implementation of an o1-like multimodal large language model (MLLM) that performs step-by-step reasoning and reflect…	49	1243	active
dmlc/dlpack DLPack is a minimal, stable open in-memory tensor structure defined as a single C ABI header, enabling zero-copy tensor exchange between de…	66	1242	stable
cvg/depthsplat DepthSplat is a PyTorch research library implementing a CVPR 2025 model that connects Gaussian splatting with single/multi-view depth estim…	56	1242	active
Roblox/cube Cube is Roblox's open-source family of foundation models for 3D intelligence, including text-to-3D shape generation and part-controllable m…	58	1241	active
wannesm/dtaidistance A Python library for computing time series distances, most notably Dynamic Time Warping (DTW), with a fast C implementation exposed via Cyt…	64	1240	stable
marian42/mesh_to_sdf A Python library that computes approximate signed distance fields (SDFs) for arbitrary triangle meshes, including non-watertight, self-inte…	32	1240	stable
ZHKKKe/MODNet MODNet is a deep learning model for real-time portrait matting (background removal) that requires only an RGB image as input, with no trima…	32	4355	maintenance
apchenstu/TensoRF TensoRF is a PyTorch implementation of the ECCV 2022 paper 'TensoRF: Tensorial Radiance Fields', which models and reconstructs radiance fie…	44	1239	stable
sh-lee-prml/HierSpeechpp Official PyTorch implementation of HierSpeech++, a fast zero-shot speech synthesizer for text-to-speech and voice conversion based on hiera…	28	1238	active
pytorch/serve TorchServe is a flexible, production-ready model server for serving, optimizing, and scaling PyTorch models over HTTP with support for CPU,…	10	4346	maintenance
X-Square-Robot/wall-x Wall-X is the open-source training and inference stack for X Square Robot's WALL series of embodied foundation models (VLAs) for general-pu…	61	1236	active
FenTechSolutions/CausalDiscoveryToolbox A Python library for causal inference in graphs and pairwise settings, implementing many algorithms for graph structure recovery from obser…	53	1236	active
open-mmlab/playground OpenMMLab Playground is a central hub collecting and showcasing community projects that extend OpenMMLab libraries with Segment Anything Mo…	30	1236	active
meta-pytorch/attention-gym Attention Gym is a collection of tools, examples, and reference implementations for working with PyTorch's FlexAttention API. It provides a…	91	1234	active
xtreme1-io/xtreme1 Xtreme1 is an open-source, self-hosted data labeling and annotation platform for multimodal training data, supporting images, 3D LiDAR poin…	62	1234	active
facebookresearch/home-robot HomeRobot is an open-source robotics stack from Meta AI for mobile manipulation tasks on low-cost hardware like the Hello Robot Stretch. It…	23	1234	active
facioquo/stock-indicators-dotnet A C# NuGet library (FacioQuo.Stock.Indicators, formerly Skender.Stock.Indicators) that transforms historical OHLCV price bars from equities…	92	1233	active
Equim-chan/mjai-reviewer A Rust CLI tool that reviews riichi mahjong game logs using mjai-compatible AI engines such as Mortal and akochan. It fetches logs from Ten…	26	1233	active
Acly/comfyui-inpaint-nodes A set of custom nodes for ComfyUI that improve image inpainting and outpainting workflows. It integrates the Fooocus inpaint model for SDXL…	64	1232	active
CSSLab/maia-chess Maia is a collection of human-like neural network chess engines trained on millions of human games, targeting skill levels from ELO 1100 to…	61	1228	active
python-adaptive/adaptive Adaptive is an open-source Python library for parallel active learning of mathematical functions. It intelligently selects the most informa…	91	1227	active
ai-dynamo/nixl NVIDIA Inference Xfer Library (NIXL) is a C++/Python library that accelerates point-to-point communication in AI inference frameworks like …	88	1227	active
NVIDIA/BigVGAN BigVGAN is NVIDIA's official PyTorch implementation of a universal neural vocoder (ICLR 2023) that generates high-fidelity raw audio wavefo…	23	1227	stable
alibaba/x-deeplearning X-DeepLearning (XDL) is an industrial deep learning framework from Alibaba optimized for high-dimension sparse data scenarios such as adver…	23	4304	maintenance
Bolin97/GongBU GongBU is a self-hosted, no-code web platform for fine-tuning, evaluating, and deploying large language models, built on Transformers and P…	52	1225	active
run-house/kubetorch Kubetorch is a Python library that lets you distribute and run ML workloads (training, inference, data processing) on Kubernetes directly f…	83	1224	active
yeyupiaoling/Whisper-Finetune A toolkit for fine-tuning OpenAI's Whisper speech recognition models using LoRA, supporting training with or without timestamps and even wi…	66	1223	active
ElectricAlexis/NotaGen NotaGen is a symbolic music generation model that produces high-quality classical sheet music using LLM-style training paradigms: pre-train…	32	1223	active
mrousavy/react-native-fast-tflite A high-performance TensorFlow Lite library for React Native built on Nitro Modules, using the low-level C/C++ TFLite core API with zero-cop…	84	1222	active
eduardoleao052/js-pytorch JS-PyTorch is a deep learning library for JavaScript that closely mirrors PyTorch's syntax, providing tensor operations, automatic differen…	16	1222	active
fastgs/FastGS FastGS is a general acceleration framework for 3D Gaussian Splatting that trains scenes in roughly 100 seconds using multi-view consistent …	49	1221	active
SciML/NeuralPDE.jl NeuralPDE.jl is a Julia library of physics-informed neural network (PINN) solvers for ordinary, stochastic, and partial differential equati…	99	1220	active
MotrixLab/SMPLer-X Official code for SMPLer-X, a family of foundation models for expressive human pose and shape estimation (EHPS) that unifies body, hand, an…	59	1220	stable
Project-MONAI/research-contributions A collection of peer-reviewed research prototype implementations built on the MONAI framework for medical imaging AI. It serves as a fast-t…	43	1220	active
gabrielchua/RAGxplorer RAGxplorer is a Python library and hosted Streamlit app for visualizing Retrieval Augmented Generation pipelines. It chunks PDFs, embeds th…	16	1220	active
zkonduit/ezkl EZKL is a Rust-based library and command-line tool that converts deep learning models and arbitrary computational graphs (exported as ONNX)…	75	1219	active
Aratako/Irodori-TTS Irodori-TTS is a Flow Matching-based text-to-speech model with training and inference code, built on a Rectified Flow Diffusion Transformer…	58	1219	active
microsoft/malmo Project Malmo is a platform for artificial intelligence experimentation and research built on top of Minecraft, providing a gym-like enviro…	10	4270	maintenance
lucidrains/perceiver-pytorch A PyTorch implementation of the Perceiver architecture (General Perception with Iterative Attention) and its follow-up Perceiver IO. It pro…	62	1217	active
kijai/ComfyUI-segment-anything-2 A set of ComfyUI custom nodes that bring Meta's Segment Anything 2 (SAM2) models into ComfyUI workflows for promptable image and video segm…	43	1214	active
chelsea0x3b/cudarc cudarc is a safe, minimal Rust wrapper around the NVIDIA CUDA toolkit, exposing the CUDA driver API plus libraries such as NVRTC, cuBLAS/cu…	98	1213	active
InternScience/GraphGen GraphGen is a Python framework for knowledge-graph-guided synthetic data generation for LLM training. It builds fine-grained knowledge grap…	59	1212	active
hasktorch/hasktorch Hasktorch is a Haskell library for tensor math and neural networks, built on bindings to the C++ libtorch libraries that power PyTorch. It …	76	1211	active
OpenTSLM/OpenTSLM OpenTSLM is a family of Time-Series Language Models that integrate time series as a native modality into pretrained LLMs (Llama, Gemma), en…	61	1211	active
GuitarML/NeuralPi NeuralPi is a DIY Raspberry Pi 4-based guitar pedal that uses neural networks to emulate real amplifiers and distortion/overdrive pedals in…	23	1211	active
MyoHub/myosuite MyoSuite is a collection of musculoskeletal environments and tasks simulated with the MuJoCo physics engine and wrapped in the OpenAI gym A…	93	1210	active
Zefan-Cai/R-KV R-KV is a training-free, redundancy-aware KV cache compression method for reasoning LLMs, discarding repetitive tokens on-the-fly during de…	60	1209	active
datawhalechina/torch-rechub Torch-RecHub is a lightweight PyTorch framework for building recommendation system models with 30+ out-of-the-box algorithms covering ranki…	94	1207	active
ardha27/AI-Song-Cover-RVC A collection of Google Colab and Kaggle notebooks that form an all-in-one toolkit for creating AI song covers with RVC (Retrieval-based Voi…	68	1207	active