function: machine-learning

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

Product	Health v2	Stars	Maturity
PRIME-RL/TTRL TTRL is an open-source implementation of Test-Time Reinforcement Learning, a method for training LLMs with RL on unlabeled data using major…	51	1120	active
blackjax-devs/blackjax BlackJAX is a library of fast, modular MCMC samplers (NUTS, HMC, and more) built on JAX, running on CPU, GPU, and TPU. It is not a full pro…	89	1119	active
princeton-vl/RAFT-Stereo RAFT-Stereo is a PyTorch implementation of a deep learning model for stereo matching that estimates disparity maps from stereo image pairs …	75	1119	stable
PaddlePaddle/PaddleHelix PaddleHelix is a bio-computing platform built on PaddlePaddle featuring large-scale representation learning and multi-task deep learning fo…	57	1119	active
alibaba-damo-academy/RynnVLA-002 RynnVLA-002 is a unified autoregressive Vision-Language-Action and world model that generates robot actions from text and image observation…	43	1119	active
caiyuanhao1998/MST A Python toolbox for spectral compressive imaging reconstruction that implements over 15 algorithms including MST, CST, DAUHST, BiSCI, HDNe…	53	1118	active
yangxue0827/RotationDetection AlphaRotate is a TensorFlow-based benchmark and toolbox for rotated (oriented) object detection, implementing detectors such as R2CNN, Reti…	23	1118	active
Lasagne/Lasagne Lasagne is a lightweight Python library for building and training neural networks on top of Theano. It supports feed-forward, convolutional…	23	3857	maintenance
HarryR/z80ai Z80-μLM is a 2-bit quantized micro language model designed to run on 8-bit Z80 processors with only 64KB of RAM, exportable as ~40KB CP/M .…	51	1117	active
datadreamer-dev/DataDreamer DataDreamer is an open-source Python library for prompting LLMs, generating synthetic datasets, and training or aligning models in reproduc…	33	1117	active
microsoft/onnxruntime-genai ONNX Runtime GenAI is a C++ library with Python, C#, C/C++, and Java APIs for running generative AI models (LLMs, Whisper, vision-language …	93	1116	active
HITsz-TMG/Uni-MoE Uni-MoE is a family of open-source Mixture-of-Experts (MoE) based omnimodal large language models that understand and generate across text,…	68	1116	active
NTMC-Community/MatchZoo MatchZoo is a Python toolkit for designing, comparing, and sharing deep text matching models. It provides a unified data pipeline, pre-buil…	23	3849	maintenance
yahoo/TensorFlowOnSpark TensorFlowOnSpark is a Python library that lets existing TensorFlow programs run distributed training and inference on Apache Spark and Had…	23	3845	maintenance
openai/improved-diffusion The official codebase for OpenAI's Improved Denoising Diffusion Probabilistic Models paper, providing a Python package for training and sam…	32	3844	maintenance
storyicon/comfyui_segment_anything A ComfyUI custom node that combines GroundingDINO and Segment Anything (SAM) to segment any element in an image using semantic text prompts…	27	1113	active
NVIDIA/earth2studio Earth2Studio is a Python deep-learning framework from NVIDIA for building, exploring, and deploying AI-driven weather and climate workflows…	88	1112	active
AdaptiveMotorControlLab/CEBRA CEBRA is a Python library for self-supervised learning of consistent latent embeddings from high-dimensional time-series recordings, using …	86	1111	active
luxkun/ReGoap ReGoap is a generic C# GOAP (Goal Oriented Action Planning) library with adapters for Unity and Godot. It provides an engine-agnostic plann…	58	1111	active
jina-ai/discoart DiscoArt is a Python library that wraps Disco Diffusion (CLIP-guided diffusion) so generative artists and developers can create AI artworks…	23	3826	maintenance
baofff/U-ViT U-ViT is the official PyTorch implementation of a ViT-based backbone architecture for diffusion models from the CVPR 2023 paper 'All are Wo…	32	1110	stable
unitreerobotics/unifolm-world-model-action UnifoLM-WMA-0 is Unitree's open-source world-model-action framework for general-purpose robot learning across multiple robotic embodiments.…	50	1109	active
princeton-vl/DPVO DPVO is a deep learning-based visual odometry and SLAM system that estimates camera trajectories from video or image sequences using patch-…	32	1108	active
THU-MIG/RepViT Official PyTorch implementation of RepViT, a family of lightweight CNNs designed by integrating efficient ViT architectural designs into Mo…	19	1108	stable
gxtrobot/bustag Bustag is a self-hosted web application that periodically scrapes new media title (番号) listings, lets users tag items as liked or disliked,…	23	3814	maintenance
open-mmlab/PowerPaint PowerPaint is a versatile image inpainting model (ECCV 2024) built on diffusion models that handles text-guided object insertion, object re…	70	1107	active
prometheus-eval/prometheus-eval Prometheus-Eval is a Python library for evaluating LLM generation outputs using the Prometheus family of open evaluator models and GPT-4 as…	23	1107	active
grf-labs/grf GRF is an R package (with C++ core) for forest-based statistical estimation and inference, providing non-parametric methods for heterogeneo…	68	1106	stable
szymanowiczs/splatter-image Official PyTorch implementation of 'Splatter Image: Ultra-Fast Single-View 3D Reconstruction' (CVPR 2024), which uses an image-to-image net…	26	1106	active
google-research/multinerf Google Research's official code release for three NeRF papers: Mip-NeRF 360, Ref-NeRF, and RawNeRF, written in JAX. It trains neural radian…	10	3808	maintenance
rusty1s/pytorch_sparse A PyTorch extension library providing optimized sparse matrix operations (coalesce, transpose, sparse-dense and sparse-sparse multiplicatio…	61	1104	active
TencentARC/T2I-Adapter Official implementation of T2I-Adapter, lightweight adapter models that add controllable conditioning (sketch, canny, lineart, depth, pose)…	31	3801	maintenance
PharMolix/OpenBioMed OpenBioMed is an open-source toolkit and agent platform for biomedicine and life science, offering multimodal models like BioMedGPT-Mol, 45…	72	1103	active
yangheng95/PyABSA PyABSA is a PyTorch-based library providing state-of-the-art models for aspect-based sentiment analysis, including aspect term extraction, …	67	1102	active
MIT-SPARK/VGGT-SLAM VGGT-SLAM is a dense RGB SLAM system that performs real-time feed-forward 3D scene reconstruction, optimizing on the SL(4) manifold using t…	59	1102	active
yzslab/gaussian-splatting-lightning A PyTorch Lightning implementation of 3D Gaussian Splatting with many derived algorithms (Mip-Splatting, LightGaussian, 2DGS, deformable Ga…	58	1101	active
MoonshotAI/MoonEP MoonEP is an Expert Parallelism communication library for Mixture-of-Experts training that keeps token loads perfectly balanced across rank…	56	1101	active
zai-org/CogView4 CogView4, CogView3-Plus and CogView3 are open-source text-to-image generation models from Zhipu AI, with CogView4 being a 6B-parameter DiT-…	28	1101	active
naturomics/CapsNet-Tensorflow A TensorFlow implementation of CapsNet (Capsule Networks) based on Geoffrey Hinton's paper 'Dynamic Routing Between Capsules'. It supports …	32	3786	maintenance
yongliang-wu/DFT DFT (Dynamic Fine-Tuning) is the official implementation of an ICLR 2026 paper that improves Supervised Fine-Tuning of LLMs by dynamically …	61	1100	active
lucidrains/stylegan2-pytorch A simple PyTorch implementation of StyleGAN2, a state-of-the-art generative adversarial network, trainable entirely from the command line w…	23	3783	maintenance
Gen-Verse/LatentMAS LatentMAS is a training-free multi-agent reasoning framework that lets LLM agents collaborate in continuous latent space by passing hidden-…	56	1099	active
yandex-research/tabm TabM is a PyTorch-based deep learning model for tabular data that efficiently imitates an ensemble of MLPs through parameter-efficient ense…	36	1099	active
luchris429/purejaxrl PureJaxRL provides end-to-end reinforcement learning training pipelines implemented entirely in JAX, including environments, enabling massi…	30	1099	active
fraunhoferportugal/tsfel TSFEL is an open-source Python library for extracting features from time series signals across statistical, temporal, spectral, and fractal…	53	1098	active
LAMDA-CL/PyCIL PyCIL is a PyTorch-based Python toolbox for class-incremental learning, implementing the largest collection of CIL methods for reproducible…	52	1098	active
apache/systemds Apache SystemDS is an open-source machine learning system covering the end-to-end data science lifecycle, from data cleaning and feature en…	87	1097	active
HswAI2026/JuZhou-V1 JuZhou 1.0 is an ultra-lightweight 0.387B-parameter text-to-image foundation model designed for fully offline, on-device execution on mobil…	54	1097	active
qiskit-community/qiskit-machine-learning Qiskit Machine Learning is an open-source Python library built on the Qiskit SDK that provides building blocks for quantum machine learning…	90	1096	active
boona13/image-extender An open-source Next.js web app for AI image outpainting that extends images in any direction using Gemini models via OpenRouter, with Poiss…	52	1096	active
waymo-research/waymax Waymax is a lightweight, multi-agent autonomous driving simulator built entirely in JAX, based on the Waymo Open Motion Dataset. It support…	49	1096	active
WangLibo1995/GeoSeg GeoSeg is an open-source PyTorch-based semantic segmentation toolbox focused on Vision Transformers for remote sensing imagery, featuring t…	32	1096	active
gunrock/gunrock Gunrock is a CUDA/C++ library for GPU-accelerated graph analytics built on a high-level, data-centric abstraction over vertex and edge fron…	63	1095	active
hkchengrex/Cutie Cutie is a video object segmentation framework with object-level memory reading, a follow-up to XMem offering better consistency, robustnes…	18	1095	active
LujiaJin/One-Pot_Multi-Frame_Denoising Official PyTorch implementation of the One-Pot Multi-frame Denoising (OPD) method published at BMVC 2022 and extended in IJCV. It provides …	60	1094	stable
Eyeline-Labs/Go-with-the-Flow Official implementation of the CVPR 2025 Oral paper 'Go-with-the-Flow', which controls motion in video diffusion models by replacing i.i.d.…	41	1093	active
SCLBD/DeepfakeBench DeepfakeBench is a comprehensive benchmark framework for deepfake detection, providing a unified platform for data management, implementati…	36	1093	active
lizhe00/AnimatableGaussians Official PyTorch implementation of the CVPR 2024 paper 'Animatable Gaussians', which learns pose-dependent Gaussian maps for high-fidelity …	27	1093	active
yandex/YaLM-100B YaLM-100B is a GPT-like pretrained language model with 100 billion parameters, trained by Yandex on English and Russian text using DeepSpee…	32	3757	maintenance
welltop-cn/ComfyUI-TeaCache A ComfyUI plugin integrating TeaCache, a training-free caching method that accelerates diffusion model inference by exploiting output diffe…	35	1092	active
zju3dv/PGSR PGSR is a Python implementation of a planar-based Gaussian Splatting method for efficient, high-fidelity 3D surface reconstruction from mul…	24	1092	active
google-ai-edge/litert-torch LiteRT Torch is a Python library that converts PyTorch models into .tflite format for running with LiteRT on Android, iOS, and IoT devices.…	87	1091	active
lucidrains/tab-transformer-pytorch A PyTorch implementation of the TabTransformer architecture, an attention-based neural network for tabular data, also including the FT Tran…	70	1091	stable
geoopt/geoopt Geoopt is a PyTorch library providing manifold-aware optimizers and manifold tensor containers for Riemannian optimization. It implements R…	59	1091	active
yerfor/Real3DPortrait Official PyTorch implementation of Real3D-Portrait, an ICLR 2024 Spotlight paper for one-shot realistic 3D talking portrait synthesis. It g…	26	1091	active
hanruihua/NeuPAN NeuPAN is a real-time, map-free robot motion planner that maps raw obstacle point data directly to control actions using an end-to-end MPC-…	69	1090	active
localai-org/depth-anything.cpp A from-scratch C++17/ggml port of ByteDance's Depth Anything 2 and 3 models for dependency-free monocular metric depth and camera pose infe…	58	1090	active
flagos-ai/FlagGems FlagGems is a high-performance operator library for large language models written in the Triton language, providing backend-neutral GPU ker…	89	1089	active
Toni-SM/skrl skrl is an open-source modular Reinforcement Learning library written in Python, implemented in PyTorch, JAX, and NVIDIA Warp. It supports …	81	1089	active
mymusise/ChatGLM-Tuning A Python toolkit for fine-tuning the ChatGLM-6B large language model using LoRA (Low-Rank Adaptation) with the Alpaca dataset. It provides …	10	3740	maintenance
rhymes-ai/Aria Aria is an open multimodal native Mixture-of-Experts (MoE) model with 25.3B total parameters (3.9B activated per token) and a 64K multimoda…	23	1087	active
QwenLM/Qwen2.5-Math Qwen2.5-Math is a series of math-specialized large language models (1.5B/7B/72B base and instruct variants plus a 72B reward model) built o…	23	1087	active
FujiwaraChoki/supoclip SupoClip is an open-source, AI-powered video clipping tool that turns long videos like podcasts and streams into vertical 9:16 short clips …	62	1086	active
ndif-team/nnsight nnsight is a Python library for interpreting and intervening on the internals of deep learning models, built on PyTorch. It lets researcher…	88	1085	active
microsoft/TimeCraft TimeCraft is a diffusion model-based framework for generating high-quality synthetic time series data across domains. It uses learned seman…	64	1085	active
DSE-MSU/DeepRobust DeepRobust is a PyTorch library for adversarial robustness research, providing implementations of attack and defense methods for both image…	45	1085	active
mlc-ai/web-stable-diffusion A project that compiles and runs Stable Diffusion text-to-image models entirely inside web browsers using WebGPU and WebAssembly, with no s…	30	3721	maintenance
Alpha-VLLM/Lumina-mGPT-2.0 Lumina-mGPT 2.0 is a stand-alone decoder-only autoregressive model trained from scratch that unifies a broad range of image generation task…	42	1084	active
GraphSAGE Reference implementation of the GraphSAGE algorithm for inductive representation learning on large graphs using stochastic graph convolutio…	32	3719	maintenance
bytedance/byteps BytePS is a high-performance parameter server framework for distributed deep neural network training, supporting TensorFlow, Keras, PyTorch…	10	3717	maintenance
GAIR-NLP/LIMO LIMO is a research project and training framework demonstrating that large language models can achieve strong mathematical reasoning with o…	36	1083	active
nihui/rife-ncnn-vulkan A command-line tool implementing RIFE (Real-Time Intermediate Flow Estimation) for video frame interpolation using the ncnn neural network …	23	1083	active
vladmandic/face-api FaceAPI is a JavaScript library built on TensorFlow/JS that provides AI-powered face detection, rotation tracking, face description and rec…	10	1083	active
kimiyoung/transformer-xl Official implementation of Transformer-XL, an attention-based language model architecture that extends context beyond a fixed length via se…	32	3714	maintenance
agi-brain/xuance XuanCe is an open-source Python library of deep reinforcement learning (DRL) and multi-agent reinforcement learning (MARL) algorithm implem…	95	1082	active
NVlabs/Fast-dLLM NVIDIA's official implementation of Fast-dLLM, a family of training-free and fine-tuning-based acceleration techniques for diffusion-based …	57	1082	active
mlr mlr/mlr3 is a machine learning framework for R providing a unified, object-oriented interface to many learning algorithms. It supports resa…	95	1080	active
GradientHQ/symphony-coord Symphony-Coord is a decentralized multi-agent framework that treats agent selection as an online multi-armed bandit problem, allowing roles…	54	1080	active
charlesq34/pointnet2 Official TensorFlow implementation of PointNet++, a deep neural network that learns hierarchical features on 3D point clouds using metric-s…	32	3700	maintenance
XiaomiMiMo/MiMo-Audio Xiaomi's open-source 7B audio language model family (Base and Instruct) plus a 1.2B RVQ audio tokenizer, pretrained on 100M+ hours of audio…	57	1077	active
minghanqin/LangSplat Official implementation of LangSplat, a CVPR 2024 Highlight paper that constructs a 3D language field using 3D Gaussian Splatting with CLIP…	47	1077	active
Geekgineer/YOLOs-CPP YOLOs-CPP is a production-ready, cross-platform C++ inference library for the YOLO model family (v5 through YOLO26), built on ONNX Runtime …	88	1076	active
louiszengCN/CarlaAir CarlaAir is an open-source simulation infrastructure that combines CARLA's high-fidelity urban driving environments with physics-accurate m…	74	1076	active
autonomousvision/navsim NAVSIM is a data-driven pseudo-simulation framework and benchmark for autonomous vehicle planning, evaluating driving agents non-reactively…	50	1076	active
lasgroup/SDPO SDPO (Self-Distilled Policy Optimization) is a research library implementing a reinforcement learning framework for post-training large lan…	56	1075	active
openai/glide-text2im Official codebase for GLIDE, a diffusion-based text-conditional image synthesis model from OpenAI. It provides pretrained models and notebo…	10	3684	maintenance
facebookresearch/hiera Hiera is the official PyTorch implementation of a hierarchical vision transformer from Meta AI (ICML 2023 Oral). It achieves state-of-the-a…	20	1074	active
cleardusk/3DDFA A PyTorch implementation of the TPAMI 2017 paper 'Face Alignment in Full Pose Range: A 3D Total Solution' (3DDFA). It fits a 3D Morphable M…	23	3677	maintenance
ARM-software/CMSIS-DSP CMSIS-DSP is ARM's optimized embedded compute library providing DSP and math kernels for Cortex-M and Cortex-A processors, including FFT, f…	96	1073	stable
meta-pytorch/monarch Monarch is a distributed programming framework for PyTorch built on scalable actor messaging, with actors grouped into meshes, supervision-…	80	1073	active