function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| Stability-AI/stable-audio-tools Stability AI's training and inference toolkit for conditional audio generation models, including Stable Audio Open. It supports training cu… | 73 | 3849 | active |
| Avatarify Avatarify is an open-source application that drives photorealistic avatars in real time for video-conferencing apps like Zoom and Skype, ba… | 23 | 16515 | maintenance |
| shenweichen/GraphEmbedding A Python library providing implementations of classic graph embedding algorithms including DeepWalk, LINE, Node2Vec, SDNE, and Struc2Vec. I… | 68 | 3845 | active |
| aurelio-labs/semantic-router Semantic Router is a Python library that provides a fast decision-making layer for LLMs and agents, routing requests based on semantic vect… | 94 | 3841 | active |
| piskvorky/gensim Gensim is a Python library for topic modelling, document indexing, and similarity retrieval over large text corpora. It provides memory-ind… | 50 | 16481 | maintenance |
| neuraloperator/neuraloperator A PyTorch library for learning neural operators, which map between function spaces rather than finite-dimensional vectors. It provides the … | 82 | 3836 | active |
| TransformerLensOrg/TransformerLens TransformerLens is a Python library for mechanistic interpretability of GPT-style transformer language models. It lets researchers load tho… | 99 | 3825 | active |
| google-research/scenic Scenic is a JAX-based library from Google Research focused on attention-based models for computer vision, providing shared lightweight libr… | 76 | 3821 | active |
| benfred/implicit A fast Python library implementing collaborative filtering recommendation algorithms for implicit feedback datasets, including ALS, BPR, lo… | 74 | 3815 | active |
| ddbourgin/numpy-ml numpy-ml is a collection of machine learning models and algorithms implemented exclusively in NumPy and the Python standard library, coveri… | 32 | 16330 | maintenance |
| meta-llama/codellama Official Meta repository with inference code for the Code Llama family of large language models for code, based on Llama 2. It provides min… | 10 | 16265 | maintenance |
| lightly-ai/lightly LightlySSL is a Python library built on PyTorch for self-supervised learning on images, offering modular implementations of methods like Si… | 93 | 3797 | active |
| google/deepvariant DeepVariant is a deep learning-based genomic variant caller that converts aligned DNA sequencing reads (BAM/CRAM) into pileup image tensors… | 70 | 3791 | stable |
| SandAI-org/MAGI-1 MAGI-1 is an open-source autoregressive video generation model from Sand.ai, released with Apache-2.0 licensed code and weights. It generat… | 59 | 3772 | active |
| StuckAtPrototype/Racer An open-source hardware and firmware project for a tiny PCB-based RC car built around the ESP32-H2, including KiCad schematics, gerbers, 3D… | 46 | 3765 | active |
| mathnet/mathnet-numerics Math.NET Numerics is an open-source numerical library for .NET and Mono, providing methods and algorithms for numerical computation in scie… | 36 | 3765 | stable |
| xtensor-stack/xtensor A header-only C++ library for numerical analysis with multi-dimensional array expressions, offering lazy broadcasting and a NumPy-inspired … | 74 | 3762 | active |
| Hunyuan-PromptEnhancer/PromptEnhancer PromptEnhancer is a prompt-rewriting framework from Tencent Hunyuan that uses a Chain-of-Thought rewriter trained via reinforcement learnin… | 61 | 3758 | active |
| aubio/aubio aubio is a C library (with Python bindings and CLI tools) for labeling music and sounds, detecting events such as note onsets, pitch, and t… | 57 | 3754 | stable |
| HeartMuLa/heartlib HeartMuLa is a family of open-source music foundation models that generate music conditioned on lyrics and tags with multilingual support. … | 50 | 3749 | active |
| astroautomata/PySR PySR is a high-performance symbolic regression library that searches for symbolic mathematical expressions optimizing a given objective, im… | 99 | 3739 | stable |
| fudan-generative-vision/hallo2 Hallo2 is a Python research library from Fudan University that animates a single portrait image using audio input, producing long-duration … | 26 | 3734 | active |
| vllm-project/llm-compressor LLM Compressor is a Python library for applying quantization and pruning algorithms to large language models, producing compressed-tensors … | 90 | 3726 | active |
| roboflow/trackers A Python library of clean-room, Apache 2.0 implementations of multi-object tracking algorithms including SORT, ByteTrack, OC-SORT, BoT-SORT… | 85 | 3717 | active |
| genmoai/mochi Mochi 1 is Genmo's open-source, state-of-the-art text-to-video generation model released under Apache 2.0, with a Python API, CLI, and Grad… | 46 | 3713 | active |
| IDEA-Research/Grounded-SAM-2 Grounded SAM 2 is a foundation-model pipeline that combines open-set detectors (Grounding DINO, Grounding DINO 1.5/1.6, Florence-2, DINO-X)… | 37 | 3708 | active |
| xinyu1205/recognize-anything Recognize Anything is a collection of open-source image recognition foundation models, including RAM, RAM++, and Tag2Text, that perform ima… | 33 | 3708 | active |
| MTG/essentia Essentia is an open-source C++ library for audio analysis and music information retrieval, offering a large collection of reusable DSP algo… | 65 | 3705 | stable |
| danijar/dreamerv3 A Python/JAX reimplementation of DreamerV3, a model-based reinforcement learning algorithm that learns a world model and trains an actor-cr… | 69 | 3704 | active |
| ant-research/MagicQuill MagicQuill is an intelligent interactive image editing system from a CVPR 2025 paper, combining a brush-based UI with AI-powered suggestion… | 46 | 3688 | active |
| microsoft/Bringing-Old-Photos-Back-to-Life The official PyTorch implementation of 'Bringing Old Photos Back to Life' (CVPR 2020 Oral), a deep learning model that restores old photos … | 23 | 15704 | maintenance |
| DLR-RM/BlenderProc BlenderProc is a procedural Python pipeline built on Blender for generating photorealistic synthetic training images with ground-truth anno… | 62 | 3684 | active |
| thu-ml/SageAttention SageAttention is a family of quantized attention kernels (INT8/FP8/FP4) that accelerate transformer inference 2-5x over FlashAttention with… | 40 | 3684 | active |
| Stability-AI/StableLM StableLM is Stability AI's repository of open-weight decoder-only transformer language models, including the 3B-parameter StableLM-3B-4E1T … | 30 | 15684 | maintenance |
| facebookresearch/map-anything MapAnything is an open-source research framework from Meta and CMU for universal feed-forward metric 3D reconstruction using an end-to-end … | 77 | 3682 | active |
| ferdous-alam/GenCAD GenCAD is a research codebase for image-conditioned CAD model generation using transformer-based contrastive representations (CCIP) and dif… | 36 | 3669 | active |
| PAIR-code/lit The Learning Interpretability Tool (LIT) is a visual, interactive web-based tool for understanding and debugging ML model behavior across t… | 73 | 3660 | active |
| HazyResearch/ThunderKittens ThunderKittens is a C++/CUDA framework of tile-based primitives for writing fast deep learning GPU kernels. It embeds natively into CUDA so… | 70 | 3659 | active |
| borglab/gtsam GTSAM is a C++ library implementing smoothing and mapping (SAM) for robotics and vision using factor graphs and Bayes networks as its core … | 92 | 3656 | active |
| zai-org/ChatGLM2-6B ChatGLM2-6B is an open-source bilingual (Chinese-English) 6B-parameter conversational large language model built on the GLM architecture. I… | 29 | 15528 | maintenance |
| shitagaki-lab/see-through A research framework from a SIGGRAPH 2026 paper that decomposes a single anime character illustration into up to 23 fully inpainted, semant… | 58 | 3641 | active |
| kaldi-asr/kaldi Kaldi is a C++ toolkit for speech recognition research and development, including acoustic modeling, feature extraction, decoding, and spea… | 52 | 15469 | maintenance |
| opendilab/DI-engine DI-engine is an open-source reinforcement learning framework from OpenDILab that provides comprehensive implementations of deep RL algorith… | 48 | 3638 | active |
| NExT-GPT/NExT-GPT NExT-GPT is an end-to-end any-to-any multimodal large language model that accepts and generates arbitrary combinations of text, image, vide… | 37 | 3638 | active |
| thu-ml/TurboDiffusion TurboDiffusion is a Python framework that accelerates end-to-end video diffusion model generation by 100-200x using SageAttention, Sparse-L… | 59 | 3623 | active |
| ploomber/ploomber Ploomber is a Python framework for building maintainable data pipelines from scripts and Jupyter notebooks, with iterative local developmen… | 10 | 3622 | active |
| mit-han-lab/llm-awq AWQ is a Python library implementing Activation-aware Weight Quantization for compressing large language models to low-bit (INT3/4) weights… | 44 | 3621 | active |
| facebookresearch/detr DETR is Facebook Research's PyTorch implementation of Detection Transformer, an end-to-end object detection model that replaces hand-crafte… | 10 | 15354 | maintenance |
| facebookresearch/sam-audio SAM-Audio is Meta's foundation model for isolating any sound in audio using text, visual, or temporal prompts. This repository provides inf… | 55 | 3612 | active |
| albumentations-team/albumentations Albumentations is a fast, flexible Python image augmentation library for computer vision, supporting images, masks, bounding boxes, keypoin… | 10 | 15315 | maintenance |
| MrNeRF/LichtFeld-Studio LichtFeld Studio is a native open-source desktop application for 3D Gaussian Splatting that combines training, real-time inspection, splat … | 92 | 3594 | active |
| AI4Finance-Foundation/FinRL-Trading FinRL-X is an open-source, AI-native modular infrastructure for quantitative trading that unifies data processing, strategy composition, ba… | 71 | 3592 | active |
| meta-pytorch/botorch BoTorch is a Python library for Bayesian Optimization research built on top of PyTorch, part of the PyTorch ecosystem. It provides modular,… | 91 | 3589 | active |
| ZhaoJ9014/face.evoLVe A high-performance face recognition library built on PaddlePaddle and PyTorch, providing comprehensive tools for face-related analytics and… | 38 | 3589 | active |
| 0x90d/videoduplicatefinder A cross-platform application that finds duplicate video and image files based on visual similarity, including duplicates with different res… | 97 | 3588 | active |
| zhaoyingjun/chatbot A Chinese chatbot project that trains conversational models (Seq2Seq and GPT-style, including small SLM variants) on custom corpora with lo… | 53 | 3581 | active |
| AiuniAI/Unique3D Unique3D is the official implementation of a NeurIPS 2024 paper that generates high-quality textured 3D meshes from a single image in about… | 38 | 3579 | active |
| urchade/GLiNER GLiNER is a Python framework for training and deploying lightweight, zero-shot Named Entity Recognition models that can extract arbitrary e… | 87 | 3574 | active |
| rdkit/rdkit RDKit is an open-source cheminformatics toolkit written in C++ with Python, Java, C#, and JavaScript wrappers. It provides core data struct… | 98 | 3565 | stable |
| SkalskiP/make-sense makesense.ai is a free, browser-based tool for labeling photos to prepare datasets for computer vision projects. It runs entirely client-si… | 23 | 3562 | active |
| alibaba/GraphScope GraphScope is a unified distributed graph computing platform from Alibaba that combines graph analytics (GRAPE), interactive graph queries … | 73 | 3556 | active |
| ToTheBeginning/PuLID PuLID is the official PyTorch implementation of a NeurIPS 2024 method for inserting a specific person's identity into text-to-image generat… | 40 | 3550 | active |
| sdv-dev/SDV SDV (Synthetic Data Vault) is a Python library for generating synthetic tabular data using machine learning models ranging from GaussianCop… | 99 | 3549 | stable |
| microsoft/hummingbird Hummingbird is a Microsoft library that compiles trained traditional ML models (decision trees, random forests, XGBoost, LightGBM) into ten… | 37 | 3544 | active |
| jmschrei/pomegranate Pomegranate is a Python library for fast and flexible probabilistic modeling, built on PyTorch. It supports distributions, mixture models, … | 27 | 3542 | active |
| pytorch/rl TorchRL is a PyTorch-native library of composable primitives for building reinforcement learning systems, including environments, policies,… | 97 | 3538 | active |
| Soul-AILab/SoulX-Podcast SoulX-Podcast is the official inference codebase for a text-to-speech model that generates long-form, multi-turn, multi-speaker podcast-sty… | 42 | 3535 | active |
| Kosinkadink/ComfyUI-AnimateDiff-Evolved A ComfyUI custom node pack providing an improved AnimateDiff integration plus advanced 'Evolved Sampling' options for animated video genera… | 70 | 3531 | active |
| starVLA/starVLA StarVLA is an open-source, Lego-like modular codebase for developing Vision-Language-Action (VLA) models for generalist robots. It unifies … | 69 | 3530 | active |
| fnnx-ai/scikit-llm Scikit-LLM is a Python library that integrates large language models like GPT into scikit-learn workflows as estimators. It enables tasks s… | 72 | 3529 | active |
| google-research/big_vision Google Research's official Jax/Flax codebase for training large-scale vision models such as Vision Transformer, SigLIP, MLP-Mixer, and LiT … | 42 | 3528 | active |
| cszn/KAIR A PyTorch image restoration toolbox providing training and testing code for many restoration models including DnCNN, FFDNet, SRMD, USRNet, … | 23 | 3523 | active |
| apirrone/Open_Duck_Mini Open Duck Mini is an open-source project to build a miniature (~42cm) bipedal robot inspired by Disney's BDX droid, with a bill of material… | 53 | 3521 | active |
| neonbjb/tortoise-tts Tortoise TTS is a multi-voice text-to-speech library built on PyTorch that prioritizes highly realistic prosody and intonation. It combines… | 32 | 14870 | maintenance |
| pathwaycom/bdh BDH (Dragon Hatchling) is a biologically inspired large language model architecture that bridges deep learning and neuroscience, implemente… | 54 | 3519 | active |
| MashiroSaber03/Saber-Translator Saber-Translator is an AI-powered manga translation application that detects speech bubbles, OCRs Japanese text, translates it, inpaints th… | 78 | 3516 | active |
| NVlabs/FoundationPose FoundationPose is NVIDIA's unified foundation model for 6D object pose estimation and tracking of novel objects, supporting both model-base… | 62 | 3516 | active |
| edtechre/pybroker PyBroker is a Python framework for developing algorithmic trading strategies with a focus on machine learning. It provides a fast NumPy/Num… | 93 | 3515 | active |
| EverMind-AI/MSA MSA (Memory Sparse Attention) is a Python framework for end-to-end trainable sparse latent-memory attention that scales LLM context to 100M… | 53 | 3515 | active |
| MooreThreads/Moore-AnimateAnyone An open-source reproduction of AnimateAnyone that animates a character from a single reference image using pose sequences from a driving vi… | 26 | 3514 | active |
| visionml/pytracking PyTracking is a PyTorch-based framework for visual object tracking and video object segmentation, providing official implementations of tra… | 23 | 3514 | active |
| google-deepmind/alphafold Open-source implementation of the AlphaFold 2 inference pipeline for predicting protein structures from amino acid sequences, including Alp… | 58 | 14811 | maintenance |
| NVIDIA/TransformerEngine Transformer Engine is an NVIDIA library for accelerating Transformer model training and inference on NVIDIA GPUs using low-precision format… | 99 | 3504 | active |
| PKU-YuanGroup/Video-LLaVA Video-LLaVA is a large vision-language model that aligns image and video representations into a unified visual space before projection into… | 27 | 3500 | active |
| aiming-lab/MetaClaw MetaClaw is a continual meta-learning framework that lets an LLM agent evolve from real conversations, combining skill synthesis from failu… | 68 | 3497 | active |
| autorope/donkeycar Donkeycar is an open-source Python library and hardware platform for building small-scale self-driving RC cars with Raspberry Pi or Jetson … | 88 | 3495 | active |
| mne-tools/mne-python MNE-Python is an open-source Python library for exploring, visualizing, and analyzing human neurophysiological data such as MEG, EEG, sEEG,… | 88 | 3495 | stable |
| Anttwo/SuGaR SuGaR is the official PyTorch implementation of a CVPR 2024 method that extracts accurate, editable meshes from 3D Gaussian Splatting recon… | 27 | 3495 | active |
| aleju/imgaug imgaug is a Python library for augmenting images in machine learning experiments, converting a small set of input images into a much larger… | 23 | 14741 | maintenance |
| borisdayma/dalle-mini DALL·E Mini is a Python library and model that generates images from a text prompt, available via pip and hosted on Hugging Face Model Hub.… | 23 | 14740 | maintenance |
| helblazer811/ManimML ManimML is a Python library built on the Manim Community Library for creating animations and visualizations of machine learning concepts li… | 23 | 3492 | active |
| facebookresearch/ijepa Official PyTorch implementation of I-JEPA, a self-supervised learning method that predicts latent representations of image regions from oth… | 10 | 3489 | active |
| NVIDIA/Model-Optimizer NVIDIA Model Optimizer (ModelOpt) is a Python library of state-of-the-art model optimization techniques including quantization, pruning, di… | 91 | 3488 | active |
| guandeh17/Self-Forcing Official implementation of Self Forcing, a training method for autoregressive video diffusion models that simulates inference during traini… | 37 | 3488 | active |
| POSTECH-CVLab/PyTorch-StudioGAN PyTorch-StudioGAN is a PyTorch library providing unified implementations of representative GAN architectures (BigGAN, StyleGAN2/3, etc.) fo… | 23 | 3487 | stable |
| Docta-ai/docta Docta is a Python library for data-centric AI that diagnoses and fixes issues in datasets such as label errors, noise, and misannotations. … | 69 | 3486 | active |
| Tencent-Hunyuan/Hunyuan3D-1 Tencent Hunyuan3D-1.0 is an open-source two-stage diffusion-based model for generating 3D assets from text prompts or images. It provides i… | 46 | 3482 | active |
| huggingface/optimum Optimum is a Hugging Face library that extends Transformers, Diffusers, timm, and Sentence Transformers with hardware-specific optimization… | 98 | 3469 | active |
| QData/TextAttack TextAttack is a Python framework for generating adversarial examples against NLP models, as well as for data augmentation and model trainin… | 87 | 3469 | active |
| MiniMax-AI/MiniMax-01 Official repository for MiniMax-Text-01 and MiniMax-VL-01, open-weight large language and vision-language models built on a linear attentio… | 34 | 3466 | active |