function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| yehengchen/Object-Detection-and-Tracking A collection of Python implementations combining YOLO-based object detection with SORT and DeepSORT multi-object tracking. It includes exam… | 32 | 1962 | maintenance |
| ml-jku/hopfield-layers A PyTorch library implementing modern Hopfield networks with continuous states, whose update rule is equivalent to transformer attention. I… | 32 | 1961 | maintenance |
| devzwy/open_nsfw_android An Android library for offline NSFW (pornographic) image detection, ported from Yahoo's open_nsfw project using TensorFlow Lite. It classif… | 10 | 1957 | maintenance |
| mit-han-lab/once-for-all Once-for-All (OFA) is a PyTorch library implementing the ICLR 2020 Once-for-All network, which trains a single supernet that can be special… | 23 | 1955 | maintenance |
| guillaumegenthial/sequence_tagging A TensorFlow implementation of a Bi-LSTM + CRF model with character embeddings for named entity recognition and sequence tagging. It trains… | 32 | 1952 | maintenance |
| hukaixuan19970627/yolov5_obb A PyTorch implementation of YOLOv5 extended for oriented (rotated) object detection using Circular Smooth Label (CSL) angle encoding. It pr… | 32 | 1947 | maintenance |
| peng-zhihui/DeepVision DeepVision is an Android-based computer vision algorithm inference and validation framework by peng-zhihui. It decouples data flow control,… | 32 | 1943 | maintenance |
| google/sling SLING is a natural language frame semantics parser that annotates text with frame semantic graph representations using bi-directional LSTMs… | 10 | 1931 | maintenance |
| uTensor/uTensor uTensor is an extremely lightweight machine learning inference framework built on TensorFlow and optimized for Arm Cortex-M microcontroller… | 32 | 1929 | maintenance |
| danieldjohnson/biaxial-rnn-music-composition A Python implementation of a biaxial recurrent neural network (LSTM-based) trained to generate classical music from MIDI data. It includes … | 32 | 1926 | maintenance |
| S-LoRA/S-LoRA S-LoRA is a serving system for large language models that scales to thousands of concurrent LoRA adapters on one or more GPUs. It uses Unif… | 10 | 1923 | maintenance |
| Tencent/NeuralNLP-NeuralClassifier NeuralClassifier is a PyTorch-based open-source toolkit for neural text classification, with a focus on hierarchical multi-label classifica… | 56 | 1922 | maintenance |
| black0017/MedicalZooPytorch A PyTorch-based open-source library implementing state-of-the-art 3D (and some 2D) deep neural networks for multi-modal medical image segme… | 32 | 1918 | maintenance |
| Tiiiger/bert_score BERTScore is a PyTorch implementation of the BERTScore automatic evaluation metric for text generation, which scores generated text against… | 23 | 1918 | maintenance |
| appvision-ai/fast-bert Fast-Bert is a Python deep learning library for training and deploying BERT, RoBERTa, and XLNet based models for NLP tasks, starting with m… | 23 | 1917 | maintenance |
| Tencent/plato Plato is Tencent's high-performance distributed graph computation framework, developed for graph processing and graph machine learning at W… | 23 | 1910 | maintenance |
| DmitryUlyanov/Multicore-TSNE A multicore modification of Barnes-Hut t-SNE written in C++ with Python (CFFI) and Torch wrappers for fast dimensionality reduction of high… | 32 | 1909 | maintenance |
| WongKinYiu/PyTorch_YOLOv4 A PyTorch implementation of the YOLOv4 real-time object detection model, based on ultralytics/yolov3. It includes research extensions such … | 23 | 1908 | maintenance |
| Palashio/libra Libra is a Python autoML library that lets users build, train, and evaluate machine learning models with one-line natural-language-style qu… | 40 | 1907 | maintenance |
| snap-stanford/GraphGym GraphGym is a platform for designing and evaluating Graph Neural Networks (GNNs), built around a highly modularized pipeline covering data … | 23 | 1903 | maintenance |
| ronghuaiyang/arcface-pytorch A PyTorch implementation of ArcFace, a deep metric learning approach for face recognition that adds angular margin penalties to face embedd… | 32 | 1901 | maintenance |
| whai362/PVT Official PyTorch implementation of the Pyramid Vision Transformer (PVTv1 and PVTv2) model series, providing pretrained backbone networks fo… | 23 | 1901 | maintenance |
| jiesutd/NCRFpp NCRF++ is a PyTorch-based toolkit for neural sequence labeling tasks such as NER, POS tagging, and word segmentation. Model architectures c… | 32 | 1899 | maintenance |
| ChenyangSi/FreeU FreeU is a research codebase implementing a training-free method that improves diffusion model sample quality by re-weighting U-Net backbon… | 28 | 1899 | maintenance |
| facebookresearch/denoiser A PyTorch library implementing a causal, real-time speech enhancement model that operates on raw waveforms to remove background noise and r… | 10 | 1899 | maintenance |
| zjhuang22/maskscoring_rcnn Official PyTorch implementation of Mask Scoring R-CNN (CVPR 2019), built on maskrcnn-benchmark. It adds a network block that learns the qua… | 32 | 1895 | maintenance |
| TreB1eN/InsightFace_Pytorch A PyTorch reimplementation of InsightFace/ArcFace for face recognition, including backbone models (IR-SE50, MobileFacenet) and pretrained w… | 32 | 1893 | maintenance |
| uber/petastorm Petastorm is a Python data access library from Uber that enables single-machine or distributed training and evaluation of deep learning mod… | 58 | 1892 | maintenance |
| imgcook/imgcook imgcook is a design-to-code tool from Alibaba that intelligently generates front-end code (views, data bindings, components, and some busin… | 23 | 1892 | maintenance |
| junshutang/Make-It-3D Make-It-3D is a research codebase from an ICCV 2023 paper that creates high-fidelity 3D content from a single image using a 2D diffusion mo… | 30 | 1891 | maintenance |
| microsoft/BlingFire Bling Fire is a fast finite state machine and regular expression manipulation library from Microsoft, primarily used for high-performance n… | 23 | 1890 | maintenance |
| Ucas-HaoranWei/Vary Official ECCV 2024 implementation of Vary, a method for scaling up the vision vocabulary of large vision-language models. It provides train… | 26 | 1889 | maintenance |
| NVIDIA-AI-IOT/deepstream_python_apps Python bindings and sample applications for NVIDIA's DeepStream SDK, built with Pybind11 and Gst Python for constructing video analytics pi… | 59 | 1887 | maintenance |
| marella/ctransformers CTransformers provides Python bindings for transformer language models implemented in C/C++ using the GGML library. It offers a unified int… | 20 | 1886 | maintenance |
| Duankaiwen/CenterNet Official PyTorch implementation of CenterNet, a one-stage keypoint-triplet-based object detector built on CornerNet. It trains from scratch… | 32 | 1884 | maintenance |
| hexiangnan/neural_collaborative_filtering Reference implementation of the WWW'17 Neural Collaborative Filtering paper, providing GMF, MLP, and NeuMF models for implicit-feedback rec… | 32 | 1884 | maintenance |
| zhoubolei/CAM Reference implementation of Class Activation Mapping (CAM), a CVPR'16 technique that visualizes which image regions a CNN attends to when p… | 32 | 1883 | maintenance |
| mpatacchiola/deepgaze Deepgaze is a Python computer vision library for human-computer interaction built on OpenCV and TensorFlow. It provides CNN-based head pose… | 32 | 1880 | maintenance |
| salesforce/ctrl CTRL is a 1.63 billion-parameter conditional Transformer language model from Salesforce that generates text controllable via control codes … | 10 | 1879 | maintenance |
| tradytics/surpriver Surpriver is a Python application that uses machine learning and anomaly detection to find stocks with unusual price and volume patterns th… | 32 | 1877 | maintenance |
| open-mmlab/mmaction MMAction is an open-source PyTorch toolbox for video action understanding, covering action recognition, temporal action detection, and spat… | 32 | 1876 | maintenance |
| google/TensorNetwork TensorNetwork is a Python library for building and contracting tensor networks, with backends for TensorFlow, JAX, PyTorch, and NumPy. It p… | 10 | 1870 | maintenance |
| facebookresearch/DPR DPR is a set of tools and pretrained models for dense passage retrieval in open-domain question answering, based on the EMNLP 2020 paper fr… | 10 | 1870 | maintenance |
| pierluigiferrari/ssd_keras A Keras implementation of the Single Shot MultiBox Detector (SSD) object detection architecture, with ports of the original trained weights… | 23 | 1869 | maintenance |
| waleedka/hiddenlayer A lightweight Python library for rendering readable neural network architecture graphs and plotting training metrics for PyTorch, TensorFlo… | 23 | 1867 | maintenance |
| minimaxir/automl-gs automl-gs is a Python AutoML tool that takes an input CSV and a target prediction field and automatically generates a trained machine learn… | 23 | 1866 | maintenance |
| xiaolai-sqlai/mobilenetv3 A PyTorch implementation of the MobileNetV3 architecture with retrained pre-trained weights that outperform the original paper and torchvis… | 32 | 1864 | maintenance |
| facebookresearch/TimeSformer Official PyTorch implementation of TimeSformer, a space-time attention transformer model for video understanding from an ICML 2021 paper. I… | 10 | 1863 | maintenance |
| anthropics/jacobian-lens A Python library implementing the Jacobian lens, an interpretability technique that linearly transports residual-stream activations at any … | 56 | 1862 | maintenance |
| raminmh/liquid_time_constant_networks Official code repository for Liquid Time-Constant Networks (LTCs), a class of continuous-time recurrent neural networks with varying time-c… | 32 | 1861 | maintenance |
| hanzhanggit/StackGAN TensorFlow implementation of StackGAN, a stacked generative adversarial network that synthesizes photo-realistic images from text descripti… | 32 | 1860 | maintenance |
| magicleap/Atlas Atlas is a PyTorch-based deep learning model from Magic Leap that performs end-to-end 3D scene reconstruction from posed RGB images, produc… | 32 | 1859 | maintenance |
| MrNothing/AI-Blocks AI-Blocks is a WYSIWYG desktop application for visually building machine learning models by dragging objects with attached scripts in a sce… | 23 | 1857 | maintenance |
| nlpodyssey/spago Spago is a self-contained machine learning and NLP library written in pure Go, featuring an asynchronous computation graph with automatic d… | 29 | 1850 | maintenance |
| ZhengyaoJiang/PGPortfolio PGPortfolio is a Python library implementing a deep reinforcement learning framework for financial portfolio management, based on the paper… | 23 | 1850 | maintenance |
| xinshuoweng/AB3DMOT Official Python implementation of AB3DMOT, a simple real-time baseline for 3D multi-object tracking using oriented 3D bounding boxes from L… | 32 | 1847 | maintenance |
| WangYueFt/dgcnn Author's reference implementation of Dynamic Graph CNN (DGCNN), a neural network module called EdgeConv for learning on 3D point clouds. It… | 32 | 1846 | maintenance |
| ermongroup/ddim Official PyTorch implementation of Denoising Diffusion Implicit Models (DDIM), a faster sampling method for diffusion probabilistic models.… | 32 | 1846 | maintenance |
| Score-Based Generative Modeling through SDEs Official PyTorch implementation of the ICLR 2021 paper 'Score-Based Generative Modeling through Stochastic Differential Equations', unifyin… | 32 | 1845 | maintenance |
| google/gemmlowp gemmlowp is a small, self-contained, headers-only C++ library for low-precision (quantized) general matrix multiplication (GEMM). It provid… | 32 | 1844 | maintenance |
| AlexeyAB/Yolo_mark A Windows and Linux GUI application for drawing bounding boxes around objects in images to create labeled training data for YOLO v2/v3 obje… | 32 | 1841 | maintenance |
| microsoft/Cream A collection of Microsoft research projects on neural architecture search (NAS) and efficient Vision Transformers, including AutoFormer, Ti… | 23 | 1841 | maintenance |
| google-deepmind/kinetics-i3d A repository of pre-trained Inflated 3D Convnet (I3D) models for video action classification, trained on the Kinetics dataset, released alo… | 32 | 1838 | maintenance |
| zetane/viewer Zetane Viewer is a free desktop application that visualizes machine learning models and their internal tensors in 3D, showing architecture,… | 23 | 1838 | maintenance |
| cubiq/ComfyUI_InstantID A ComfyUI custom node providing native InstantID support for identity-preserving image generation with SDXL, implemented without diffusers.… | 33 | 1837 | maintenance |
| minimaxir/aitextgen aitextgen is a Python library for training and generating text with GPT-2 and GPT Neo models, built on PyTorch, Hugging Face Transformers, … | 23 | 1836 | maintenance |
| Janspiry/Palette-Image-to-Image-Diffusion-Models An unofficial PyTorch implementation of the Palette paper on image-to-image diffusion models, supporting tasks like inpainting, uncropping,… | 32 | 1833 | maintenance |
| Kyubyong/tacotron A heavily documented TensorFlow implementation of Tacotron, a fully end-to-end text-to-speech synthesis model. It includes training, prepro… | 32 | 1832 | maintenance |
| bubbliiiing/faster-rcnn-pytorch A PyTorch implementation of the Faster R-CNN two-stage object detection model, supporting training on VOC-format datasets with ResNet or VG… | 23 | 1832 | maintenance |
| NVIDIA/semantic-segmentation NVIDIA's PyTorch monorepo implementing the paper 'Hierarchical Multi-Scale Attention for Semantic Segmentation', with pretrained weights an… | 32 | 1829 | maintenance |
| NVlabs/stylegan2-ada NVIDIA's official TensorFlow implementation of StyleGAN2 with adaptive discriminator augmentation (ADA), enabling high-quality GAN training… | 32 | 1828 | maintenance |
| THUDM/CogDL CogDL is a Python/PyTorch toolkit for graph deep learning that lets researchers train and compare GNN and graph embedding models on tasks l… | 23 | 1821 | maintenance |
| OpenPPL/ppq PPQ (PPL Quantization Tool) is an extensible, high-performance offline neural network quantization framework written in Python. It parses a… | 23 | 1820 | maintenance |
| yassouali/pytorch-segmentation A PyTorch library implementing multiple semantic segmentation models (DeepLab V3+, PSPNet, U-Net, SegNet, FCN, ENet, and others) with datas… | 26 | 1818 | maintenance |
| sshaoshuai/PointRCNN PyTorch implementation of PointRCNN, a two-stage 3D object detector that generates 3D box proposals directly from raw LiDAR point clouds. I… | 32 | 1817 | maintenance |
| kohya-ss/sd-webui-additional-networks An extension for AUTOMATIC1111's Stable Diffusion web UI that applies additional networks such as LoRA models to the base model on-the-fly … | 22 | 1816 | maintenance |
| google/compare_gan A TensorFlow library from Google Research providing configurable implementations of GAN components: losses, penalties, normalization techni… | 10 | 1815 | maintenance |
| yongzhuo/Keras-TextClassification A Keras-based library for Chinese text classification supporting long/short text, multi-label classification, and sentence pair similarity.… | 23 | 1807 | maintenance |
| minivision-ai/Silent-Face-Anti-Spoofing An open-source silent face anti-spoofing (liveness detection) project by MiniVision, providing model training code, data preprocessing, tes… | 32 | 1805 | maintenance |
| rosinality/vq-vae-2-pytorch A PyTorch implementation of VQ-VAE-2 for generating diverse high-fidelity images, including training scripts for the VQ-VAE and PixelSNAIL … | 32 | 1804 | maintenance |
| xmartlabs/Bender Bender is a Swift library that provides an abstraction layer over Apple's MetalPerformanceShaders for defining and running neural networks … | 23 | 1801 | maintenance |
| zai-org/CogView CogView is a 4-billion-parameter pretrained transformer model for text-to-image image generation, released alongside a NeurIPS 2021 paper. … | 32 | 1800 | maintenance |
| Trusted-AI/AIX360 AI Explainability 360 is an open-source Python library from IBM Research offering a comprehensive set of algorithms for interpreting and ex… | 66 | 1799 | maintenance |
| symisc/sod SOD is an embedded, cross-platform computer vision and machine learning library written in C, distributed as a single dependency-free amalg… | 23 | 1799 | maintenance |
| harskish/ganspace GANSpace is a research codebase implementing the NeurIPS 2020 paper on discovering interpretable controls for GANs via PCA in activation sp… | 32 | 1794 | maintenance |
| jkjung-avt/tensorrt_demos A collection of Python demo programs showing how to optimize and run deep learning models (YOLOv4, YOLOv3, SSD, MTCNN, GoogLeNet, MODNet) w… | 50 | 1792 | maintenance |
| dreamoving/dreamoving-project DreaMoving is the official implementation of a diffusion-based controllable video generation framework from Alibaba that produces high-qual… | 26 | 1790 | maintenance |
| MCG-NJU/VideoMAE Official PyTorch implementation of VideoMAE, a masked autoencoder method for data-efficient self-supervised video pre-training with video t… | 32 | 1788 | maintenance |
| scikit-learn-contrib/lightning A Python library for large-scale linear classification, regression, and ranking that follows scikit-learn API conventions. It supports dens… | 10 | 1783 | maintenance |
| jquesnelle/yarn Reference implementation of YaRN, an efficient method for extending the context window of large language models, published as an ICLR 2024 … | 29 | 1782 | maintenance |
| OryxProject/oryx Oryx 2 is a Java framework implementing the lambda architecture on Apache Spark and Apache Kafka, specialized for real-time large-scale mac… | 10 | 1782 | maintenance |
| idiap/fast-transformers A PyTorch library providing fast transformer implementations with efficient attention mechanisms such as linear attention, designed to scal… | 32 | 1780 | maintenance |
| bazingagin/npc_gzip A Python library and research codebase implementing the parameter-free 'kNN-gzip' text classification method from the ACL 2023 paper, which… | 20 | 1780 | maintenance |
| wkentaro/pytorch-fcn A PyTorch implementation of Fully Convolutional Networks (FCN) for semantic segmentation, including training code that reproduces the origi… | 10 | 1777 | maintenance |
| qiuqiangkong/audioset_tagging_cnn PANNs is a collection of large-scale pretrained CNN audio neural networks trained on AudioSet (5000 hours, 527 classes) for audio pattern r… | 32 | 1774 | maintenance |
| princeton-vl/CornerNet-Lite CornerNet-Lite is the official PyTorch implementation of the paper 'CornerNet-Lite: Efficient Keypoint Based Object Detection', providing t… | 32 | 1771 | maintenance |
| flowtyone/flowty-realtime-lcm-canvas A real-time sketch-to-image demo application that uses Latent Consistency Models (LCM) with a Gradio web UI. Users draw on a canvas and see… | 27 | 1770 | maintenance |
| facebookresearch/TensorComprehensions Tensor Comprehensions is a C++ library with a Python API that provides a domain-specific language for expressing machine learning workloads… | 10 | 1767 | maintenance |
| nywang16/Pixel2Mesh Pixel2Mesh is the official TensorFlow implementation of the ECCV 2018 paper that generates 3D mesh models from a single RGB image using a g… | 32 | 1766 | maintenance |
| cmu-db/noisepage NoisePage is a relational database management system from Carnegie Mellon University designed for autonomous, self-driving operation, with … | 10 | 1766 | maintenance |