domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| google-research/frame-interpolation FILM is the official TensorFlow 2 implementation of a state-of-the-art frame interpolation neural network from Google Research, presented a… | 10 | 3150 | maintenance |
| migueldeicaza/TensorFlowSharp TensorFlowSharp provides .NET bindings to the TensorFlow C API, exposing a strongly-typed low-level API for C# and F#. It is designed mainl… | 10 | 3147 | maintenance |
| FudanNLP/fastNLP fastNLP is a lightweight, modularized and extensible NLP framework in Python that reduces engineering boilerplate such as data processing l… | 23 | 3141 | maintenance |
| linyiLYi/bilibot A local chatbot fine-tuned from Bilibili user comments, built on Qwen1.5-32B-Chat using Apple's MLX LoRA fine-tuning. It supports text chat… | 24 | 3139 | maintenance |
| open-mmlab/mmdeploy MMDeploy is the OpenMMLab model deployment framework that converts PyTorch-based OpenMMLab models (mmdetection, mmsegmentation, etc.) into … | 23 | 3137 | maintenance |
| google-deepmind/trfl TRFL is a Python library built on TensorFlow that provides building-block loss operations (e.g., Q-learning, TD learning, distributional RL… | 32 | 3131 | maintenance |
| open-mmlab/mmskeleton MMSkeleton is an OpenMMLAB toolbox for skeleton-based human understanding, built on PyTorch. It supports 2D pose estimation, skeleton-based… | 23 | 3127 | maintenance |
| diegoantognini/pyGAT A PyTorch implementation of the Graph Attention Network (GAT) model from Veličković et al. (2017), including a sparse-matrix variant. It re… | 32 | 3123 | maintenance |
| dbiir/UER-py UER-py is a PyTorch framework for pre-training transformer language models (BERT, GPT-2, T5, ELMo, etc.) and fine-tuning them on downstream… | 32 | 3112 | maintenance |
| lucasjinreal/yolov7_d2 A detectron2-based implementation of YOLOv7 that extends YOLO-style detection to instance segmentation, keypoint detection, and multi-head … | 23 | 3109 | maintenance |
| aksnzhy/xlearn xLearn is a high-performance C++ machine learning package implementing linear models (LR), factorization machines (FM), and field-aware fac… | 23 | 3090 | maintenance |
| tusen-ai/simpledet SimpleDet is a Python framework built on MXNet for object detection and instance recognition. It provides state-of-the-art detection models… | 32 | 3085 | maintenance |
| shogun-toolbox/shogun Shogun is a C++ machine learning toolbox offering unified, efficient implementations of many ML algorithms with automatically generated bin… | 23 | 3079 | maintenance |
| qwopqwop200/GPTQ-for-LLaMa A Python library that applies GPTQ 4-bit weight quantization to LLaMA large language models, drastically reducing memory usage and checkpoi… | 30 | 3073 | maintenance |
| Tramac/awesome-semantic-segmentation-pytorch A PyTorch library providing concise, modifiable reference implementations of many semantic segmentation models such as FCN, PSPNet, DeepLab… | 32 | 3069 | maintenance |
| stellargraph/stellargraph StellarGraph is a Python library for machine learning on graphs and networks, offering state-of-the-art graph neural network algorithms suc… | 23 | 3060 | maintenance |
| argman/EAST A TensorFlow re-implementation of the EAST (Efficient and Accurate Scene Text Detector) deep learning model for detecting text in natural s… | 32 | 3059 | maintenance |
| rinongal/textual_inversion Official implementation of the Textual Inversion paper, which learns new word embeddings in a frozen text-to-image (Latent Diffusion) model… | 32 | 3055 | maintenance |
| bigscience-workshop/promptsource PromptSource is a Python toolkit and repository for creating, sharing, and applying natural language prompts to Hugging Face datasets. It h… | 23 | 3031 | maintenance |
| avinashpaliwal/Super-SloMo A PyTorch implementation of the Super SloMo paper for high-quality video frame interpolation, generating multiple intermediate frames to co… | 10 | 3025 | maintenance |
| CodeRayZhang/Movie_Recommend A full-stack movie recommendation system built on Spark, including a Scrapy crawler, an SSM-based movie website, an admin backend, and a Sp… | 32 | 3007 | maintenance |
| jfzhang95/pytorch-deeplab-xception A PyTorch implementation of the DeepLab v3+ semantic segmentation model with support for multiple backbones (Xception, ResNet, MobileNet, D… | 32 | 3000 | maintenance |
| google-research/t5x T5X is a modular, composable framework built on JAX and Flax for high-performance training, evaluation, and inference of sequence models at… | 75 | 2998 | maintenance |
| BayesWitnesses/m2cgen m2cgen is a lightweight Python library that transpiles trained statistical machine learning models into native code in languages such as Py… | 23 | 2998 | maintenance |
| keithito/tacotron An unofficial open-source TensorFlow implementation of Google's Tacotron end-to-end neural text-to-speech model, with a pre-trained model a… | 23 | 2995 | maintenance |
| libffcv/ffcv FFCV is a fast data loading system for PyTorch that dramatically increases data throughput in model training by replacing standard data loa… | 23 | 2993 | maintenance |
| Cartucho/mAP A Python library and script that computes mean Average Precision (mAP) for object detection models, adapted from the official PASCAL VOC 20… | 23 | 2966 | maintenance |
| johnma2006/mamba-minimal A single-file, readable PyTorch implementation of the Mamba selective state space model architecture. It produces numerically equivalent ou… | 26 | 2965 | maintenance |
| NVIDIA/MinkowskiEngine Minkowski Engine is an auto-differentiation neural network library for high-dimensional sparse tensors, built on PyTorch with CUDA accelera… | 23 | 2956 | maintenance |
| FreedomIntelligence/LLMZoo LLM Zoo is a project providing data, models, and evaluation benchmarks for large language models, including the multilingual Phoenix and Ch… | 30 | 2935 | maintenance |
| rail-berkeley/rlkit RLkit is a PyTorch-based reinforcement learning framework and algorithm collection from UC Berkeley RAIL. It provides reference implementat… | 32 | 2932 | maintenance |
| Handtrack Handtrack.js is a JavaScript library for prototyping realtime hand detection directly in the browser, framing handtracking as an object det… | 23 | 2930 | maintenance |
| pytorch/audio TorchAudio is PyTorch's audio library providing data manipulation, transforms, and dataset loaders for audio and speech machine learning. I… | 87 | 2927 | maintenance |
| ajbrock/BigGAN-PyTorch A PyTorch implementation of BigGAN for large-scale high-fidelity image synthesis, written by the paper's first author. It supports multi-GP… | 32 | 2923 | maintenance |
| facebookresearch/XLM PyTorch implementation of Cross-lingual Language Model Pretraining (XLM) from Facebook AI Research, covering MLM, CLM, and TLM objectives p… | 10 | 2920 | maintenance |
| Tencent/PhoenixGo PhoenixGo is a Go (weiqi/baduk) AI engine from Tencent that implements the AlphaGo Zero approach of learning the game without human knowled… | 23 | 2917 | maintenance |
| zhaipro/easy12306 A Python project that uses deep learning models to automatically recognize 12306 (China Railway) captchas, identifying both the Chinese tex… | 23 | 2910 | maintenance |
| Tencent/PocketFlow PocketFlow is an open-source AutoML framework from Tencent AI Lab for automatically compressing and accelerating deep learning models. Deve… | 32 | 2909 | maintenance |
| jbesomi/texthero Texthero is a Python toolkit for text preprocessing, representation, and visualization, designed to work on top of Pandas Series and DataFr… | 23 | 2907 | maintenance |
| Luolc/AdaBound AdaBound is a PyTorch implementation of the AdaBound optimizer from an ICLR 2019 paper, which behaves like Adam early in training and gradu… | 23 | 2902 | maintenance |
| alibaba/euler Euler is a distributed graph deep learning framework developed by Alibaba for running graph neural network algorithms on large-scale graphs… | 23 | 2897 | maintenance |
| huggingface/neuralcoref NeuralCoref is a spaCy pipeline extension that annotates and resolves coreference clusters using a neural network, with a pre-trained Engli… | 23 | 2892 | maintenance |
| sxyu/svox2 Official PyTorch/CUDA implementation of Plenoxels, a method for optimizing sparse voxel radiance fields for novel view synthesis without ne… | 32 | 2889 | maintenance |
| stanford-crfm/helm HELM is a Python framework from Stanford's CRFM for holistic, reproducible evaluation of foundation models including LLMs and multimodal mo… | 87 | 2887 | maintenance |
| nickliqian/cnn_captcha A Python project that uses convolutional neural networks built with TensorFlow to recognize character-based image captchas. It packages val… | 32 | 2881 | maintenance |
| tensorflow/lingvo Lingvo is a TensorFlow-based framework for building neural networks, particularly sequence models, with a focus on speech recognition, mach… | 72 | 2864 | maintenance |
| PlayVoice/whisper-vits-svc A PyTorch-based singing voice conversion and voice cloning engine built on VITS with Whisper, BigVGAN, and diffusion components. It lets us… | 23 | 2864 | maintenance |
| kpzhang93/MTCNN_face_detection_alignment Reference MATLAB/Caffe implementation of MTCNN, a multi-task cascaded convolutional neural network for joint face detection and facial land… | 32 | 2862 | maintenance |
| cybertronai/gradient-checkpointing A Python library that reduces GPU memory usage when training very deep neural networks via gradient checkpointing, trading computation for … | 32 | 2843 | maintenance |
| scikit-learn-contrib/sklearn-pandas A Python library that bridges pandas DataFrames and scikit-learn by mapping DataFrame columns to sklearn transformations. Its DataFrameMapp… | 23 | 2842 | maintenance |
| IDEA-Research/DINO Official PyTorch implementation of DINO, a state-of-the-art end-to-end object detection model based on DETR with improved denoising anchor … | 32 | 2834 | maintenance |
| rosinality/stylegan2-pytorch A PyTorch implementation of StyleGAN 2, the image-quality-improved version of the StyleGAN generative adversarial network, including traini… | 32 | 2833 | maintenance |
| kzl/decision-transformer Official codebase for the Decision Transformer paper, which frames reinforcement learning as sequence modeling with transformers. It contai… | 32 | 2830 | maintenance |
| OpenPipe/OpenPipe OpenPipe is an open-source fine-tuning and model-hosting platform that turns expensive LLM prompts into cheaper fine-tuned models. It offer… | 29 | 2830 | maintenance |
| huggingface/knockknock A small Python library that sends notifications when machine learning training completes or crashes, requiring only a decorator on your mai… | 32 | 2828 | maintenance |
| scikit-optimize/scikit-optimize Scikit-Optimize (skopt) is a Python library for sequential model-based optimization of expensive and noisy black-box functions, built on Nu… | 10 | 2828 | maintenance |
| microsoftarchive/promptbench PromptBench is a unified Python library from Microsoft for evaluating and understanding large language models across many datasets, models,… | 10 | 2819 | maintenance |
| philipperemy/keras-attention A Keras layer implementing attention mechanisms with Luong (multiplicative) and Bahdanau (additive) score functions for TensorFlow 2.x. It … | 55 | 2809 | maintenance |
| roytseng-tw/Detectron.pytorch A PyTorch reimplementation of Facebook's Detectron object detection framework, supporting Mask R-CNN, keypoint/pose estimation, and instanc… | 10 | 2808 | maintenance |
| IDEA-Research/DWPose DWPose is the official implementation of 'Effective Whole-body Pose Estimation with Two-stages Distillation' (ICCV 2023), providing whole-b… | 28 | 2807 | maintenance |
| ivanseidel/IAMDinosaur A Node.js application that uses neural networks and a genetic algorithm to teach Google Chrome's offline dinosaur game to jump cacti autono… | 32 | 2804 | maintenance |
| facebookresearch/hiplot HiPlot is a lightweight interactive visualization library from Facebook Research for exploring high-dimensional data using parallel coordin… | 10 | 2799 | maintenance |
| illuin-tech/colpali ColPali Engine is the Python library for training and running inference with ColVision visual document retrieval models such as ColPali, Co… | 91 | 2798 | maintenance |
| TeamHG-Memex/eli5 ELI5 is a Python library for debugging, inspecting, and explaining machine learning classifiers and regressors. It supports scikit-learn, X… | 66 | 2798 | maintenance |
| PhoebusSi/Alpaca-CoT Alpaca-CoT is an instruction-tuning platform that unifies interfaces for instruction data collection, parameter-efficient fine-tuning metho… | 30 | 2791 | maintenance |
| ethereon/caffe-tensorflow A Python tool that converts Caffe-trained neural network models into TensorFlow format, producing a NumPy parameter file and a Python class… | 32 | 2786 | maintenance |
| microsoft/CodeBERT A collection of pre-trained code models from Microsoft, including CodeBERT and successors like GraphCodeBERT and UniXcoder, usable via Hugg… | 32 | 2785 | maintenance |
| YCG09/chinese_ocr An end-to-end Chinese OCR system implemented with TensorFlow and Keras, combining CTPN for text detection with DenseNet + CTC for text reco… | 32 | 2782 | maintenance |
| tensorflow/graphics TensorFlow Graphics is a library of differentiable graphics layers for TensorFlow, including differentiable renderers, spatial transformers… | 64 | 2781 | maintenance |
| yfeng95/face3d A lightweight Python library implementing core 3D face processing functions: 3D morphable model (3DMM) generation and fitting, mesh I/O, tr… | 32 | 2779 | maintenance |
| henrywoo/pyllama A Python library that is a modified version of Meta's LLaMA implementation, making it easy to download model weights and run LLaMA models o… | 21 | 2778 | maintenance |
| tensorflow/ranking TensorFlow Ranking is a Python library for building Learning-to-Rank (LTR) models on top of TensorFlow. It provides pointwise, pairwise, an… | 10 | 2773 | maintenance |
| liucongg/ChatGLM-Finetuning A Python toolkit for fine-tuning ChatGLM-6B, ChatGLM2-6B, and ChatGLM3-6B large language models using Freeze, LoRA, P-Tuning, and full-para… | 21 | 2771 | maintenance |
| phanein/deepwalk DeepWalk is a Python command-line tool that learns low-dimensional vector representations of vertices in graphs using short random walks an… | 23 | 2759 | maintenance |
| boy1dr/SpleeterGui SpleeterGUI is a Windows desktop application providing a graphical front end for Spleeter, an AI music source separation tool. It bundles a… | 32 | 2755 | maintenance |
| mars-project/mars Mars is a tensor-based unified framework for large-scale data computation that provides NumPy-, pandas-, and scikit-learn-compatible APIs w… | 23 | 2742 | maintenance |
| saharmor/dalle-playground A web-based playground for generating images from text prompts using Stable Diffusion V2 (originally DALL-E Mini). It pairs a Python backen… | 32 | 2740 | maintenance |
| aditya-grover/node2vec Reference implementation of the node2vec algorithm for learning continuous low-dimensional node representations from arbitrary (un)directed… | 32 | 2739 | maintenance |
| apple/ml-ane-transformers A reference PyTorch implementation of the Transformer architecture optimized for Apple's Neural Engine (ANE), including optimized Hugging F… | 23 | 2736 | maintenance |
| cbfinn/maml Reference implementation of Model-Agnostic Meta-Learning (MAML) from Finn et al., ICML 2017, for fast adaptation of deep networks. It inclu… | 32 | 2727 | maintenance |
| kyegomez/OpenMythos OpenMythos is an open-source, theoretical PyTorch implementation of a Recurrent-Depth Transformer (RDT) architecture inspired by speculatio… | 51 | 14805 | experimental |
| carpedm20/ENAS-pytorch A PyTorch implementation of Efficient Neural Architecture Search (ENAS), which reduces NAS GPU-hours by 1000x via parameter sharing between… | 32 | 2719 | maintenance |
| torch-points3d/torch-points3d A PyTorch-based framework for deep learning on 3D point clouds, supporting models like PointNet, KPConv, and MinkowskiEngine sparse convolu… | 67 | 2711 | maintenance |
| VainF/DeepLabV3Plus-Pytorch A PyTorch library providing pretrained DeepLabv3 and DeepLabv3+ semantic segmentation models for Pascal VOC and Cityscapes datasets. It inc… | 32 | 2699 | maintenance |
| mahyarnajibi/SNIPER SNIPER is an efficient multi-scale training algorithm for object detection and instance segmentation that processes only context regions (c… | 32 | 2690 | maintenance |
| pengzhiliang/MAE-pytorch An unofficial PyTorch implementation of the MAE (Masked Autoencoders Are Scalable Vision Learners) paper, providing pretraining and finetun… | 32 | 2689 | maintenance |
| microsoft/pai OpenPAI is an open-source AI platform from Microsoft that provides resource scheduling and cluster management for machine learning workload… | 66 | 2688 | maintenance |
| JIA-Lab-research/LongLoRA LongLoRA is an efficient fine-tuning approach and codebase that extends the context length of pre-trained LLMs (Llama2 7B/13B/70B) using sh… | 27 | 2686 | maintenance |
| vchoutas/smplx A PyTorch implementation of SMPL-X, a unified parametric 3D model of the human body with fully articulated hands and an expressive face, al… | 32 | 2685 | maintenance |
| thisandagain/sentiment A Node.js module that performs sentiment analysis on text using the AFINN-165 wordlist and Emoji Sentiment Ranking. It supports custom lang… | 23 | 2678 | maintenance |
| graphdeeplearning/benchmarking-gnns An open-source benchmarking framework for graph neural networks (GNNs) published as a JMLR 2023 paper. It provides reproducible pipelines, … | 32 | 2671 | maintenance |
| hora-search/hora Hora is a Rust library of approximate nearest neighbor (ANN) search algorithms with SIMD acceleration and multithreading. It provides multi… | 63 | 2658 | maintenance |
| nerdyrodent/VQGAN-CLIP A Python application for running VQGAN+CLIP text-to-image generation locally on your own GPU, derived from Katherine Crowson's Google Colab… | 32 | 2647 | maintenance |
| baidu/Familia Familia is a C++ toolkit for industrial topic modeling, providing document topic inference and semantic matching tools along with pretraine… | 23 | 2644 | maintenance |
| linyiLYi/pose-monitor An Android app that uses the camera to detect bad sitting posture in real time and gives voice reminders. It runs MoveNet pose estimation p… | 23 | 2622 | maintenance |
| microsoft/GLIP GLIP is Microsoft's official implementation of Grounded Language-Image Pre-training, a vision-language model that unifies object detection … | 32 | 2607 | maintenance |
| alibaba/pipcook Pipcook is a JavaScript application framework for machine learning and its engineering, aimed at enabling JavaScript and front-end engineer… | 67 | 2594 | maintenance |
| zllrunning/face-parsing.PyTorch A PyTorch implementation of face parsing using a modified BiSeNet architecture, trained on the CelebAMask-HQ dataset. It provides training … | 32 | 2586 | maintenance |
| microsoft/DirectML DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning that provides GPU acceleration across DirectX … | 58 | 2580 | maintenance |
| baidu/AnyQ AnyQ is Baidu's open-source FAQ-based question answering framework with a configurable, plugin-based architecture covering question analysi… | 32 | 2575 | maintenance |