domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| kijai/ComfyUI-Hunyuan3DWrapper A ComfyUI custom node wrapper for Tencent's Hunyuan3D-2 model, enabling 3D asset generation from images or text directly inside ComfyUI wor… | 53 | 1035 | active |
| tensorflow/hub TensorFlow Hub is a Python library for reusing parts of trained TensorFlow models (SavedModels) for transfer learning, wrapping them as Ker… | 24 | 3523 | maintenance |
| BabitMF/bmf BMF (Babit Multimedia Framework) is a cross-platform, multi-language multimedia and video processing framework developed by ByteDance, offe… | 75 | 1034 | active |
| videoflow/videoflow Videoflow is a Python framework for building distributed video and stream processing pipelines as directed acyclic graphs of producers, pro… | 66 | 1034 | active |
| liyiteng/AlohaMini AlohaMini is an open-source, 3D-printable dual-arm mobile robot with a motorized lift, buildable for under $1,000. It is LeRobot-ready for … | 57 | 1034 | active |
| NVlabs/DiffusionNFT DiffusionNFT is a research library implementing an online reinforcement learning paradigm for diffusion models that optimizes policy direct… | 47 | 1034 | active |
| OpenMined/TenSEAL TenSEAL is a Python library (with a C++ core) for performing homomorphic encryption operations on tensors, built on top of Microsoft SEAL. … | 87 | 1033 | active |
| ustcllm/RecFM RecFM is a collection of tools and frameworks from USTCLLM for developing foundation models tailored to recommendation systems. It bundles … | 40 | 1033 | active |
| samuel-vitorino/lm.rs A minimal Rust implementation for running LLM inference locally on CPU without ML libraries, supporting Gemma 2, Llama 3.2, and Phi 3.5 (in… | 23 | 1033 | active |
| NVIDIA-NeMo/Skills Nemo-Skills is a collection of Python pipelines for improving the skills of large language models, covering synthetic data generation, mode… | 70 | 1031 | active |
| adrida/tracer TRACER is a Python library that learns a routing policy from LLM classification traces, fitting a lightweight traditional ML surrogate to h… | 78 | 1030 | active |
| Jumpat/SegmentAnythingin3D SA3D is a research framework that lifts 2D Segment Anything (SAM) masks into 3D segmentation of objects within a NeRF or 3D Gaussian Splatt… | 40 | 1030 | active |
| kuleshov-group/bd3lms BD3-LMs is a research implementation of Block Discrete Denoising Diffusion Language Models that interpolate between autoregressive and diff… | 34 | 1029 | active |
| rl-tools/rl-tools RLtools is a pure C++ header-only, dependency-free deep reinforcement learning library supporting algorithms like SAC, TD3, and PPO. It com… | 65 | 1028 | active |
| motional/nuplan-devkit The official devkit for the nuPlan dataset, providing tools to load, visualize, and process the large-scale autonomous driving planning dat… | 40 | 1028 | active |
| jfilter/clean-text A Python package for cleaning and normalizing messy text, especially user-generated content from the web and social media. It fixes unicode… | 81 | 1027 | active |
| soCzech/TransNetV2 TransNet V2 is a deep neural network for shot boundary detection in videos, achieving state-of-the-art results on benchmarks like ClipShots… | 32 | 1027 | stable |
| zhyever/PatchFusion PatchFusion is a CVPR 2024 end-to-end tile-based framework for high-resolution monocular metric depth estimation from single images. It fus… | 57 | 1026 | active |
| andyzoujm/representation-engineering Official library for Representation Engineering (RepE), a top-down approach to AI transparency that monitors and manipulates population-lev… | 18 | 1024 | active |
| aim-uofa/AdelaiDet AdelaiDet is an open-source Python toolbox built on Detectron2 that implements multiple instance-level detection and recognition algorithms… | 32 | 3478 | maintenance |
| TutteInstitute/datamapplot DataMapPlot is a Python library for creating beautiful, presentation- and publication-ready plots of data maps (2D embeddings of clustered … | 89 | 1023 | active |
| google-deepmind/tree tree (dm-tree) is a Python library for working with nested data structures, generalizing the builtin map function to apply functions to lea… | 79 | 1022 | stable |
| microsoft/tensorwatch TensorWatch is a Python library from Microsoft Research for debugging, monitoring, and visualizing machine learning training in real time, … | 66 | 3471 | maintenance |
| open-mmlab/mmyolo MMYOLO is the OpenMMLab toolbox and benchmark for the YOLO series of object detection models, implemented on PyTorch. It provides unified i… | 23 | 3468 | maintenance |
| TJU-DRL-LAB/AI-Optimizer AI-Optimizer is a deep reinforcement learning toolkit from TJU's DRL lab offering algorithm libraries spanning model-free, model-based, off… | 32 | 3465 | maintenance |
| silvery107/rl-mpc-locomotion A Python framework combining deep reinforcement learning with model predictive control (MPC) for quadruped robot locomotion, where a policy… | 68 | 1020 | active |
| fla-org/native-sparse-attention Efficient Triton kernel implementations of Native Sparse Attention (NSA), a hardware-aligned, natively trainable sparse attention mechanism… | 49 | 1020 | active |
| scalanlp/breeze Breeze is a numerical processing library for Scala providing linear algebra, optimization, and statistical functionality. It is now mostly … | 52 | 3454 | maintenance |
| towhee-io/towhee Towhee is a Python framework for building ETL pipelines that process unstructured data (images, video, text, audio) into embeddings using s… | 23 | 3454 | maintenance |
| tensorflow/adanet AdaNet is a lightweight TensorFlow-based AutoML framework that automatically learns high-quality neural network architectures and ensembles… | 10 | 3454 | maintenance |
| facebookresearch/PyTorch-BigGraph PyTorch-BigGraph is a distributed system for learning embeddings of very large graph-structured data, scaling to billions of entities and t… | 10 | 3454 | maintenance |
| alex-petrenko/sample-factory Sample Factory is a high-throughput Python reinforcement learning library implementing synchronous and asynchronous policy gradient algorit… | 63 | 1017 | active |
| autonomousvision/gaussian-opacity-fields Gaussian Opacity Fields (GOF) is a Python/CUDA research implementation for efficient, adaptive surface reconstruction in unbounded scenes u… | 25 | 1017 | active |
| HobbitLong/SupContrast A PyTorch reference implementation of the Supervised Contrastive Learning paper (SupCon loss) that also supports SimCLR when labels are omi… | 32 | 3449 | maintenance |
| ShangtongZhang/DeepRL A modularized PyTorch implementation of popular deep reinforcement learning algorithms including DQN variants, PPO, DDPG, TD3, A2C, and Opt… | 23 | 3448 | maintenance |
| YingfanWang/PaCMAP PaCMAP is a Python library implementing Pairwise Controlled Manifold Approximation, a dimensionality reduction technique that preserves bot… | 87 | 1016 | stable |
| microsoft/Tutel Tutel is Microsoft's optimized Mixture-of-Experts (MoE) library for efficient training and inference of large language models, featuring dy… | 79 | 1016 | active |
| maximeraafat/BlenderNeRF BlenderNeRF is a Blender add-on that generates synthetic NeRF and Gaussian Splatting datasets with a single click, exporting renders and ca… | 23 | 1014 | active |
| clab/dynet DyNet is a C++ neural network toolkit with Python bindings, designed for efficient CPU/GPU training of networks with dynamic per-instance s… | 23 | 3436 | maintenance |
| hezarai/hezar Hezar is an all-in-one Python AI library for the Persian language, covering NLP, speech recognition, OCR, and image captioning through a ta… | 78 | 1013 | active |
| dswah/pyGAM pyGAM is a Python library for building Generalized Additive Models (GAMs) with an emphasis on modularity and performance. Its API is design… | 77 | 1013 | active |
| eragonruan/text-detection-ctpn A TensorFlow implementation of the Connectionist Text Proposal Network (CTPN) for detecting horizontal scene text in images. It includes pr… | 23 | 3429 | maintenance |
| ownthink/Jiagu Jiagu is a Python library for Chinese natural language processing built on deep learning models trained on large-scale corpora. It provides… | 32 | 3425 | maintenance |
| microsoft/aurora Aurora is Microsoft's implementation of a deep learning foundation model for Earth system forecasting, predicting atmospheric variables lik… | 89 | 1010 | active |
| TensorSpeech/TensorFlowASR TensorFlowASR is a Python library implementing automatic speech recognition architectures such as DeepSpeech2, Jasper, RNN Transducer, Cont… | 66 | 1010 | active |
| SunOner/sunone_aimbot An AI-powered aimbot for first-person shooter games that uses YOLO object detection models (YOLOv8/v10/v12) with TensorRT/ONNX acceleration… | 65 | 1010 | active |
| thuml/Large-Time-Series-Model Official code, datasets, and checkpoints for Timer and Sundial, generative pre-trained Transformer foundation models for general time serie… | 58 | 1010 | active |
| kaito-project/kaito KAITO is a Kubernetes operator suite that automates LLM inference, fine-tuning, and RAG engine deployment using simplified CRD APIs. It aut… | 95 | 1009 | active |
| braintrustdata/autoevals Autoevals is a Python and TypeScript library for automatically evaluating AI model outputs using LLM-as-a-judge, heuristic, and statistical… | 85 | 1009 | active |
| wang-rui/phishguard-scaffold PhishGuard is a Python research framework that jointly performs phishing detection and dissemination control on social media using LLaMA-ba… | 47 | 1009 | active |
| TRI-ML/prismatic-vlms Prismatic VLMs is a PyTorch-based codebase for training visually-conditioned language models (VLMs) with flexible vision backbones like CLI… | 25 | 1009 | active |
| waifu2x (nunif) waifu2x is an image super-resolution and noise-reduction tool for anime-style art (and photos) using deep convolutional neural networks, or… | 78 | 3418 | maintenance |
| hmmlearn/hmmlearn hmmlearn is a Python library providing algorithms for unsupervised learning and inference of Hidden Markov Models, with an API modeled on s… | 32 | 3416 | maintenance |
| graspologic-org/graspologic graspologic is a Python package providing algorithms and utilities for graph (network) statistics, including embedding, clustering, matchin… | 76 | 1008 | active |
| neuralgcm/neuralgcm NeuralGCM is a Python library for building hybrid machine-learning/physics models of the Earth's atmosphere for weather and climate simulat… | 69 | 1008 | active |
| davidtvs/pytorch-lr-finder A PyTorch library implementing the learning rate range test from Leslie Smith's cyclical learning rates paper, including the fastai-tweaked… | 35 | 1008 | active |
| shinkuan/Akagi Akagi is a real-time mahjong AI assistant desktop app (Rust + Tauri) that observes game connections from Mahjong Soul, Tenhou, Riichi City,… | 93 | 1007 | active |
| AutoArk/open-audio-opd An industrial training stack for online policy distillation (OPD) of audio models, distilling compact ASR (and planned TTS) student models … | 52 | 1007 | active |
| pytorch/ELF ELF is an extensive, lightweight, and flexible C++/Python platform for game research, best known for ELF OpenGo, a full AlphaGoZero/AlphaZe… | 10 | 3411 | maintenance |
| google-research/inksight InkSight is a Google Research system that converts photos of offline handwritten text into digital ink strokes using a ViT and mT5 encoder-… | 65 | 1006 | active |
| facebookresearch/fairscale FairScale is a PyTorch extension library providing composable modules and APIs for high-performance, large-scale distributed training, incl… | 10 | 3407 | maintenance |
| MoonshotAI/checkpoint-engine Checkpoint-engine is a lightweight Python middleware for updating model weights in-place across LLM inference engines, a critical step in r… | 80 | 1005 | active |
| meetps/pytorch-semseg A PyTorch library implementing popular semantic segmentation architectures such as FCN, U-Net, SegNet, PSPNet, ICNet, FRRN, and LinkNet, wi… | 23 | 3402 | maintenance |
| minimaxir/gpt-2-simple A Python package that simplifies fine-tuning OpenAI's GPT-2 text-generation model (124M/355M) on custom text and generating text from the r… | 23 | 3400 | maintenance |
| JuliaDiff/ForwardDiff.jl ForwardDiff.jl is a Julia library implementing forward mode automatic differentiation for computing derivatives, gradients, Jacobians, Hess… | 99 | 1004 | stable |
| huawei-noah/noah-research A collection of research code subprojects released by Huawei Noah's Ark Lab, each in its own directory. It is not an official Huawei produc… | 76 | 1004 | active |
| TinyLLaVA/TinyLLaVA_Factory TinyLLaVA Factory is an open-source modular PyTorch/HuggingFace codebase for training small-scale large multimodal models (LMMs) that combi… | 68 | 1004 | active |
| SciSharp/TensorFlow.NET TensorFlow.NET provides .NET Standard bindings for Google's TensorFlow, aiming to implement the complete TensorFlow API in C# and F#. It in… | 24 | 3398 | maintenance |
| PrithivirajDamodaran/FlashRank FlashRank is an ultra-light, fast Python library for re-ranking search and retrieval results using SoTA cross-encoders and listwise LLM-bas… | 58 | 1003 | active |
| sematic-ai/sematic Sematic is an open-source ML pipeline development platform that lets engineers write end-to-end pipelines in pure Python. Pipelines can run… | 30 | 1001 | active |
| pjlab-sys4nlp/llama-moe LLaMA-MoE is a Python toolkit and model series for building Mixture-of-Experts (MoE) language models from dense LLaMA models via expert con… | 19 | 1001 | active |
| catalyst-team/catalyst Catalyst is a high-level PyTorch framework for deep learning research and development, focused on reproducibility, rapid experimentation, a… | 64 | 3382 | maintenance |
| argilla-io/distilabel Distilabel is a Python framework for building scalable pipelines that generate synthetic data and AI feedback, based on verified research p… | 74 | 3378 | maintenance |
| shelhamer/fcn.berkeleyvision.org Reference implementation of Fully Convolutional Networks (FCN) for semantic segmentation from the CVPR 2015 / PAMI 2016 papers, built on Ca… | 32 | 3350 | maintenance |
| hamuchiwa/AutoRCCar An open-source project that turns a hobby RC car into an autonomous self-driving vehicle using a Raspberry Pi, Arduino, camera, and ultraso… | 32 | 3344 | maintenance |
| pytorch/glow Glow is a machine learning compiler and execution engine for neural network hardware accelerators, developed under the PyTorch project. It … | 10 | 3320 | maintenance |
| pytorch-yolo-v3 A minimal PyTorch implementation of the YOLO v3 object detection algorithm, supporting detection on images and video with configurable reso… | 32 | 3312 | maintenance |
| open-mmlab/mmselfsup OpenMMLab's PyTorch-based toolbox and benchmark for self-supervised and unsupervised visual representation learning. It provides implementa… | 23 | 3302 | maintenance |
| resemble-ai/Resemblyzer Resemblyzer is a Python package that uses a deep learning voice encoder to convert speech audio into 256-dimensional voice embeddings. Thes… | 23 | 3300 | maintenance |
| bytedance/lightseq LightSeq is a high-performance CUDA-based library for training and inference of sequence models like Transformer, BERT, GPT, and BART, with… | 10 | 3295 | maintenance |
| facebookresearch/vissl VISSL is Facebook AI Research's extensible, modular and scalable PyTorch library for state-of-the-art self-supervised learning with images.… | 10 | 3293 | maintenance |
| tradytics/eiten Eiten is an open-source Python toolkit for building statistical and algorithmic investment portfolios, implementing strategies like Eigen P… | 32 | 3290 | maintenance |
| NVIDIA/flownet2-pytorch A PyTorch implementation of FlowNet 2.0 for deep-learning-based optical flow estimation, released by NVIDIA. It provides multiple network a… | 66 | 3289 | maintenance |
| google-research/albert Official TensorFlow implementation and pretrained checkpoints of ALBERT, a lite version of BERT for self-supervised learning of language re… | 10 | 3278 | maintenance |
| google-deepmind/dm-haiku Haiku is a simple neural network library for JAX from Google DeepMind, providing an object-oriented module abstraction (hk.Module) and a fu… | 89 | 3275 | maintenance |
| guoguibing/librec LibRec is a Java library for recommender systems implementing over 70 state-of-the-art recommendation algorithms. It supports the two class… | 23 | 3257 | maintenance |
| facebookresearch/MUSE MUSE is a Python library from Facebook AI Research for creating and aligning multilingual word embeddings, using both supervised (bilingual… | 10 | 3244 | maintenance |
| martinarjovsky/WassersteinGAN Reference PyTorch implementation of the Wasserstein GAN paper, providing training scripts for DCGAN and MLP architectures on datasets like … | 32 | 3243 | maintenance |
| google/model_search Model Search is a Google AutoML framework that implements neural architecture search algorithms at scale to find optimal DNN architectures … | 10 | 3238 | maintenance |
| JoePenna/Dreambooth-Stable-Diffusion A Jupyter Notebook-based implementation of Dreambooth fine-tuning for Stable Diffusion, adapted from XavierXiao's repo with tweaks for trai… | 32 | 3211 | maintenance |
| mkocabas/VIBE Official PyTorch implementation of VIBE (CVPR 2020), a video-based method for 3D human body pose and shape estimation that predicts SMPL bo… | 23 | 3211 | maintenance |
| tomgoldstein/loss-landscape PyTorch code for visualizing the loss landscape of neural networks, from the NIPS 2018 paper 'Visualizing the Loss Landscape of Neural Nets… | 32 | 3199 | maintenance |
| tinyvision/DAMO-YOLO DAMO-YOLO is a fast and accurate object detection framework built on PyTorch, featuring NAS-searched backbones, RepGFPN, a lightweight Zero… | 32 | 3183 | maintenance |
| yusugomori/DeepLearning A multi-language (Python, C, C++, Java, Scala, Go) educational implementation of classic deep learning algorithms such as Deep Belief Nets,… | 32 | 3177 | maintenance |
| adambielski/siamese-triplet A PyTorch library implementing siamese and triplet networks for learning image embeddings, with online pair/triplet mining strategies. It p… | 32 | 3174 | maintenance |
| jettify/pytorch-optimizer torch-optimizer is a Python library providing a collection of additional optimization algorithms for PyTorch, drop-in compatible with the t… | 23 | 3171 | maintenance |
| cemoody/lda2vec A Python library implementing lda2vec, a hybrid topic model that combines word2vec word embeddings with LDA-style interpretable document to… | 32 | 3169 | maintenance |
| farizrahman4u/seq2seq A sequence-to-sequence learning add-on library for Keras, providing modular encoder-decoder layers and ready-made Seq2Seq models. It suppor… | 32 | 3169 | maintenance |
| huawei-noah/Pretrained-Language-Model A collection of pretrained language models and optimization techniques from Huawei Noah's Ark Lab, including PanGu-α (200B-parameter Chines… | 32 | 3165 | maintenance |
| project-baize/baize-chatbot Baize is an open-source chat model built on LLaMA using LoRA parameter-efficient tuning, trained on 100k self-chat dialogs generated by Cha… | 21 | 3150 | maintenance |