function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| facebookresearch/pytorch_GAN_zoo A PyTorch toolbox of GAN implementations including Progressive Growing GAN, DCGAN, and StyleGAN (beta), plus tools like GDPP, AC-GAN condit… | 10 | 1627 | maintenance |
| facebookresearch/higher higher is a PyTorch library that enables higher-order gradients over losses spanning entire training loops rather than individual steps. It… | 10 | 1626 | maintenance |
| moses-smt/mosesdecoder Moses is a statistical machine translation (SMT) system that trains translation models from parallel corpora and decodes with efficient phr… | 29 | 1624 | maintenance |
| timoschick/pet Official implementation of Pattern-Exploiting Training (PET), a semi-supervised method that reformulates text classification and natural la… | 32 | 1622 | maintenance |
| Xtra-Computing/thundersvm ThunderSVM is a fast SVM library that accelerates support vector machine training on GPUs (via CUDA) and multi-core CPUs. It supports all L… | 23 | 1622 | maintenance |
| WisdomShell/codeshell CodeShell is a family of 7-billion-parameter code large language models developed by PKU-KCL, trained on 500 billion tokens with an 8192-to… | 27 | 1620 | maintenance |
| aimacode/aima-java A Java implementation of the algorithms from Russell and Norvig's 'Artificial Intelligence: A Modern Approach' (3rd edition), covering agen… | 62 | 1619 | maintenance |
| wangzhaode/mnn-llm mnn-llm is a C++ library for deploying large language models on-device using Alibaba's MNN inference engine, with demos for CLI, web, Andro… | 22 | 1617 | maintenance |
| lucidrains/PaLM-rlhf-pytorch A PyTorch library implementing Reinforcement Learning from Human Feedback (RLHF) on top of the PaLM transformer architecture, aiming to rep… | 78 | 7865 | experimental |
| dotnet/infer Infer.NET is a .NET framework for running Bayesian inference in graphical models, also usable for probabilistic programming. It compiles mo… | 74 | 1614 | maintenance |
| cdipaolo/goml goml is a machine learning library written entirely in Go, offering both traditional batch learning and online, stream-based learning via c… | 32 | 1614 | maintenance |
| allenai/bilm-tf A TensorFlow implementation of the bidirectional language model (biLM) used to compute ELMo deep contextualized word representations. It su… | 32 | 1612 | maintenance |
| DT42/BerryNet BerryNet is a deep learning gateway that turns edge devices like Raspberry Pi into intelligent, offline AI hubs for analyzing camera images… | 23 | 1610 | maintenance |
| divamgupta/stable-diffusion-tensorflow A TensorFlow/Keras implementation of Stable Diffusion with weights ported from the original implementation. It provides a Python API and Co… | 32 | 1608 | maintenance |
| experiencor/keras-yolo3 A Keras/TensorFlow implementation of YOLOv3 for object detection, supporting detection with pretrained weights, custom model training with … | 32 | 1608 | maintenance |
| Delta-ML/delta DELTA is a deep learning based end-to-end natural language and speech processing platform built on TensorFlow and Python 3. It provides one… | 10 | 1607 | maintenance |
| ialhashim/DenseDepth Official Keras/TensorFlow implementation (with experimental PyTorch and TF2 code) of the DenseDepth paper for high-quality monocular depth … | 32 | 1606 | maintenance |
| kakaobrain/fast-autoaugment Official PyTorch implementation of Fast AutoAugment (NeurIPS 2019), which learns image augmentation policies via density-matching search. I… | 32 | 1605 | maintenance |
| tensorflow/haskell Haskell bindings to the TensorFlow machine learning library, providing a typed API for building and training neural network models. It incl… | 10 | 1605 | maintenance |
| chandrikadeb7/Face-Mask-Detection A face mask detection system built with OpenCV and TensorFlow/Keras that uses deep learning (SSD MobileNetV2) to detect whether people are … | 23 | 1604 | maintenance |
| PeterWang512/FALdetector FALdetector is the official PyTorch implementation of the ICCV 2019 paper 'Detecting Photoshopped Faces by Scripting Photoshop'. It provide… | 32 | 1600 | maintenance |
| jiupinjia/stylized-neural-painting Official PyTorch implementation of the CVPR 2021 paper 'Stylized Neural Painting', which translates photos into vectorized painting artwork… | 32 | 1600 | maintenance |
| ycjuan/libffm LIBFFM is a C++ library implementing field-aware factorization machines (FFM), a model especially effective for click-through-rate (CTR) pr… | 10 | 1599 | maintenance |
| cvg/nice-slam NICE-SLAM is a dense RGB-D SLAM system that combines neural implicit decoders with hierarchical grid-based scene representations, published… | 32 | 1596 | maintenance |
| tmikolov/word2vec The original C implementation of word2vec, which trains word embeddings using the Continuous Bag-of-Words (CBOW) and Skip-gram neural netwo… | 32 | 1593 | maintenance |
| boudinfl/pke pke is an open-source Python toolkit for keyphrase extraction that provides an end-to-end pipeline where each component (candidate selectio… | 23 | 1590 | maintenance |
| google-research/parti Parti is Google Research's Pathways Autoregressive Text-to-Image model, which generates high-fidelity photorealistic images from text by tr… | 32 | 1589 | maintenance |
| google/uis-rnn A Python library implementing the Unbounded Interleaved-State Recurrent Neural Network (UIS-RNN) algorithm for segmenting and clustering se… | 10 | 1588 | maintenance |
| NVlabs/FUNIT FUNIT is NVIDIA's PyTorch implementation of a few-shot unsupervised image-to-image translation model (ICCV 2019) that can translate images … | 32 | 1586 | maintenance |
| PaddlePaddle/PGL Paddle Graph Learning (PGL) is an efficient and flexible graph learning framework built on PaddlePaddle. It supports heterogeneous graph le… | 23 | 1585 | maintenance |
| jundongl/scikit-feature scikit-feature is an open-source Python feature selection library built on scikit-learn, NumPy, and SciPy. It implements around 40 popular … | 32 | 1584 | maintenance |
| lufficc/SSD A high-quality, fast, modular reference implementation of the SSD (Single Shot MultiBox Detector) object detection model in PyTorch. It sup… | 23 | 1584 | maintenance |
| PrithivirajDamodaran/Gramformer Gramformer is a Python library that exposes interfaces to detect, highlight, and correct grammatical errors in natural language text, built… | 23 | 1584 | maintenance |
| liuhuanyong/CrimeKgAssitant A Chinese legal intelligence application built around a crime knowledge graph covering 856 crime types. It provides crime type prediction f… | 32 | 1583 | maintenance |
| LingBot-World LingBot-World is an open-source world model that generates high-fidelity, interactive simulated environments from video generation, support… | 54 | 1581 | maintenance |
| D-X-Y/AutoDL-Projects AutoDL-Projects is a PyTorch library implementing automated deep learning algorithms, including neural architecture search (NAS) methods li… | 32 | 1581 | maintenance |
| TsinghuaAI/CPM-1-Generate CPM-Generate provides inference code for CPM-LM, a 2.6B-parameter Chinese pre-trained language model based on the GPT-2 architecture and Me… | 32 | 1579 | maintenance |
| melodyguan/enas Authors' TensorFlow implementation of Efficient Neural Architecture Search (ENAS), which discovers neural network architectures via paramet… | 32 | 1578 | maintenance |
| xinntao/EDVR EDVR is the winning solution of the NTIRE19 video restoration challenges, built on enhanced deformable convolutional networks. The repo is … | 32 | 1578 | maintenance |
| Temporal Segment Networks (TSN) Official code and pretrained models for Temporal Segment Networks (TSN), a deep learning framework for video action recognition published a… | 32 | 1577 | maintenance |
| maxpumperla/elephas Elephas is a Python library that extends Keras to run distributed deep learning training on Apache Spark. It serializes Keras models on the… | 23 | 1577 | maintenance |
| Kotlin/kotlindl KotlinDL is a high-level deep learning framework written in Kotlin and inspired by Keras, built on top of the TensorFlow Java API and ONNX … | 23 | 1577 | maintenance |
| vturrisi/solo-learn solo-learn is a Python library of state-of-the-art self-supervised methods for unsupervised visual representation learning, built on PyTorc… | 65 | 1576 | maintenance |
| OsciiArt/DeepAA DeepAA is a Python project that generates ASCII art from grayscale line images using convolutional neural networks built with TensorFlow an… | 32 | 1576 | maintenance |
| Ewenwan/ORB_SLAM2_SSD_Semantic A C++ research project extending ORB_SLAM2 with dynamic object detection and semantic mapping. It combines SSD-based object detection (Mobi… | 32 | 1575 | maintenance |
| cmdbug/YOLOv5_NCNN A mobile demo application that deploys the ncnn inference framework on Android and iOS, running a variety of computer vision models includi… | 32 | 1573 | maintenance |
| HIPS/Spearmint Spearmint is a Python package for Bayesian optimization that automatically tunes experiment parameters to minimize an objective in as few r… | 32 | 1572 | maintenance |
| kumar-shridhar/PyTorch-BayesianCNN A PyTorch library implementing Bayesian convolutional neural networks with variational inference via Bayes by Backprop. It provides drop-in… | 32 | 1570 | maintenance |
| sniklaus/3d-ken-burns A PyTorch reference implementation of the 3D Ken Burns Effect from a Single Image paper, which animates a still photo with a virtual camera… | 70 | 1569 | maintenance |
| HumanAIGC/EMO EMO (Emote Portrait Alive) is a research codebase from Alibaba's Institute for Intelligent Computing that generates expressive talking port… | 25 | 7590 | experimental |
| AlexHex7/Non-local_pytorch A PyTorch implementation of the Non-local Neural Block from the paper 'Non-local Neural Networks', providing multiple variants (concatenati… | 32 | 1564 | maintenance |
| NVlabs/noise2noise Official TensorFlow implementation of the Noise2Noise ICML 2018 paper, which trains image restoration networks using only corrupted (noisy)… | 32 | 1564 | maintenance |
| tobegit3hub/advisor Advisor is an open-source implementation of Google Vizier, a hyperparameter tuning system for black-box optimization. It provides a server … | 32 | 1563 | maintenance |
| thu-ml/prolificdreamer Official PyTorch implementation of ProlificDreamer, a NeurIPS 2023 method for high-fidelity text-to-3D generation using Variational Score D… | 29 | 1562 | maintenance |
| facebookresearch/DeepSDF DeepSDF is Facebook Research's official implementation of the CVPR 2019 paper on learning continuous signed distance functions for 3D shape… | 10 | 1562 | maintenance |
| genforce/interfacegan InterFaceGAN is a Python research library that interprets the latent space of pretrained GANs (PGGAN, StyleGAN) to find semantic subspaces … | 32 | 1561 | maintenance |
| msracver/FCIS FCIS is the official MXNet implementation of the CVPR 2017 paper 'Fully Convolutional Instance-aware Semantic Segmentation', which won firs… | 32 | 1561 | maintenance |
| QingyongHu/RandLA-Net Official TensorFlow implementation of RandLA-Net, a neural architecture for efficient semantic segmentation of large-scale 3D point clouds,… | 32 | 1561 | maintenance |
| V2AI/Det3D Det3D is a PyTorch-based toolbox for 3D object detection from point clouds, offering implementations of models like PointPillars, SECOND, a… | 32 | 1561 | maintenance |
| IDEA-Research/MaskDINO Official PyTorch implementation of Mask DINO, a unified transformer-based framework for object detection and segmentation, built on detectr… | 23 | 1558 | maintenance |
| petercorke/robotics-toolbox-matlab A MATLAB toolbox providing robotics functionality including kinematics and dynamics of serial-link manipulators, mobile robot motion models… | 23 | 1556 | maintenance |
| vlawhern/arl-eegmodels A collection of well-validated convolutional neural network models (EEGNet, DeepConvNet, ShallowConvNet) for EEG signal classification, imp… | 32 | 1555 | maintenance |
| bfelbo/DeepMoji DeepMoji is a deep learning model trained on 1.2 billion tweets with emojis to understand emotional language, with code for scoring texts, … | 32 | 1553 | maintenance |
| vt-vl-lab/FGVC FGVC is a PyTorch implementation of the ECCV 2020 paper 'Flow-edge Guided Video Completion'. It completes missing regions in videos by extr… | 32 | 1553 | maintenance |
| pytorch/QNNPACK QNNPACK is a mobile-optimized C library of high-performance kernels for 8-bit quantized neural network operators such as convolution, pooli… | 10 | 1552 | maintenance |
| Tencent/TurboTransformers A fast, user-friendly C++/Python runtime for transformer model inference (BERT, ALBERT, RoBERTa, GPT2, decoders) on CPU and GPU. It works a… | 37 | 1550 | maintenance |
| DevashishPrasad/CascadeTabNet CascadeTabNet is a PyTorch/mmdetection implementation of a CVPR 2020 paper for end-to-end table detection and structure recognition from im… | 32 | 1549 | maintenance |
| JiaRenChang/PSMNet PSMNet is the official PyTorch implementation of the CVPR 2018 paper 'Pyramid Stereo Matching Network' for stereo depth estimation. It uses… | 32 | 1549 | maintenance |
| JingyunLiang/VRT VRT is the official PyTorch implementation of the paper 'VRT: A Video Restoration Transformer', a transformer-based model for video restora… | 23 | 1548 | maintenance |
| Javacr/PyQt5-YOLOv5 A desktop GUI application built with PyQt5 that wraps YOLOv5 (v6.1) object detection models. It supports running detection on images, video… | 32 | 1547 | maintenance |
| caogang/wgan-gp A PyTorch implementation of the 'Improved Training of Wasserstein GANs' (WGAN-GP) paper, covering toy datasets, MNIST, CIFAR-10, and charac… | 32 | 1547 | maintenance |
| YoYo000/MVSNet MVSNet is a deep learning architecture for depth map inference from unstructured multi-view images, and R-MVSNet is its recurrent extension… | 32 | 1546 | maintenance |
| google-research/big_transfer Official repository for the Big Transfer (BiT) paper, providing ResNet models pre-trained on ImageNet and ImageNet-21k for transfer learnin… | 10 | 1541 | maintenance |
| dandelin/ViLT Official PyTorch code for the ICML 2021 paper ViLT, a vision-and-language transformer that performs multimodal pre-training without convolu… | 23 | 1536 | maintenance |
| AxeldeRomblay/MLBox MLBox is a Python automated machine learning (AutoML) library that handles data preprocessing, feature selection, leak detection, and hyper… | 23 | 1536 | maintenance |
| poloclub/ganlab GAN Lab is an interactive, browser-based visualization tool for learning and experimenting with Generative Adversarial Networks. Built with… | 56 | 1535 | maintenance |
| gpleiss/efficient_densenet_pytorch A memory-efficient PyTorch implementation of DenseNets that uses gradient checkpointing to reduce feature map memory consumption from quadr… | 32 | 1534 | maintenance |
| AsuharietYgvar/AppleNeuralHash2ONNX A Python tool that converts Apple's NeuralHash perceptual hashing model (used in Apple's CSAM Detection system) into ONNX format. It includ… | 32 | 1534 | maintenance |
| yongzhuo/nlp_xiaojiang A Chinese natural language processing toolkit covering retrieval-based chatbots, text classification, NER (BERT+BiLSTM+CRF), sentence embed… | 32 | 1534 | maintenance |
| crownpku/Rasa_NLU_Chi A Chinese-language fork of Rasa NLU that turns Chinese natural language into structured intent and entity data using MITIE, Jieba tokenizat… | 32 | 1532 | maintenance |
| Stability-AI/sd3.5 A minimal inference-only reference implementation of Stable Diffusion 3.5 and SD3, including text encoders, VAE decoder, MM-DiT core, and S… | 22 | 1532 | maintenance |
| lucidrains/lambda-networks A PyTorch implementation of Lambda Networks, featuring the λ layer that models long-range interactions by transforming contexts into linear… | 23 | 1529 | maintenance |
| zhan-xu/RigNet RigNet is a PyTorch implementation of the SIGGRAPH 2020 paper on neural rigging for articulated 3D characters. It takes a character mesh as… | 32 | 1521 | maintenance |
| yu4u/age-gender-estimation A Keras/TensorFlow implementation of a convolutional neural network that estimates age and gender from face images, trained on the IMDB-WIK… | 23 | 1520 | maintenance |
| shaoxiongji/federated-learning A PyTorch implementation of the FedAvg federated learning algorithm reproducing the McMahan et al. AISTATS 2017 paper. It includes experime… | 32 | 1516 | maintenance |
| tanluren/yolov3-channel-and-layer-pruning A Python toolkit built on ultralytics/yolov3 that implements channel pruning, layer pruning, and knowledge distillation for YOLOv3/v4 (incl… | 32 | 1515 | maintenance |
| IBM/pytorch-seq2seq An open-source framework for building sequence-to-sequence (seq2seq) models in PyTorch, with modular and extensible components for models, … | 10 | 1515 | maintenance |
| junyanz/BicycleGAN A PyTorch implementation of BicycleGAN, a model for multimodal image-to-image translation that generates diverse outputs from a single inpu… | 32 | 1514 | maintenance |
| megvii-model/ShuffleNet-Series A collection of ShuffleNet-series efficient convolutional neural network models (V1, V2, V2+, Large, ExLarge) plus NAS-derived backbones li… | 32 | 1513 | maintenance |
| Eric-mingjie/rethinking-network-pruning A PyTorch research codebase reproducing the ICLR 2019 paper 'Rethinking the Value of Network Pruning', which shows pruned models trained fr… | 32 | 1512 | maintenance |
| open-mmlab/Multimodal-GPT Multimodal-GPT is an open-source project for training a multimodal chatbot that combines vision and language instructions, built on OpenFla… | 30 | 1512 | maintenance |
| maderix/ANE A research project demonstrating backpropagation and neural network training directly on Apple's Neural Engine using reverse-engineered pri… | 47 | 7255 | experimental |
| krasserm/super-resolution A TensorFlow 2.x implementation of EDSR, WDSR, and SRGAN models for single image super-resolution, with a high-level training API and DIV2K… | 32 | 1511 | maintenance |
| marsauto/europilot Europilot is a Python toolkit that bridges Euro Truck Simulator 2 with deep-learning frameworks, enabling end-to-end self-driving research … | 32 | 1511 | maintenance |
| akai-katto/dandere2x Dandere2x is a tool that speeds up waifu2x video upscaling by exploiting temporal redundancy in anime footage, using P-frame compression so… | 23 | 1510 | maintenance |
| ZJU-OmniAI/Data-Copilot Data-Copilot is an LLM-based data analysis agent that autonomously designs and invokes interface tools to query, process, analyze, predict,… | 29 | 1509 | maintenance |
| compphoto/BoostingMonocularDepth A Python research implementation for boosting monocular depth estimation to high resolution using a double-estimation merging operator, sup… | 32 | 1507 | maintenance |
| yulunzhang/RCAN PyTorch implementation of RCAN, a very deep residual channel attention network for single image super-resolution from an ECCV 2018 paper. I… | 39 | 1506 | maintenance |
| nlpyang/BertSum BertSum is the official PyTorch implementation of the paper 'Fine-tune BERT for Extractive Summarization', providing preprocessing pipeline… | 32 | 1506 | maintenance |
| sigsep/open-unmix-pytorch Open-Unmix is a PyTorch deep learning reference implementation for music source separation, with pre-trained models that split pop music in… | 23 | 1504 | maintenance |