function: machine-learning

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

Product	Health v2	Stars	Maturity
facebookresearch/coconut Official PyTorch implementation of Coconut, a method for training large language models to reason in a continuous latent space instead of e…	61	1689	active
elixir-nx/axon Axon is a neural network library for Elixir built on top of Nx numerical definitions. It provides a functional API, a high-level model crea…	79	1688	active
kijai/ComfyUI-FramePackWrapper A ComfyUI custom node wrapper for FramePack, enabling next-frame video generation from images using HunyuanVideo-based diffusion models. It…	47	1686	active
sedthh/pyxelate Pyxelate is a Python library and CLI tool that converts images into 8-bit pixel art by downsampling and learning a reduced color palette. I…	51	1685	active
ckiplab/ckiptagger CkipTagger is a Python library implementing neural CKIP-style Chinese NLP tools for word segmentation, part-of-speech tagging, and named en…	36	1684	stable
trailofbits/buttercup Buttercup is a Cyber Reasoning System (CRS) developed by Trail of Bits for the DARPA AI Cyber Challenge that automatically finds and patche…	56	1683	active
williamyang1991/DualStyleGAN Official PyTorch implementation of DualStyleGAN, a CVPR 2022 model for exemplar-based high-resolution (1024px) portrait style transfer. It …	32	1683	stable
absadiki/subsai Subs AI is a subtitles generation tool available as a Web-UI, CLI, and Python package, powered by OpenAI's Whisper and its variants (faster…	66	1682	active
chungmin99/pyroki PyRoki is a modular Python toolkit for robot kinematic optimization built on JAX, providing differentiable forward kinematics from URDFs, c…	53	1682	active
scverse/scvi-tools scvi-tools is a Python library for deep probabilistic modeling and analysis of single-cell and spatial omics data, built on PyTorch, PyTorc…	93	1681	stable
birdnet-team/BirdNET-Analyzer BirdNET-Analyzer is an open-source Python tool from the Cornell Lab of Ornithology that uses deep learning to identify bird species from au…	80	1681	active
intelligent-machine-learning/dlrover DLRover is an automatic distributed deep learning system that manages training of large AI models on Kubernetes and Ray clusters. It provid…	79	1680	active
EnzymeAD/Enzyme Enzyme is a high-performance automatic differentiation plugin for LLVM and MLIR that computes derivatives and gradients of arbitrary existi…	95	1679	active
explosion/sense2vec A Python library implementing sense2vec, contextually-keyed word vectors that learn embeddings for multi-word phrases and part-of-speech-di…	56	1679	stable
takuseno/d3rlpy d3rlpy is a Python library for offline and online deep reinforcement learning built on PyTorch, offering state-of-the-art algorithms throug…	52	1679	active
py-why/causal-learn causal-learn is a Python library for causal discovery, implementing classical and state-of-the-art algorithms to recover causal structure f…	89	1677	active
robocasa/robocasa RoboCasa is a large-scale simulation framework for training and benchmarking generally capable robots on everyday tasks, focused on kitchen…	81	1677	active
pytorch-tabular/pytorch_tabular PyTorch Tabular is a Python library providing a unified, high-level interface for deep learning models on tabular data, built on PyTorch an…	67	1677	active
PRML/PRMLT A self-contained MATLAB toolbox implementing the machine learning algorithms from Bishop's textbook 'Pattern Recognition and Machine Learni…	23	6201	maintenance
franciszzj/Leffa Leffa is a diffusion-based framework for controllable person image generation, supporting virtual try-on and pose transfer via a regulariza…	40	1672	active
google/vizier Open Source Vizier is a Python-based service and library for black-box and hyperparameter optimization, based on Google's internal Vizier t…	79	1671	active
shubham-goel/4D-Humans 4DHumans is a Python research codebase implementing HMR 2.0, a transformer-based model for 3D human mesh recovery from single images, plus …	59	1671	active
jd-opensource/JoyAI-Video-Edit JoyAI-Video-Edit is a real-time, instruction-guided video editing system that applies natural-language edits to live or uploaded video stre…	57	1671	active
ZrrSkywalker/Personalize-SAM PerSAM is the official implementation of 'Personalize Segment Anything Model with One Shot', which customizes the Segment Anything Model (S…	29	1671	active
zihangdai/xlnet XLNet is a generalized autoregressive pretraining method for language representation learning, built on Transformer-XL. This repo provides …	32	6188	maintenance
tkarras/progressive_growing_of_gans Official TensorFlow implementation of the ICLR 2018 NVIDIA paper 'Progressive Growing of GANs', which trains generators and discriminators …	32	6179	maintenance
Gen-Verse/MMaDA MMaDA is an open-source family of multimodal large diffusion language models that unify textual reasoning, multimodal understanding, and te…	49	1668	active
nwojke/deep_sort A Python implementation of Deep SORT, a multi-object tracking algorithm that extends SORT with a deep appearance descriptor for robust data…	36	6169	maintenance
bytedance/Sa2VA Sa2VA is a family of research models and codebases from ByteDance that combine SAM-2 with multimodal LLMs for pixel-level grounded understa…	70	1666	active
JiuhaiChen/BLIP3o Official implementation of the BLIP3o-Series, a unified autoregressive-plus-diffusion model for text-to-image generation and editing. It co…	44	1666	active
elixir-nx/bumblebee Bumblebee is an Elixir library providing pre-trained neural network models built on Axon, with integration for downloading models from Hugg…	89	1662	active
mit-han-lab/torchquantum TorchQuantum is a PyTorch-based framework for quantum computing simulation, supporting statevector and pulse-level simulation on GPUs with …	64	1662	active
meyda/meyda Meyda is a JavaScript audio feature extraction library that computes standard signal features like RMS, spectral centroid, and spectral fla…	23	1662	stable
lotus-data/lotus LOTUS is a Python library providing a Pandas-like API of LLM-powered semantic operators (map, filter, extract, aggregate, top-k) for bulk p…	83	1661	active
hecomi/uLipSync uLipSync is a Unity plug-in that performs lip-sync by analyzing audio in real time or pre-baked, extracting MFCC features via the Job Syste…	50	1660	active
thunil/TecoGAN TecoGAN is the official source code for a temporally coherent GAN for video super-resolution, published at SIGGRAPH/ACM TOG. It includes in…	32	6141	maintenance
thtrieu/darkflow Darkflow is a Python library that translates Darknet's YOLO neural network definitions to TensorFlow, enabling real-time object detection a…	32	6139	maintenance
MgArcher/Text_select_captcha A PyTorch-based deep learning system that recognizes click-based (text-select) CAPTCHAs by detecting and ordering Chinese character positio…	69	1656	active
chineseocr A Python OCR toolkit that combines YOLO3-based text detection with CRNN/Dense recognition for Chinese and English text in natural scene ima…	32	6123	maintenance
tableau/TabPy TabPy is the Tableau Python Server, an Analytics Extensions implementation that lets Tableau users execute Python scripts and saved functio…	95	1654	active
Data-Centric-AI-Community/fg-data-synthetic A Python library for generating synthetic tabular and time-series data using state-of-the-art generative models such as GANs (including CTG…	72	1654	active
ZJU4HealthCare/HealthGPT HealthGPT is a medical multimodal large language model family unifying medical image comprehension and generation via heterogeneous knowled…	63	1654	active
mhamilton723/FeatUp FeatUp is a model-agnostic framework that upsamples the spatial resolution of deep neural network features by 16-32x without changing their…	16	1654	active
taki0112/UGATIT Official TensorFlow implementation of U-GAT-IT, an unsupervised image-to-image translation model using attention modules and adaptive layer…	32	6116	maintenance
maziarraissi/PINNs The original reference implementation of Physics-Informed Neural Networks (PINNs), which train neural networks to solve and discover nonlin…	62	6111	maintenance
cubiq/ComfyUI_IPAdapter_plus A ComfyUI custom node implementation of IPAdapter models for image-to-image conditioning in Stable Diffusion workflows. It transfers the su…	35	6110	maintenance
lightly-ai/lightly-train LightlyTrain is a Python framework for training computer vision models, covering pretraining of vision foundation models (DINOv2/v3) on unl…	85	1652	active
Stability-AI/stable-virtual-camera Stable Virtual Camera (SEVA) is a generalist diffusion model for novel view synthesis that generates 3D-consistent views of a scene from an…	54	1652	active
skrub-data/skrub skrub is a Python library that facilitates machine learning with dataframes, providing preprocessing, encoding, and wrangling tools for tab…	93	1650	active
sunlabuiuc/PyHealth PyHealth is an open-source Python toolkit for clinical deep learning, unifying healthcare datasets (EHRs, physiological signals, medical im…	91	1650	active
bytedance/DreamO DreamO is the official PyTorch implementation of a unified framework for image customization, built on FLUX diffusion models. It supports t…	35	1649	active
XueZeyue/DanceGRPO Official implementation of DanceGRPO, a framework applying Group Relative Policy Optimization (GRPO) to fine-tune visual generation models …	40	1648	active
vllm-project/vllm-metal A community-maintained hardware plugin that enables vLLM to run LLM inference on Apple Silicon Macs using MLX as the primary compute backen…	79	1647	active
Anemll/Anemll ANEMLL is an open-source Python library and toolchain for converting and running large language models on the Apple Neural Engine (ANE) via…	61	1647	active
JIA-Lab-research/ControlNeXt ControlNeXt is the official implementation of a controllable generation method for images and videos, built on Stable Diffusion XL, Stable …	24	1646	active
airockchip/rknn-llm RKLLM is Rockchip's software stack for converting, quantizing, and running large language models on Rockchip NPU chips such as RK3588 and R…	79	1644	active
tenstorrent/tt-metal TT-Metal is Tenstorrent's open-source software stack containing TT-NN, a Python and C++ neural network operator library, and TT-Metalium, a…	97	1640	active
CoinCheung/BiSeNet A PyTorch implementation of the BiSeNet V1 and V2 real-time semantic segmentation models, with pretrained weights for Cityscapes, COCO-Stuf…	57	1638	active
yenchenlin/nerf-pytorch A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the original paper's results for synthesizing novel views of comp…	32	6044	maintenance
RQLuo/MixTeX-Latex-OCR MixTeX is a multimodal OCR application that recognizes LaTeX formulas, tables, and mixed Chinese/English text from images, running entirely…	22	1637	active
luanfujun/deep-painterly-harmonization Research code implementing the 'Deep Painterly Harmonization' algorithm, which seamlessly blends a pasted object into a painting's style us…	32	6042	maintenance
symforce-org/symforce SymForce is a Python/C++ library for fast symbolic computation, code generation, and tangent-space nonlinear optimization aimed at robotics…	94	1636	active
opendilab/LightZero LightZero is a lightweight, open-source Python toolkit that unifies Monte Carlo Tree Search (MCTS) with deep reinforcement learning, implem…	67	1636	active
datamade/usaddress usaddress is a Python library that parses unstructured United States address strings into labeled address components using a conditional ra…	48	1636	active
TonyJiangWJ/Ant-Forest An AutoJS-based automation script for Android that automatically collects energy in Alipay's Ant Forest feature. It uses image recognition …	58	1635	active
NVIDIAGameWorks/rtx-remix NVIDIA RTX Remix is an open-source platform for remastering classic DirectX 8/9 games with RTX path tracing, DLSS, and AI-upgraded textures…	81	1631	active
ShengranHu/ADAS ADAS (Automated Design of Agentic Systems) is the official research code for the ICLR 2025 paper introducing Meta Agent Search, where a met…	24	1631	active
Picsart-AI-Research/StreamingT2V StreamingT2V is a research codebase (CVPR 2025) implementing an autoregressive technique that turns short text-to-video diffusion models li…	31	1630	active
facebookresearch/DomainBed DomainBed is a PyTorch suite containing benchmark datasets and algorithms for domain generalization research. It implements many algorithms…	10	1630	active
Lucchetto/SuperImage SuperImage is an Android application that upscales low-resolution images using a Real-ESRGAN neural network running on-device via the MNN d…	10	1630	active
yoshitomo-matsubara/torchdistill torchdistill is a modular, configuration-driven PyTorch framework for knowledge distillation and general deep learning experiments, requiri…	86	1629	active
AnswerDotAI/rerankers A lightweight, dependency-free Python library providing a unified API for using all common reranking and cross-encoder models. It lets deve…	42	1629	active
Fantasy-AMAP/fantasy-talking FantasyTalking is a research codebase and model for generating realistic talking portrait videos from a single image and an audio clip, bui…	48	1628	active
snorkel-team/snorkel Snorkel is a Python library for programmatically building and managing training data using weak supervision, letting users write labeling f…	62	6001	maintenance
InterDigitalInc/CompressAI CompressAI is a PyTorch library and evaluation platform for end-to-end learned data compression research. It provides custom layers, entrop…	73	1627	active
apple/ml-mobileclip Official implementation of Apple's MobileCLIP and MobileCLIP2, fast image-text contrastive (CLIP-style) models optimized for mobile latency…	60	1627	active
scikit-learn-contrib/boruta_py A Python implementation of the Boruta all-relevant feature selection method, built on scikit-learn. It identifies all features carrying pre…	46	1627	active
ZiqiaoPeng/SyncTalk SyncTalk is the official PyTorch implementation of a CVPR 2024 paper that synthesizes speech-driven, synchronized talking head videos using…	46	1626	active
guochengqian/Magic123 Magic123 is the official PyTorch implementation of an ICLR 2024 paper that generates high-quality textured 3D meshes from a single unposed …	90	1623	stable
HumanSignal/Adala Adala is a Python framework for building autonomous data labeling agents powered by LLMs. Agents learn skills iteratively from a ground tru…	62	1623	active
gomlx/gomlx GoMLX is an accelerated machine learning and math framework for Go, comparable to PyTorch/JAX/TensorFlow. It offers differentiable operator…	95	1621	active
mlcommons/inference Reference implementations of the MLPerf Inference benchmark suite, the industry-standard benchmarks for measuring how fast ML systems run m…	83	1621	active
pavlin-policar/openTSNE openTSNE is a modular Python library implementing t-Distributed Stochastic Neighbor Embedding (t-SNE) for visualizing high-dimensional data…	66	1621	stable
bowang-lab/scGPT scGPT is a foundation model for single-cell multi-omics built with a generative transformer, providing pretrained checkpoints and a Python …	67	1620	active
jeffffffli/HybrIK HybrIK is the official PyTorch implementation of a hybrid analytical-neural inverse kinematics method for 3D human pose and shape estimatio…	23	1618	stable
houseofsecrets/SdPaint A Python desktop application that provides a painting canvas where each stroke is sent to a Stable Diffusion (automatic1111) API and the ge…	20	1618	active
csinva/imodels A Python package providing scikit-learn-compatible implementations of interpretable machine-learning models such as rule lists, RuleFit, an…	85	1617	active
vinhhien112/img2obj img2obj is a Codex plugin that turns an attached object image into a code-only, animation-ready procedural Three.js model. It guides the AI…	56	1615	active
OminousIndustries/PhoneDriver PhoneDriver is a Python-based mobile automation agent that uses Qwen3-VL vision-language models to visually understand and control Android …	38	1614	active
PaddlePaddle/PaddleSlim PaddleSlim is an open-source library built on PaddlePaddle for deep learning model compression and architecture search. It provides low-bit…	50	1612	active
open-gigaai/giga-world-0 GigaWorld-0 is a unified world model framework that acts as a data engine for Vision-Language-Action (VLA) learning in embodied AI. It comb…	41	1612	active
PKU-Alignment/safe-rlhf Beaver is a modular open-source framework from Peking University for training large language models with SFT, RLHF, and Safe RLHF (constrai…	53	1611	active
gorgonia/gorgonia Gorgonia is a Go library for machine learning that lets you define and evaluate mathematical equations over multidimensional arrays using a…	23	5929	maintenance
yossdotpro/removerized Removerized is an open-source, browser-based AI image toolkit for background removal and image upscaling, running entirely client-side via …	81	1610	active
Robbyant/lingbot-depth LingBot-Depth is a PyTorch-based model and toolkit for masked depth modeling that transforms incomplete, noisy depth sensor data into metri…	56	1610	active
chainer/chainer Chainer is a Python-based deep learning framework that pioneered the define-by-run approach with dynamic computational graphs and automatic…	23	5924	maintenance
dmlc/gluon-cv GluonCV is a deep learning toolkit providing state-of-the-art computer vision model implementations with 170+ pre-trained models. It suppor…	23	5916	maintenance
OpenGVLab/LLaMA-Adapter Official implementation of LLaMA-Adapter and its V2 successor, a parameter-efficient fine-tuning method that adapts LLaMA models to follow …	21	5914	maintenance
clusterzx/paperless-ai A self-hosted web application that automatically analyzes, tags, and classifies documents in Paperless-ngx using LLMs via OpenAI-compatible…	73	5910	maintenance
pypose/pypose PyPose is a PyTorch-based Python library for differentiable robotics on manifolds, combining deep perceptual models with physics-based opti…	91	1605	active