domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| HumanCompatibleAI/imitation A Python library providing clean PyTorch implementations of imitation and reward learning algorithms such as Behavioral Cloning, DAgger, GA… | 23 | 1779 | active |
| thu-ml/RoboticsDiffusionTransformer RDT-1B is a 1B-parameter diffusion foundation model for robot bimanual manipulation, pre-trained on 1M+ multi-robot episodes to predict rob… | 50 | 1778 | active |
| DataArcTech/DataArc-SynData-Toolkit DataArc SynData Toolkit is a Python-based synthetic data generation platform for creating customized LLM training data from local corpora, … | 57 | 1777 | active |
| Robbyant/lingbot-vla LingBot-VLA is a Vision-Language-Action foundation model for robot manipulation, pretrained on 20,000 hours of real-world dual-arm robot da… | 54 | 1777 | active |
| microsoft/MarS MarS is a financial market simulation engine powered by a generative foundation model, developed by Microsoft. It provides tools for simula… | 61 | 1776 | active |
| nsfw-filter/nsfw-filter A free, open-source, privacy-focused browser extension that blocks NSFW images using on-device AI classification with TensorFlow.js. It hid… | 85 | 1775 | active |
| xhluca/bm25s BM25S is a pure Python library implementing BM25 lexical ranking using Numpy/Scipy sparse matrices with an optional Numba backend, achievin… | 92 | 1774 | active |
| CalculatedContent/WeightWatcher WeightWatcher is an open-source Python diagnostic tool for analyzing pre-trained deep neural networks without access to training or test da… | 59 | 1772 | active |
| AkaliKong/MiniOneRec MiniOneRec is an open-source framework for generative recommendation built on large language models, covering the full pipeline of semantic… | 54 | 1770 | active |
| GAP-LAB-CUHK-SZ/gaustudio GauStudio is a modular PyTorch framework for 3D Gaussian Splatting (3DGS) research and development, supporting novel view synthesis, 3D rec… | 49 | 1762 | active |
| tsurumeso/vocal-remover A Python command-line tool that uses deep neural networks to separate vocals from instrumental tracks in songs. It includes pretrained mode… | 23 | 1757 | stable |
| SamsungSAILMontreal/TinyRecursiveModels Official codebase for the Tiny Recursive Model (TRM) paper, a recursive reasoning approach where a tiny 7M-parameter neural network iterati… | 10 | 6566 | maintenance |
| beam-cloud/beta9 Beam (beta9) is an open-source serverless runtime for AI workloads, providing GPU inference endpoints, isolated sandboxes for running untru… | 89 | 1755 | active |
| High-Logic/Genie-TTS GENIE is a lightweight Python inference engine for the open-source GPT-SoVITS text-to-speech project, optimized for fast CPU-based speech s… | 61 | 1754 | active |
| webonnx/wonnx Wonnx is a GPU-accelerated ONNX inference runtime written entirely in Rust, built on wgpu and usable natively or in the browser via WebGPU … | 10 | 1754 | active |
| microsoft/mup The `mup` Python package implements Maximal Update Parametrization (μP) for PyTorch models, enabling optimal hyperparameters to remain stab… | 23 | 1753 | stable |
| facebookresearch/metaseq Metaseq is a PyTorch codebase from Meta AI for training and working with large-scale Open Pre-trained Transformers (OPT), forked from fairs… | 10 | 6548 | maintenance |
| google-research/text-to-text-transfer-transformer The official T5 library from Google Research, implementing the text-to-text transfer transformer for NLP tasks like summarization, question… | 64 | 6544 | maintenance |
| octo-models/octo Octo is an open-source generalist robot policy: a transformer-based diffusion policy pretrained on 800k robot trajectories from the Open X-… | 17 | 1751 | active |
| ExtensityAI/symbolicai SymbolicAI is a Python neuro-symbolic framework that blends classical programming with LLM capabilities through composable Symbol primitive… | 94 | 1750 | active |
| parthsarthi03/raptor The official Python implementation of RAPTOR (Recursive Abstractive Processing for Tree-Organized Retrieval), a retrieval-augmented generat… | 25 | 1750 | active |
| codertimo/BERT-pytorch A PyTorch implementation of Google AI's 2018 BERT model with simple, readable code. It provides CLI tools for building vocabulary and pre-t… | 23 | 6527 | maintenance |
| Vchitect/VBench VBench is a comprehensive benchmark suite for evaluating video generative models across hierarchical evaluation dimensions, with prompt sui… | 61 | 1747 | active |
| zhouhaoyi/Informer2020 The official PyTorch implementation of Informer, an efficient Transformer architecture for long sequence time-series forecasting that won t… | 44 | 6516 | maintenance |
| rusty1s/pytorch_scatter A PyTorch extension library providing highly optimized scatter and segment (sparse update) operations with sum, mean, min, and max reductio… | 61 | 1745 | stable |
| datastax/jvector JVector is an embedded approximate nearest neighbor (ANN) search engine for the JVM, implementing a graph index that merges HNSW's hierarch… | 89 | 1741 | active |
| magenta/magenta-realtime Magenta RealTime 2 (MRT2) is an open-weights model for real-time, interactive music generation controllable via text, MIDI, and audio. It s… | 73 | 1739 | active |
| 0xSero/turboquant TurboQuant is a Python library implementing near-optimal KV cache quantization for LLM inference, compressing keys to 3-bit and values to 2… | 48 | 1739 | active |
| openai/consistency_models Official PyTorch implementation of Consistency Models, a generative image model family from OpenAI supporting consistency distillation, con… | 10 | 6486 | maintenance |
| NVIDIA-NeMo/Curator NVIDIA NeMo Curator is a scalable, GPU-accelerated toolkit for preprocessing and curating large text, image, video, and audio datasets for … | 86 | 1736 | active |
| google/automl Google Brain's AutoML repository containing implementations of AutoML models and libraries such as EfficientNet, EfficientNetV2, and Effici… | 10 | 6474 | maintenance |
| decisionintelligence/TFB TFB is a comprehensive and fair benchmarking framework for time series forecasting methods, covering deep learning, machine learning, and s… | 68 | 1734 | active |
| alkaline-ml/pmdarima pmdarima is a Python statistical library that brings R's auto.arima functionality to Python, wrapping statsmodels ARIMA/SARIMAX models behi… | 63 | 1733 | active |
| facebookresearch/multimodal TorchMultimodal is a PyTorch library from Meta for training state-of-the-art multimodal multi-task models at scale, covering both content u… | 77 | 1732 | active |
| quantumlib/OpenFermion OpenFermion is an open-source Python library for compiling and analyzing quantum algorithms that simulate fermionic systems, including quan… | 89 | 1731 | active |
| Zaneham/Booth Booth is an open-source compiler that takes CUDA C, HIP, or Triton kernel source and emits binaries for AMD RDNA 2/3/4 GPUs, NVIDIA PTX, Te… | 77 | 1731 | active |
| cleverhans-lab/cleverhans CleverHans is a Python library for benchmarking machine learning systems' vulnerability to adversarial examples, providing reference implem… | 23 | 6450 | maintenance |
| MultimediaTechLab/YOLO Official MIT-licensed implementation of the YOLOv9, YOLOv7, and YOLO-RD real-time object detection models, including pre-trained weights, t… | 56 | 1723 | active |
| meta-pytorch/tnt TNT (torchtnt) is a lightweight library from Meta providing tools and utilities for PyTorch training workflows. It offers building blocks f… | 76 | 1721 | active |
| bespokelabsai/curator Bespoke Curator is a Python library for bulk LLM inference and scalable synthetic data curation for post-training and structured data extra… | 79 | 1719 | active |
| vdaas/vald Vald is a highly scalable, distributed approximate nearest neighbor (ANN) dense vector search engine built on cloud-native architecture and… | 91 | 1718 | active |
| VAST-AI-Research/UniRig UniRig is the official implementation of a SIGGRAPH 2025 framework that automatically rigs 3D models by predicting skeletons and skinning w… | 58 | 1717 | active |
| MaliosDark/wifi-3d-fusion WiFi-3D-Fusion is an open-source research project that estimates 3D human pose from WiFi CSI (Channel State Information) signals using deep… | 27 | 1716 | active |
| cocodataset/cocoapi Official API for the COCO (Common Objects in Context) dataset, providing Matlab, Python, and Lua interfaces to load, parse, and visualize C… | 32 | 6384 | maintenance |
| kingoflolz/mesh-transformer-jax A JAX/Haiku library implementing model-parallel training and inference of transformer models using xmap/pjit operators, similar to Megatron… | 32 | 6380 | maintenance |
| google-deepmind/mujoco_mpc MuJoCo MPC (MJPC) is an interactive application and C++ software framework from Google DeepMind for real-time predictive control with the M… | 66 | 1713 | active |
| AnswerDotAI/ModernBERT ModernBERT is the research repository for a modernized BERT-family bidirectional encoder trained on 2 trillion tokens with an 8192-token co… | 56 | 1713 | active |
| McGill-NLP/llm2vec LLM2Vec is a Python library that converts decoder-only large language models into powerful text encoders via bidirectional attention, maske… | 49 | 1712 | active |
| feizc/FluxMusic FluxMusic is the official PyTorch implementation of a rectified flow Transformer model for text-to-music generation, from the paper 'Flux t… | 23 | 1711 | active |
| 4paradigm/OpenMLDB OpenMLDB is an open-source machine learning database that acts as a feature platform, computing consistent features for both offline traini… | 67 | 1710 | active |
| jakobrunge/tigramite Tigramite is a Python package for causal inference on time series data, providing causal discovery methods such as PCMCI, PCMCIplus, LPCMCI… | 51 | 1710 | active |
| numenta/nupic-legacy NuPIC is Numenta's open-source machine intelligence platform implementing Hierarchical Temporal Memory (HTM), a computational theory of the… | 23 | 6350 | maintenance |
| whitphx/streamlit-webrtc A Python library that adds real-time video and audio streaming to Streamlit apps via WebRTC. It lets developers process live camera/microph… | 94 | 1706 | active |
| google-research/circuit_training AlphaChip (circuit_training) is Google's open-source framework for generating chip floorplans using distributed deep reinforcement learning… | 62 | 1706 | active |
| software-mansion/react-native-executorch React Native ExecuTorch is a declarative React Native library for running AI models on-device, powered by Meta's ExecuTorch runtime. It shi… | 88 | 1702 | active |
| zai-org/ImageReward ImageReward is the first general-purpose human preference reward model for text-to-image generation, trained on 137k expert comparison pair… | 52 | 1702 | stable |
| lsdefine/simple_GRPO A minimal (~200 lines) Python implementation of GRPO reinforcement learning for training LLMs to develop r1-like reasoning, built on PyTorc… | 44 | 1702 | active |
| iMoonLab/yolov13 Official PyTorch implementation of YOLOv13, a real-time object detection model family (Nano to X-Large) featuring Hypergraph-based Adaptive… | 32 | 1702 | active |
| PaddlePaddle/PaddleVideo PaddleVideo is a video understanding toolkit built on PaddlePaddle, offering state-of-the-art models for action recognition, temporal actio… | 26 | 1702 | active |
| dingodb/dingo DingoDB is an open-source distributed multi-modal vector database that combines relational (SQL) and vector semantics in a unified platform… | 64 | 1701 | active |
| sihyun-yu/REPA REPA is the official PyTorch implementation of the ICLR 2025 paper 'Representation Alignment for Generation', a regularization technique th… | 27 | 1700 | active |
| jiaweizzhao/GaLore GaLore is a PyTorch library implementing Gradient Low-Rank Projection for memory-efficient full-parameter training of large language models… | 25 | 1700 | active |
| xingyaoww/code-act CodeAct is a research framework and agent system that uses executable Python code as a unified action space for LLM agents, enabling multi-… | 26 | 1699 | active |
| youssofal/MTPLX MTPLX is a native macOS app and CLI that runs local LLMs on Apple Silicon roughly twice as fast using native multi-token prediction (MTP) s… | 81 | 1698 | active |
| kubeflow/katib Kubeflow Katib is a Kubernetes-native AutoML system for hyperparameter tuning, early stopping, and neural architecture search. It is ML-fra… | 71 | 1697 | stable |
| kyleskom/NBA-Machine-Learning-Sports-Betting A Python application that predicts NBA game winners and totals (over/under) using XGBoost and neural network models trained on historical t… | 66 | 1696 | active |
| BindsNET/bindsnet BindsNET is a Python package for simulating spiking neural networks (SNNs) built on PyTorch tensor functionality, running on CPUs or GPUs. … | 84 | 1695 | active |
| 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… | 58 | 1695 | active |
| stephansturges/WALDO WALDO is an open-source object detection model based on a YOLOv8 backbone, trained with a synthetic data pipeline to detect people, vehicle… | 32 | 1695 | active |
| NVlabs/InstantSplat InstantSplat is a research framework for photorealistic 3D scene reconstruction from extremely sparse image views using Gaussian Splatting,… | 33 | 1694 | active |
| tensorpack/tensorpack Tensorpack is a high-level neural network training interface built on graph-mode TensorFlow, focused on training speed and flexibility for … | 23 | 6286 | maintenance |
| mlrun/mlrun MLRun is an open-source MLOps and AI orchestration framework for building, training, deploying, and monitoring machine learning and generat… | 96 | 1692 | active |
| tensorly/tensorly TensorLy is a Python library for tensor learning, providing tensor decomposition, tensor algebra, and tensor regression with a simple API. … | 46 | 1691 | stable |
| 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 |
| 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 |
| 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 |
| HybridRobotics/berkeley-humanoid-lite Berkeley Humanoid Lite is the open-source codebase for a sub-$5,000 3D-printed humanoid robot platform from UC Berkeley. It includes Isaac … | 56 | 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 |
| gensyn-ai/rl-swarm RL Swarm is an open-source, permissionless peer-to-peer framework for running reinforcement learning training swarms over the internet, bui… | 54 | 1679 | active |
| 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 |
| trymirai/uzu Uzu is a high-performance inference engine written in Rust for running AI models directly on-device, with Python, TypeScript, and Swift bin… | 85 | 1678 | 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 |
| 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 |
| koaning/drawdata A Python library providing notebook widgets for drawing datasets interactively, built on anywidget. Drawn data can be exported as lists, pa… | 73 | 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 |
| mattpocock/evalite Evalite is a TypeScript-native eval runner for testing LLM-powered applications, built on Vitest with a local dev server UI for exploring o… | 60 | 1669 | active |
| 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 |
| deepsense-ai/ragbits Ragbits is a Python framework of modular building blocks for rapidly developing generative AI applications, covering LLM interaction, promp… | 75 | 1668 | active |
| 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 |
| opendatalab/labelU LabelU is an open-source multimodal data annotation platform supporting images, video, and audio with tools like bounding boxes, segmentati… | 96 | 1665 | 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 |
| gradio-app/trackio Trackio is a lightweight, free, local-first experiment tracking library from Hugging Face that stores metrics in SQLite with an optional Gr… | 82 | 1662 | active |