function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| magicleap/SuperPointPretrainedNetwork A PyTorch pre-trained implementation of the SuperPoint fully convolutional neural network for real-time interest point detection and descri… | 32 | 2189 | maintenance |
| alibaba/EasyNLP EasyNLP is a comprehensive PyTorch-based NLP toolkit from Alibaba that provides training, inference, and deployment for pre-trained languag… | 23 | 2186 | maintenance |
| ming024/FastSpeech2 A PyTorch implementation of Microsoft's FastSpeech 2 text-to-speech model, supporting English and Mandarin with single- and multi-speaker s… | 32 | 2185 | maintenance |
| tianweiy/CenterPoint Official PyTorch implementation of CenterPoint, a CVPR 2021 method that performs 3D object detection and tracking from LiDAR point clouds b… | 23 | 2185 | maintenance |
| githubharald/SimpleHTR A Handwritten Text Recognition (HTR) system implemented in TensorFlow that recognizes text from images of single words or text lines, train… | 72 | 2183 | maintenance |
| maxpumperla/hyperas Hyperas is a Python library that wraps Hyperopt to provide convenient hyperparameter optimization for Keras models. Users annotate model pa… | 32 | 2173 | maintenance |
| vchoutas/smplify-x SMPLify-X is the official PyTorch implementation of the CVPR 2019 paper 'Expressive Body Capture: 3D Hands, Face, and Body from a Single Im… | 32 | 2169 | maintenance |
| open-mmlab/mmrotate MMRotate is an open-source PyTorch toolbox for rotated object detection, part of the OpenMMLab project. It provides modular components, mul… | 23 | 2164 | maintenance |
| SheffieldML/GPy GPy is a Python framework for Gaussian process modeling, providing kernels, regression, classification, and sparse GP methods. It is a rese… | 97 | 2163 | maintenance |
| jeonsworld/ViT-pytorch A PyTorch reimplementation of Google's Vision Transformer (ViT) for image classification, based on the paper 'An Image is Worth 16x16 Words… | 32 | 2163 | maintenance |
| bubbliiiing/yolov4-pytorch A PyTorch implementation of the YOLOv4 object detection model with full training, prediction, and evaluation scripts. It supports training … | 23 | 2162 | maintenance |
| facebookresearch/pycls pycls is a PyTorch-based codebase for image classification research developed by Facebook AI Research. It provides implementations of stand… | 10 | 2161 | maintenance |
| kingyiusuen/image-to-latex A PyTorch application that converts images of LaTeX math equations into LaTeX code using a ResNet-18 encoder and Transformer decoder traine… | 32 | 2158 | maintenance |
| nvdla/hw The hardware release of NVIDIA's open-source Deep Learning Accelerator (NVDLA), containing Verilog RTL, a C-model, testbenches, and synthes… | 32 | 2146 | maintenance |
| satoshiiizuka/siggraph2016_colorization A Torch7 implementation of the SIGGRAPH 2016 paper 'Let there be Color!' that automatically colorizes grayscale images using a deep network… | 32 | 2142 | maintenance |
| Kkevsterrr/geneva Geneva is a research tool from the University of Maryland that uses a genetic algorithm to automatically evolve packet-manipulation strateg… | 32 | 2142 | maintenance |
| SeanNaren/deepspeech.pytorch A PyTorch implementation of the DeepSpeech2 speech recognition model, built on PyTorch Lightning, supporting training, testing, and inferen… | 23 | 2136 | maintenance |
| RJT1990/pyflux PyFlux is an open source Python library for time series analysis and forecasting, offering a broad range of models such as ARIMA, GARCH, GA… | 32 | 2135 | maintenance |
| glouw/tinn Tinn is a tiny, dependency-free neural network library written in under 200 lines of C99, implementing a feedforward network with one hidde… | 32 | 2134 | maintenance |
| chuanqi305/MobileNet-SSD A Caffe implementation of the MobileNet-SSD object detection network with pretrained weights on the VOC0712 dataset achieving mAP of 0.727.… | 46 | 2127 | maintenance |
| magenta/magenta-js Magenta.js is a collection of TypeScript libraries for running inference with pre-trained Magenta machine learning models directly in the b… | 72 | 2125 | maintenance |
| rlworkgroup/garage garage is a Python toolkit for developing and evaluating reinforcement learning algorithms, with a library of state-of-the-art RL algorithm… | 23 | 2125 | maintenance |
| AIZOOTech/FaceMaskDetection An open-source face mask detection project providing a lightweight SSD-based model (1.01M parameters) with inference code for PyTorch, Tens… | 32 | 2124 | maintenance |
| atriumlts/subpixel A TensorFlow reimplementation of the efficient sub-pixel convolutional neural network (ESPCN) for single-image super-resolution, based on S… | 32 | 2123 | maintenance |
| songyouwei/ABSA-PyTorch PyTorch implementations of aspect-based sentiment analysis (ABSA) models, including attention-based and BERT-based approaches for aspect-te… | 10 | 2114 | maintenance |
| bubbliiiing/yolo3-pytorch A PyTorch implementation of the YOLOv3 object detection model with full training, prediction, and evaluation scripts. It supports training … | 23 | 2113 | maintenance |
| facebookresearch/SentEval SentEval is a Python library from Facebook Research for evaluating the quality of sentence embeddings via a suite of downstream transfer ta… | 10 | 2110 | maintenance |
| sfujim/TD3 The author's official PyTorch implementation of Twin Delayed Deep Deterministic Policy Gradients (TD3), a reinforcement learning algorithm … | 32 | 2106 | maintenance |
| bgshih/crnn An implementation of the Convolutional Recurrent Neural Network (CRNN), combining CNN, RNN, and CTC loss for image-based sequence recogniti… | 32 | 2105 | maintenance |
| asappresearch/sru SRU is a PyTorch library implementing the Simple Recurrent Unit, a highly parallelizable RNN that runs 10-16x faster than cuDNN LSTM with c… | 23 | 2105 | maintenance |
| dog-qiuqiu/Yolo-Fastest Yolo-Fastest is an ultra-lightweight YOLO-based object detection algorithm and model zoo, with only ~250 MFLOPs and a 666KB ncnn model. It … | 23 | 2103 | maintenance |
| jalammar/ecco Ecco is a Python library for exploring and explaining Transformer-based NLP language models through interactive visualizations inside Jupyt… | 23 | 2103 | maintenance |
| facebookresearch/chameleon Repository for Meta Chameleon, an early-fusion token-based mixed-modal foundation model that understands and generates interleaved images a… | 10 | 2103 | maintenance |
| qubvel/efficientnet A Keras and TensorFlow Keras reimplementation of the EfficientNet convolutional neural network family (B0-B7), including ImageNet-pretraine… | 23 | 2100 | maintenance |
| archinetai/audio-diffusion-pytorch A PyTorch library for audio generation using diffusion models, supporting unconditional and text-conditional generation, diffusion autoenco… | 23 | 2096 | maintenance |
| facebookresearch/swav PyTorch implementation of SwAV, a self-supervised method for pre-training convolutional networks without labels by swapping cluster assignm… | 10 | 2096 | maintenance |
| GantMan/nsfw_model A Keras/TensorFlow deep learning model that classifies images into NSFW categories (drawings, hentai, neutral, porn, sexy) with ~93% accura… | 23 | 2091 | maintenance |
| ai-forever/ru-gpts A repository of Russian GPT-3 language models (ruGPT3XL/Large/Medium/Small and ruGPT2Large) with usage and fine-tuning examples. It provide… | 32 | 2087 | maintenance |
| generalized-intelligence/GAAS GAAS (Generalized Autonomy Aviation System) is an open-source C++ framework for fully autonomous VTOL aircraft and drones, built around lid… | 23 | 2084 | maintenance |
| THUDM/P-tuning-v2 P-tuning v2 is a Python implementation of deep prompt tuning, applying trainable continuous prompts at every transformer layer so prompt tu… | 32 | 2077 | maintenance |
| tianqiraf/DouZero_For_HappyDouDiZhu A Python desktop application that applies the DouZero reinforcement-learning Dou Dizhu (Chinese card game) AI to the popular Happy DouDiZhu… | 23 | 2068 | maintenance |
| koth/kcws A deep learning Chinese word segmentation toolkit implemented in C++ with TensorFlow, supporting BiLSTM+CRF and IDCNN+CRF models, plus part… | 23 | 2067 | maintenance |
| RootKit-Org/AI-Aimbot A Python application that uses a YOLOv5-based computer vision model to detect humanoid targets on screen and assist aiming in FPS games lik… | 10 | 2067 | maintenance |
| ChenglongChen/tensorflow-DeepFM A TensorFlow implementation of the DeepFM model combining factorization machines and deep neural networks for click-through-rate (CTR) pred… | 32 | 2066 | maintenance |
| flowdriveai/flowpilot Flowpilot is an open source driver assistance system built on top of openpilot that provides adaptive cruise control, automated lane center… | 23 | 2065 | maintenance |
| ozan-oktay/Attention-Gated-Networks A PyTorch implementation of attention gates for convolutional neural networks, applied to U-Net and VGG-16 architectures. It targets medica… | 32 | 2064 | maintenance |
| danmacnish/cartoonify A Python application that converts photographs into hand-drawn-style cartoons using a neural network for object recognition combined with t… | 32 | 2051 | maintenance |
| cfzd/Ultra-Fast-Lane-Detection A PyTorch implementation of the ECCV 2020 paper 'Ultra Fast Structure-aware Deep Lane Detection', providing training and evaluation code fo… | 32 | 2049 | maintenance |
| bytewax/bytewax Bytewax is a Python-first framework with a Rust-based distributed engine for stateful event and stream processing, inspired by Apache Flink… | 62 | 2048 | maintenance |
| amazon-science/auto-cot Official implementation of Auto-CoT (ICLR 2023), a method that automatically constructs chain-of-thought demonstrations for large language … | 32 | 2048 | maintenance |
| zhanghang1989/PyTorch-Encoding A PyTorch computer vision toolkit with optimized CUDA backend, providing encoding layers, synchronized multi-GPU batch normalization, and p… | 23 | 2044 | maintenance |
| mut-ex/gligen-gui A web-based graphical interface for GLIGEN, a technique for controlling object placement in Stable Diffusion text-to-image generation, buil… | 25 | 2043 | maintenance |
| alibaba/AliceMind AliceMind is Alibaba's collection of pre-trained encoder-decoder language models and related NLP techniques, including StructBERT, PALM, VE… | 23 | 2041 | maintenance |
| salu133445/musegan MuseGAN is a TensorFlow-based generative adversarial network for generating polyphonic, multitrack music (bass, drums, guitar, piano, strin… | 32 | 2040 | maintenance |
| AnimeGAN AnimeGAN is a family of TensorFlow-based generative adversarial network models (v1, v2, v3) that convert landscape photos and videos into a… | 40 | 2037 | maintenance |
| apple/ml-fastvit Official PyTorch implementation of FastViT, a fast hybrid vision transformer architecture using structural reparameterization, published at… | 28 | 2029 | maintenance |
| pkhungurn/talking-head-anime-demo Demo applications for the 'Talking Head Anime from a Single Image' research project, built with PyTorch. It includes a manual poser for ani… | 32 | 2027 | maintenance |
| darglein/ADOP ADOP is a point-based differentiable neural rendering pipeline for scene refinement and novel view synthesis, implemented in C++/CUDA with … | 23 | 2027 | maintenance |
| mingyuliutw/UNIT UNIT is NVIDIA's research implementation of unsupervised image-to-image translation networks (NIPS 2017), translating images across domains… | 32 | 2025 | maintenance |
| WongKinYiu/ScaledYOLOv4 PyTorch implementation of Scaled-YOLOv4, which scales the Cross Stage Partial Network (CSP) across model sizes from tiny to P7 for object d… | 23 | 2024 | maintenance |
| xlang-ai/instructor-embedding INSTRUCTOR is an instruction-finetuned text embedding model and Python library that generates task-tailored embeddings by simply providing … | 32 | 2023 | maintenance |
| lucasb-eyer/pydensecrf PyDenseCRF is a Cython-based Python wrapper around Philipp Krähenbühl's fully-connected (dense) conditional random field implementation wit… | 32 | 2021 | maintenance |
| bytedance/piano_transcription A PyTorch implementation of ByteDance's high-resolution piano transcription system that converts piano audio recordings into MIDI files. It… | 10 | 2021 | maintenance |
| neulab/prompt2model Prompt2Model is a Python library that takes a natural language task description (like an LLM prompt) and automatically generates a small, s… | 21 | 2018 | maintenance |
| tinghuiz/SfMLearner SfMLearner is a TensorFlow implementation of the CVPR 2017 paper 'Unsupervised Learning of Depth and Ego-Motion from Video'. It trains mode… | 32 | 2016 | maintenance |
| baidu/Senta Senta is Baidu's open-source sentiment analysis system built on the SKEP (Sentiment Knowledge Enhanced Pre-training) model. It provides pre… | 32 | 2016 | maintenance |
| NVlabs/alpamayo NVIDIA Alpamayo 1 is an open 10B-parameter reasoning vision-language-action (VLA) model for autonomous vehicles that pairs driving trajecto… | 59 | 2014 | maintenance |
| hila-chefer/Transformer-Explainability Official PyTorch implementation of the CVPR 2021 paper 'Transformer Interpretability Beyond Attention Visualization', providing relevance-b… | 32 | 2014 | maintenance |
| open-mmlab/mmgeneration MMGeneration is a PyTorch-based toolkit from OpenMMLab for training and applying generative models, with a focus on GANs and support for di… | 23 | 2012 | maintenance |
| minimaxir/person-blocker A Python command-line script that automatically censors ('blocks') people and up to 80 other object types in images using a pretrained Mask… | 32 | 2011 | maintenance |
| brannondorsey/PassGAN A Python command-line implementation of the PassGAN paper that uses a Wasserstein GAN to generate password guesses, with a pretrained model… | 23 | 2011 | maintenance |
| Zz-ww/SadTalker-Video-Lip-Sync A Python tool built on SadTalker that generates lip-synced video from an audio file and a source video, with configurable face/lip region e… | 30 | 2010 | maintenance |
| facebookresearch/Detic Detic is the official code release for the ECCV 2022 paper 'Detecting Twenty-thousand Classes using Image-level Supervision'. It is an open… | 32 | 2009 | maintenance |
| divelab/DIG DIG (Dive into Graphs) is a Python library built on PyTorch Geometric that provides a unified testbed for advanced graph deep learning rese… | 23 | 2009 | maintenance |
| robbiebarrat/art-DCGAN A modified Torch implementation of DCGAN focused on generating artworks, with support for 128x128 images and resuming training from checkpo… | 32 | 2003 | maintenance |
| WongKinYiu/yolor PyTorch implementation of the YOLOR paper 'You Only Learn One Representation: Unified Network for Multiple Tasks', a real-time object detec… | 23 | 2003 | maintenance |
| haitongli/knowledge-distillation-pytorch A PyTorch framework for running knowledge distillation experiments, supporting both shallow (teacher-to-small-CNN) and deep distillation on… | 32 | 1999 | maintenance |
| JDAI-CV/FaceX-Zoo FaceX-Zoo is a PyTorch toolbox for face recognition that provides training modules with various state-of-the-art supervisory heads and back… | 32 | 1997 | maintenance |
| vsitzmann/siren Official PyTorch implementation of SIREN, a neural network architecture using periodic (sine) activation functions for implicit neural repr… | 32 | 1994 | maintenance |
| HobbitLong/PyContrast PyContrast is a PyTorch library providing reference implementations of contrastive learning methods such as InstDis, CMC, and MoCo, along w… | 32 | 1993 | maintenance |
| NVlabs/edm Official PyTorch implementation of the NeurIPS 2022 paper 'Elucidating the Design Space of Diffusion-Based Generative Models' (EDM) by NVID… | 32 | 1993 | maintenance |
| salesforce/awd-lstm-lm A PyTorch toolkit for training LSTM and QRNN language models, implementing Salesforce Research's AWD-LSTM papers. It supports word-level an… | 10 | 1990 | maintenance |
| databricks/spark-deep-learning Deep Learning Pipelines for Apache Spark, now reduced to the HorovodRunner component for distributed deep learning training via Horovod on … | 23 | 1987 | maintenance |
| albertpumarola/GANimation Official PyTorch implementation of GANimation, an ECCV'18 research paper that animates facial expressions in a single image using a GAN con… | 32 | 1986 | maintenance |
| apple/ml-cvnets CVNets is Apple's open-source PyTorch library for training computer vision networks, covering classification, detection, segmentation, vide… | 32 | 1984 | maintenance |
| ZiyaoGeng/RecLearn RecLearn is a Python/TensorFlow2.x library implementing a wide collection of recommendation algorithms, covering both matching (Top-K recom… | 23 | 1984 | maintenance |
| HazyResearch/deepdive DeepDive is a Stanford-developed system for extracting structured data from unstructured sources and building knowledge bases using distant… | 23 | 1979 | maintenance |
| Tencent/ObjectDetection-OneStageDet OneStageDet (OSD) is a PyTorch-based single-stage object detection framework from Tencent implementing YOLOv2 and YOLOv3 with plans to unif… | 45 | 1978 | maintenance |
| ICT-BDA/EasyML EasyML is a general-purpose dataflow-based machine learning platform where tasks are defined as directed acyclic graphs of operations. It i… | 23 | 1976 | maintenance |
| WuJie1010/Facial-Expression-Recognition.Pytorch A PyTorch implementation of CNN-based facial expression recognition achieving state-of-the-art accuracy on FER2013 (73.112%) and CK+ (94.64… | 32 | 1975 | maintenance |
| r9y9/deepvoice3_pytorch A PyTorch implementation of Deep Voice 3 and related convolutional neural network-based text-to-speech synthesis models. It includes traini… | 23 | 1975 | maintenance |
| SummitKwan/transparent_latent_gan TL-GAN is a Python/TensorFlow project that makes a GAN's latent space transparent by discovering feature axes, enabling controlled image sy… | 32 | 1973 | maintenance |
| google/nerfies Nerfies is the official JAX implementation of Deformable Neural Radiance Fields, which reconstructs non-rigidly deforming scenes from casua… | 10 | 1973 | maintenance |
| deepseek-ai/DeepSeek-MoE DeepSeekMoE 16B is a Mixture-of-Experts language model with fine-grained expert segmentation and shared expert isolation, released with bas… | 26 | 1972 | maintenance |
| Hitachi-Automotive-And-Industry-Lab/semantic-segmentation-editor A web-based labeling tool for creating AI training datasets from bitmap images (.jpg/.png) and point clouds (.pcd). Built as a Meteor app w… | 23 | 1970 | maintenance |
| JavierAntoran/Bayesian-Neural-Networks A collection of PyTorch implementations of Bayesian neural network approximate inference methods, including Bayes by Backprop, MC Dropout, … | 32 | 1969 | maintenance |
| nobody132/masr MASR is an end-to-end Mandarin Chinese automatic speech recognition project built on a gated convolutional neural network (similar to Wav2L… | 23 | 1968 | maintenance |
| Music-and-Culture-Technology-Lab/omnizart Omnizart is a Python library and CLI for automatic music transcription, transcribing pitched instruments, vocal melody, chords, drum events… | 88 | 1966 | maintenance |
| MojoJolo/textteaser TextTeaser is an automatic text summarization algorithm written in Scala that combines natural language processing and machine learning to … | 32 | 1966 | maintenance |
| FoundationVision/LlamaGen LlamaGen is a family of autoregressive image generation models that apply next-token prediction, as in Llama-style LLMs, to visual generati… | 24 | 1966 | maintenance |