function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| Conchylicultor/DeepQA DeepQA is a TensorFlow implementation of Google's 'A Neural Conversational Model', a seq2seq RNN-based deep learning chatbot. It supports t… | 32 | 2910 | 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 |
| haoheliu/AudioLDM AudioLDM is a Python library and research model for text-to-audio generation using latent diffusion models conditioned on CLAP embeddings. … | 43 | 2907 | maintenance |
| biometrics/openbr OpenBR is an open-source biometrics library and command-line tool focused on face recognition, written in C++ on top of Qt and OpenCV. It p… | 50 | 2903 | 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 |
| 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 | 2901 | 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 |
| shouxieai/tensorRT_Pro A C++ library built on NVIDIA TensorRT that provides a high-level interface for high-performance deep learning inference, with Python bindi… | 23 | 2891 | maintenance |
| sxyu/svox2 Official PyTorch/CUDA implementation of Plenoxels, a method for optimizing sparse voxel radiance fields for novel view synthesis without ne… | 32 | 2888 | 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 | 2880 | 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 | 2844 | 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 | 2841 | 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 | 2836 | 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 |
| jina-ai/dalle-flow DALL·E Flow is a client-server workflow for generating high-definition images from text prompts using DALL·E-Mega, GLID-3 XL, and Stable Di… | 32 | 2829 | 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 | 2829 | 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 | 2818 | maintenance |
| ai-forever/Kandinsky-2 Kandinsky 2 is a multilingual text-to-image latent diffusion model library from Sber AI, supporting text2img, img2img, inpainting, outpaint… | 32 | 2813 | maintenance |
| SCUTlihaoyu/open-chat-video-editor An open-source Python tool that automatically generates short videos from a short text prompt or a web URL, producing narration, background… | 29 | 2813 | 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 | 2813 | 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 | 2809 | 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 |
| GerevAI/gerev Gerev is an AI-powered, self-hostable enterprise search engine that indexes workplace data sources like Slack, Confluence, Jira, and Google… | 30 | 2807 | maintenance |
| src-d/hercules Hercules is a fast, customizable Git repository history analysis tool written in Go, with a companion Python package (labours) for visualiz… | 23 | 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 | 2805 | maintenance |
| kwea123/nerf_pl An unofficial PyTorch implementation of NeRF (Neural Radiance Fields) and NeRF in the Wild built on pytorch-lightning, supporting multi-GPU… | 23 | 2804 | maintenance |
| Alpha-VLLM/LLaMA2-Accessory LLaMA2-Accessory is an open-source Python toolkit for pretraining, finetuning, and deploying large language models and multimodal LLMs, inc… | 29 | 2800 | maintenance |
| microsoft/NUWA Microsoft's official research repository for the NUWA family of multimodal generative models, a unified 3D transformer pipeline for visual … | 10 | 2791 | maintenance |
| PhoebusSi/Alpaca-CoT Alpaca-CoT is an instruction-tuning platform that unifies interfaces for instruction data collection, parameter-efficient fine-tuning metho… | 30 | 2788 | maintenance |
| microsoft/CodeBERT A collection of pre-trained code models from Microsoft, including CodeBERT and successors like GraphCodeBERT and UniXcoder, usable via Hugg… | 32 | 2787 | 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 |
| tensorflow/graphics TensorFlow Graphics is a library of differentiable graphics layers for TensorFlow, including differentiable renderers, spatial transformers… | 64 | 2782 | 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 |
| inspirit/jsfeat JSFEAT is a JavaScript computer vision library implementing classic CV algorithms in pure JS for browser use. It includes image processing … | 23 | 2772 | 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 |
| 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 | 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 | 2757 | 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 |
| paperswithcode/galai galai is the official Python library for loading and running inference with the GALACTICA general-purpose scientific language models from P… | 23 | 2741 | 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 | 2738 | maintenance |
| zhm-real/MotionPlanning A Python collection of motion planning and path tracking algorithms for autonomous vehicles, including Hybrid A*, Frenet Optimal Trajectory… | 32 | 2736 | maintenance |
| kyegomez/OpenMythos OpenMythos is an open-source, theoretical PyTorch implementation of a Recurrent-Depth Transformer (RDT) architecture inspired by speculatio… | 51 | 14872 | experimental |
| 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 | 2726 | maintenance |
| 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 | 2713 | maintenance |
| VainF/DeepLabV3Plus-Pytorch A PyTorch library providing pretrained DeepLabv3 and DeepLabv3+ semantic segmentation models for Pascal VOC and Cityscapes datasets. It inc… | 32 | 2702 | 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 | 2696 | maintenance |
| nikopueringer/CorridorKey CorridorKey is a Python-based neural network tool that unmixes green/blue screen footage, reconstructing true foreground color and linear a… | 53 | 14633 | experimental |
| junyanz/interactive-deep-colorization A deep learning application for interactively colorizing black and white images with user-guided hints, based on the SIGGRAPH 2017 paper 'R… | 32 | 2692 | 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 |
| cyrildiagne/ar-cutpaste An AR+ML research prototype that lets users capture objects from their physical surroundings with a phone camera and paste them into Photos… | 32 | 14570 | experimental |
| Tencent/GameAISDK Tencent's aitest is an open-source toolkit for building game AI based on game images, providing UI detection, in-game element recognition, … | 23 | 2682 | 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 | 2669 | maintenance |
| yerfor/GeneFace GeneFace is the official PyTorch implementation of an ICLR 2023 paper on generalized, high-fidelity audio-driven 3D talking face synthesis … | 21 | 2660 | 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 | 2657 | 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 |
| HypoX64/DeepMosaics DeepMosaics is a Python application that automatically removes or adds mosaics in images and videos using semantic segmentation and image-t… | 23 | 2635 | maintenance |
| sanghyun-son/EDSR-PyTorch Official PyTorch implementation of EDSR (Enhanced Deep Residual Networks for Single Image Super-Resolution, CVPRW 2017). It provides traini… | 32 | 2628 | 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 | 2621 | maintenance |
| PeterH0323/Smart_Construction A YOLOv5-based object detection application for detecting people, heads, and safety helmets on construction sites, including pretrained wei… | 23 | 2620 | maintenance |
| yangxy/GPEN GPEN (GAN Prior Embedded Network) is a deep learning library for blind face restoration of low-quality, in-the-wild photos. It also support… | 64 | 2613 | 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 | 2595 | maintenance |
| baaivision/Painter Painter and SegGPT are vision foundation models from BAAI for in-context visual learning, where a single generalist model performs diverse … | 32 | 2594 | maintenance |
| ShawnBIT/UNet-family A curated collection of UNet-family semantic segmentation models with PyTorch implementations and links to original papers and third-party … | 32 | 2592 | 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 |
| facebookarchive/device-year-class An Android library by Facebook that maps a device's RAM, CPU core count, and clock speed to a 'year class' indicating when those specs were… | 10 | 2584 | maintenance |
| microsoft/DirectML DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning that provides GPU acceleration across DirectX … | 58 | 2582 | maintenance |
| baidu/AnyQ AnyQ is Baidu's open-source FAQ-based question answering framework with a configurable, plugin-based architecture covering question analysi… | 32 | 2574 | maintenance |
| s3prl/s3prl S3PRL is a PyTorch toolkit for self-supervised speech pre-training and representation learning, bundling many upstream models like wav2vec … | 55 | 2562 | maintenance |
| tg-bomze/Face-Depixelizer A Jupyter Notebook-based tool that turns low-resolution pixelated face images into realistic high-resolution faces using the PULSE method, … | 32 | 2559 | maintenance |
| jcjohnson/torch-rnn torch-rnn provides high-performance, reusable RNN and LSTM modules for the Torch7 deep learning framework, used for character-level languag… | 32 | 2559 | maintenance |
| onnx/tensorflow-onnx tf2onnx is a tool that converts TensorFlow, Keras, TensorFlow.js, and TFLite models into the ONNX format via a command-line interface or Py… | 77 | 2557 | maintenance |
| OFA-Sys/OFA OFA is a unified sequence-to-sequence pretrained model supporting English and Chinese that unifies cross-modality, vision, and language tas… | 32 | 2557 | maintenance |
| LiyuanLucasLiu/RAdam RAdam is a Python implementation of Rectified Adam, a variant of the Adam optimizer that analytically reduces the large variance of adaptiv… | 32 | 2550 | maintenance |
| taesungp/contrastive-unpaired-translation A PyTorch implementation of Contrastive Unpaired Translation (CUT), an ECCV 2020 method for unpaired image-to-image translation using patch… | 32 | 2547 | maintenance |
| automl/Auto-PyTorch Auto-PyTorch is an AutoML framework that jointly performs neural architecture search and hyperparameter optimization for PyTorch models, us… | 23 | 2541 | maintenance |
| dmlc/gluon-nlp GluonNLP is a Python toolkit built on Apache MXNet that provides text data loading, processing tools, pretrained model zoos, and training u… | 10 | 2540 | maintenance |
| znxlwm/UGATIT-pytorch Official PyTorch implementation of U-GAT-IT, an unsupervised image-to-image translation model using attention modules and adaptive layer-in… | 32 | 2536 | maintenance |
| google-deepmind/dnc A TensorFlow implementation of the Differentiable Neural Computer (DNC), a recurrent neural network architecture with an external memory pu… | 32 | 2531 | maintenance |
| wenge-research/YAYI YAYI is a family of open-source Chinese-optimized large language models (7B/13B) instruction-tuned from LLaMA 2 and BLOOM base models by We… | 10 | 2527 | maintenance |
| yfeng95/DECA DECA is the official PyTorch implementation of a SIGGRAPH 2021 method that reconstructs a detailed 3D head model (pose, shape, facial detai… | 32 | 2518 | maintenance |
| dblalock/bolt Bolt is a C++ library (with a Python wrapper) for lossy compression of dense real-valued vectors that supports mathematical operations dire… | 32 | 2514 | maintenance |
| dmlc/decord Decord is a C++ library with Python bindings that provides efficient video and audio decoding for deep learning pipelines. It wraps hardwar… | 23 | 2514 | maintenance |
| zzh8829/yolov3-tf2 A clean implementation of YOLOv3 and YOLOv3-tiny object detection in TensorFlow 2.0, with pre-trained Darknet weight conversion, inference,… | 32 | 2513 | maintenance |
| unslothai/hyperlearn HyperLearn is a Python library providing optimized reimplementations of classical machine learning and statistical algorithms (SVD, eigende… | 32 | 2511 | maintenance |