function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| huoyijie/AdvancedEAST AdvancedEAST is a deep learning algorithm for detecting text in scene images, built on the EAST architecture with improvements for more acc… | 32 | 1226 | maintenance |
| liuhuanyong/ComplexEventExtraction A Python library for Chinese compound event extraction that identifies conditional, causal, sequential, and adversative events using explic… | 32 | 1226 | maintenance |
| PlayVoice/vits_chinese A Chinese text-to-speech library combining VITS with BERT-based prosody embeddings and NaturalSpeech infer-loss features, supporting ONNX c… | 23 | 1226 | maintenance |
| bheinzerling/bpemb BPEmb is a collection of pre-trained subword embeddings in 275 languages based on Byte-Pair Encoding, trained on Wikipedia, distributed as … | 32 | 1224 | maintenance |
| tylin/coco-caption The official evaluation code for the Microsoft COCO image captioning benchmark, implementing metrics such as BLEU, METEOR, ROUGE-L, CIDEr, … | 32 | 1224 | maintenance |
| google-research/fixmatch Official research code for FixMatch, a semi-supervised learning method that combines consistency regularization with confidence-based pseud… | 10 | 1224 | maintenance |
| ml4a/ml4a-ofx A collection of openFrameworks applications in C++ for real-time interactive machine learning, aimed at artists and creative coders. It inc… | 23 | 1223 | maintenance |
| iqiyi/FASPell FASPell is a Python-based Chinese spell checker based on the DAE-Decoder paradigm, published at the EMNLP 2019 W-NUT workshop. It detects a… | 32 | 1222 | maintenance |
| xvjiarui/GCNet Official PyTorch implementation of GCNet (Global Context Networks), a paper combining ideas from Non-Local Networks and Squeeze-Excitation … | 32 | 1220 | maintenance |
| Jhy1993/HAN Reference implementation of the Heterogeneous Graph Attention Network (HAN) model from the WWW 2019 paper, built on top of GAT in Python. I… | 32 | 1219 | maintenance |
| xingyizhou/CenterNet2 CenterNet2 is a research implementation of probabilistic two-stage object detection built on detectron2, where a class-agnostic one-stage C… | 32 | 1219 | maintenance |
| peng-zhihui/XUAN XUAN-Bike is an open-source self-driving (self-balancing) bicycle project by peng-zhihui, containing hardware design files including contro… | 32 | 5586 | experimental |
| getnamo/TensorFlow-Unreal An Unreal Engine plugin that wraps TensorFlow machine learning operations as Actor Components accessible from C++, Blueprints, and Python. … | 23 | 1218 | maintenance |
| ucbdrive/few-shot-object-detection FsDet is the official implementation of the ICML 2020 paper 'Frustratingly Simple Few-Shot Object Detection' (TFA), built on detectron2. It… | 23 | 1218 | maintenance |
| mit-han-lab/tinyml MIT Han Lab's TinyML research repository containing projects like TinyTL and NetAug for memory-efficient deep learning on microcontrollers … | 32 | 1216 | maintenance |
| awslabs/sockeye Sockeye is an open-source sequence-to-sequence framework for Neural Machine Translation built on PyTorch, with distributed training and opt… | 23 | 1216 | maintenance |
| The-AI-Summer/self-attention-cv A PyTorch library implementing various self-attention mechanisms and transformer building blocks for computer vision, including multi-head … | 23 | 1215 | maintenance |
| anderskm/gputil GPUtil is a Python module that queries NVIDIA GPU status (memory usage, load, availability) by wrapping the nvidia-smi command-line tool. I… | 64 | 1214 | maintenance |
| nihui/realsr-ncnn-vulkan A command-line tool implementing the RealSR real-world super-resolution model using the ncnn inference framework with Vulkan GPU accelerati… | 23 | 1214 | maintenance |
| UMass-Embodied-AGI/3D-LLM 3D-LLM is the research code for a large language model that takes 3D representations (objects and scenes) as input, built on BLIP-2/LAVIS. … | 28 | 1213 | maintenance |
| mouna99/dien A TensorFlow implementation of the Deep Interest Evolution Network (DIEN) for click-through rate prediction, based on the Alibaba research … | 32 | 1211 | maintenance |
| arundo/adtk ADTK is a Python package for unsupervised and rule-based anomaly detection in time series. It provides detectors, transformers, and aggrega… | 23 | 1211 | maintenance |
| Shikhargupta/Spiking-Neural-Network A pure Python implementation of a hardware-efficient spiking neural network (SNN) trained with Spike-Time Dependent Plasticity (STDP). It i… | 32 | 1210 | maintenance |
| facebookresearch/BLINK BLINK is a Python entity linking library from Facebook Research that resolves mentions in text to Wikipedia entities using a two-stage bi-e… | 10 | 1210 | maintenance |
| imfing/keras-flask-deploy-webapp A minimal, customizable Flask web app template for deploying Keras/TensorFlow image classification models. It ships with a mobile-friendly … | 32 | 1209 | maintenance |
| Alexander-H-Liu/End-to-end-ASR-Pytorch A PyTorch implementation of end-to-end automatic speech recognition (ASR), formerly known as Listen, Attend and Spell. It supports seq2seq … | 32 | 1208 | maintenance |
| NVlabs/VoxFormer Official PyTorch implementation of VoxFormer, a CVPR 2023 highlight paper presenting a sparse voxel transformer for camera-based 3D semanti… | 31 | 1208 | maintenance |
| YuliangXiu/ECON ECON is a research tool that reconstructs high-fidelity 3D clothed human avatars from a single color image by combining implicit and explic… | 32 | 1207 | maintenance |
| facebookresearch/ToMe ToMe (Token Merging) is a PyTorch library from Meta AI that speeds up existing Vision Transformers by merging similar tokens inside the net… | 10 | 1207 | maintenance |
| lessw2020/Ranger-Deep-Learning-Optimizer Ranger is a PyTorch deep learning optimizer that synergistically combines RAdam (Rectified Adam), LookAhead, and Gradient Centralization in… | 32 | 1206 | maintenance |
| bckenstler/CLR A Keras callback implementing cyclical learning rate (CLR) policies from Leslie Smith's paper, with built-in triangular, triangular2, and e… | 32 | 1206 | maintenance |
| reiinakano/arbitrary-image-stylization-tfjs A browser-based implementation of arbitrary neural style transfer built with TensorFlow.js, letting users stylize any photo in the style of… | 32 | 1205 | maintenance |
| aigamedev/scikit-neuralnetwork A Python library providing deep neural network classifiers and regressors with a scikit-learn-compatible API, wrapping Lasagne and Theano. … | 10 | 1204 | maintenance |
| twitter-research/tgn Official PyTorch implementation of Temporal Graph Networks (TGN), a framework for deep learning on dynamic graphs represented as sequences … | 32 | 1203 | maintenance |
| enazoe/yolo-tensorrt A C++ wrapper around NVIDIA TensorRT for running YOLO object detection models (YOLOv3, YOLOv4, YOLOv5) with support for FP32, FP16, and INT… | 57 | 1202 | maintenance |
| google-research/tapas TAPAS is Google Research's implementation of transformer-based models for question answering over tables, including pretrained checkpoints … | 10 | 1202 | maintenance |
| epfml/sent2vec Sent2vec is a C++ library with a Cython/Python interface that trains unsupervised distributed representations of sentences and short texts,… | 23 | 1201 | maintenance |
| henrywoo/chatllama An open-source Python library implementing RLHF-based fine-tuning of Meta's LLaMA models to build ChatGPT-style chat assistants, runnable o… | 21 | 1201 | maintenance |
| facebookresearch/House3D House3D is a virtual 3D environment of over 45k fully annotated indoor scenes from the SUNCG dataset, built for training embodied AI agents… | 10 | 1200 | maintenance |
| astroML/astroML AstroML is a Python library for machine learning, statistics, and data mining aimed at astronomy and astrophysics, built on numpy, scipy, s… | 32 | 1199 | maintenance |
| chainer/chainerrl ChainerRL is a deep reinforcement learning library implementing state-of-the-art algorithms like DQN, Rainbow, PPO, SAC, and A3C in Python … | 23 | 1199 | maintenance |
| ethan-li-coding/SemiGlobalMatching A complete, well-commented C++ implementation of the classic Semi-Global Matching (SGM) stereo matching algorithm for computing disparity a… | 32 | 1198 | maintenance |
| rotemtzaban/STIT STIT (Stitch it in Time) is a research implementation of a GAN-based framework for semantic editing of faces in real videos, based on the p… | 32 | 1197 | maintenance |
| DeepWisdom/AutoDL AutoDL is a fully automated deep learning framework from DeepWisdom that won the NeurIPS AutoDL challenge. It performs automatic feature en… | 23 | 1197 | maintenance |
| facebookresearch/mixup-cifar10 Official PyTorch implementation of mixup, a data augmentation technique that trains neural networks on convex combinations of image pairs a… | 10 | 1197 | maintenance |
| sryza/spark-timeseries A Scala/Java/Python library for time series analysis on Apache Spark, providing distributed abstractions for large time series datasets plu… | 32 | 1196 | maintenance |
| lucidrains/CoCa-pytorch A Pytorch implementation of CoCa (Contrastive Captioners), an image-text foundation model that combines contrastive learning with an encode… | 23 | 1196 | maintenance |
| KaihuaTang/Scene-Graph-Benchmark.pytorch A PyTorch codebase for Scene Graph Generation (SGG) built on maskrcnn-benchmark, implementing methods from the CVPR 2020 paper 'Unbiased Sc… | 59 | 1195 | maintenance |
| huggingface/hmtl HMTL is a Hierarchical Multi-Task Learning model for NLP that jointly trains Named Entity Recognition, Entity Mention Detection, Relation E… | 32 | 1195 | maintenance |
| rinongal/StyleGAN-nada Official PyTorch implementation of StyleGAN-NADA, a CLIP-guided method for adapting pre-trained StyleGAN image generators to new domains us… | 32 | 1195 | maintenance |
| Kagami/go-face A Go library that provides face recognition by wrapping dlib's machine learning models, including face detection, landmark prediction, and … | 32 | 1194 | maintenance |
| rockchip-linux/rknn-toolkit2 RKNN-Toolkit2 is Rockchip's software development kit for converting trained AI models to RKNN format and deploying them on Rockchip NPU chi… | 23 | 1194 | maintenance |
| ROCm/MIOpen MIOpen is AMD's open-source library providing high-performance GPU kernels for machine learning primitives (convolutions, normalization, ac… | 93 | 1193 | maintenance |
| VITA-Group/DeblurGANv2 Official PyTorch implementation of DeblurGAN-v2, an ICCV 2019 relativistic conditional GAN for single-image motion deblurring with a Featur… | 32 | 1192 | maintenance |
| altosaar/variational-autoencoder A reference implementation of a variational autoencoder (VAE) in PyTorch, TensorFlow, and JAX, including an inverse autoregressive flow var… | 23 | 1192 | maintenance |
| yitu-opensource/T2T-ViT Official PyTorch implementation of Tokens-to-Token ViT (T2T-ViT), an ICCV 2021 vision transformer architecture trained from scratch on Imag… | 23 | 1191 | maintenance |
| peng-zhihui/A-Eye A super-mini AI camera development board based on the Kendryte K210 chip, with fully open-source hardware (PCB and enclosure designs) and f… | 32 | 1190 | maintenance |
| yahoo/egads EGADS (Extensible Generic Anomaly Detection System) is an open-source Java library for automatically detecting anomalies, outliers, and cha… | 32 | 1190 | maintenance |
| khundman/telemanom Telemanom is a Python framework that uses vanilla LSTMs (Keras/TensorFlow) to detect anomalies in multivariate time series sensor data, wit… | 33 | 1189 | maintenance |
| elliottwu/unsup3d Official PyTorch implementation of the CVPR 2020 (Oral, Best Paper Award) research paper 'Unsupervised Learning of Probably Symmetric Defor… | 32 | 1189 | maintenance |
| vanhuyz/CycleGAN-TensorFlow A TensorFlow implementation of CycleGAN for unpaired image-to-image translation, based on the original CycleGAN paper. It includes scripts … | 23 | 1189 | maintenance |
| aitorzip/DeepGTAV DeepGTAV is a C++ plugin for Grand Theft Auto V that converts the game into a vision-based self-driving car research environment. It expose… | 32 | 1188 | maintenance |
| lightaime/deep_gcns_torch A PyTorch library implementing DeepGCNs, DeeperGCN, and GNN1000 - graph convolutional network architectures that borrow residual/dense conn… | 32 | 1187 | maintenance |
| whai362/PSENet Official PyTorch implementation of PSENet, a progressive scale expansion network for robust scene text detection. It includes training, tes… | 23 | 1186 | maintenance |
| openai/procgen A benchmark of 16 procedurally-generated, game-like Gym environments for measuring how quickly reinforcement learning agents learn generali… | 56 | 1185 | maintenance |
| zju3dv/snake Official research code for 'Deep Snake for Real-Time Instance Segmentation' (CVPR 2020 oral), implementing a deep contour-based instance se… | 32 | 1185 | maintenance |
| kexinhuang12345/DeepPurpose DeepPurpose is a PyTorch-based deep learning library for molecular modeling, supporting drug-target interaction (DTI), drug property, drug-… | 23 | 1185 | maintenance |
| business-science/ai-data-science-team A Python library of specialized LLM-powered agents for common data science workflows such as data loading, cleaning, wrangling, visualizati… | 59 | 5394 | experimental |
| HyperGAN/HyperGAN HyperGAN is a composable GAN (generative adversarial network) framework built on PyTorch, offering both a Python API and a CLI with a user … | 23 | 1183 | maintenance |
| lucidrains/performer-pytorch A PyTorch implementation of the Performer transformer, which uses linear attention via the FAVOR+ (Fast Attention Via positive Orthogonal R… | 23 | 1182 | maintenance |
| naoto0804/pytorch-AdaIN An unofficial PyTorch implementation of the AdaIN paper 'Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization' (Huang… | 10 | 1181 | maintenance |
| jankrepl/deepdow deepdow is a Python library built on PyTorch that connects deep learning with portfolio optimization, allowing networks that output asset w… | 32 | 1180 | maintenance |
| Smerity/sha-rnn A PyTorch implementation of the Single Headed Attention RNN (SHA-LSTM) from Stephen Merity's paper, combining an LSTM with a single attenti… | 32 | 1180 | maintenance |
| didierbrun/DBPathRecognizer A lightweight Swift library for iOS that recognizes touch gestures by matching drawn paths against 8-direction direction-sequence models. I… | 32 | 1178 | maintenance |
| clovaai/voxceleb_trainer A PyTorch framework for training and evaluating speaker recognition and verification models on the VoxCeleb datasets. It implements multipl… | 67 | 1176 | maintenance |
| VoltaML/voltaML VoltaML is a lightweight Python library that compiles and optimizes machine learning and deep learning models for high-performance inferenc… | 32 | 1176 | maintenance |
| fbcotter/pytorch_wavelets A PyTorch library implementing 2D discrete wavelet transforms (DWT), dual-tree complex wavelet transforms (DTCWT), their inverses, and a DT… | 23 | 1176 | maintenance |
| punica-ai/punica Punica is a Python system for serving many LoRA-finetuned LLMs from a single copy of the base model on one GPU, using a custom CUDA kernel … | 10 | 1175 | maintenance |
| julrog/nn_vis A Python tool that processes trained neural networks and renders them as decluttered 3D visualizations of their architecture and parameters… | 23 | 1174 | maintenance |
| ermongroup/SDEdit SDEdit is a PyTorch implementation of the ICLR 2022 paper 'Guided Image Synthesis and Editing with Stochastic Differential Equations'. It p… | 32 | 1168 | maintenance |
| daniilidis-group/neural_renderer A PyTorch port of the Neural 3D Mesh Renderer (CVPR 2018), providing differentiable rasterization of 3D meshes. It supports rendering from … | 32 | 1167 | maintenance |
| WebSpiderUtils/verification_code A research repository documenting approaches and code for solving mainstream CAPTCHA systems such as Geetest, NetEase Yidun, and Aliyun CAP… | 32 | 1165 | maintenance |
| alexandre01/deepsvg Official PyTorch code for the NeurIPS 2020 paper 'DeepSVG: A Hierarchical Generative Network for Vector Graphics Animation'. It provides a … | 32 | 1164 | maintenance |
| xiangwang1223/knowledge_graph_attention_network TensorFlow implementation of KGAT (Knowledge Graph Attention Network), a KDD 2019 paper on knowledge-aware personalized recommendation. It … | 32 | 1164 | maintenance |
| hiroharu-kato/neural_renderer A differentiable 3D mesh renderer implemented in Chainer, accompanying the CVPR 2018 paper 'Neural 3D Mesh Renderer'. It enables gradient-b… | 23 | 1164 | maintenance |
| google/active-learning A Python module for running experiments comparing different active learning algorithms on benchmark datasets. It provides a main experiment… | 10 | 1163 | maintenance |
| ChenyangQiQi/FateZero FateZero is a zero-shot text-based video editing framework built on pretrained Stable Diffusion models, introduced in an ICCV 2023 Oral pap… | 21 | 1162 | maintenance |
| facebookresearch/meshrcnn Mesh R-CNN is Facebook AI Research's official implementation of the ICCV 2019 paper, a model that detects objects in images and predicts th… | 60 | 1161 | maintenance |
| shadow2496/VITON-HD Official PyTorch implementation of VITON-HD (CVPR 2021), a high-resolution image-based virtual try-on model that transfers a clothing item … | 40 | 1161 | maintenance |
| spring-media/TransformerTTS A TensorFlow 2 implementation of a non-autoregressive Transformer-based neural network for text-to-speech synthesis, based on FastSpeech an… | 10 | 1161 | maintenance |
| cheungdaven/DeepRec DeepRec is an open-source Python/TensorFlow toolkit implementing deep learning based recommendation models for rating prediction, top-N ite… | 32 | 1160 | maintenance |
| wyharveychen/CloserLookFewShot Reference PyTorch implementation of the ICLR 2019 paper 'A Closer Look at Few-shot Classification', providing a testbed for empirical evalu… | 32 | 1159 | maintenance |
| HKUST-Aerial-Robotics/Teach-Repeat-Replan Teach-Repeat-Replan is a complete C++/ROS system enabling aggressive autonomous drone flight in complex environments. It converts a manuall… | 32 | 1159 | maintenance |
| FALCONN-LIB/FALCONN FALCONN is a C++ library (with a Python wrapper) implementing locality-sensitive hashing (LSH) data structures for fast nearest neighbor se… | 23 | 1159 | maintenance |
| Kyubyong/dc_tts A TensorFlow implementation of DC-TTS, a text-to-speech model based on deep convolutional networks with guided attention. It includes train… | 32 | 1158 | maintenance |
| soulteary/docker-prompt-generator A Docker-based web application that uses language models to generate and expand prompts for image generation tools like MidJourney and Stab… | 30 | 1157 | maintenance |
| ajay-sainy/Wav2Lip-GFPGAN A pipeline combining Wav2Lip lip-sync generation with GFPGAN face restoration to produce high-quality talking-head videos from an input vid… | 32 | 1156 | maintenance |
| Paperspace/DataAugmentationForObjectDetection A Python library implementing data augmentation transforms (flipping, scaling, rotation, shearing, translation, resizing) that adjust both … | 32 | 1156 | maintenance |
| abhishekkrthakur/tez Tez is a lightweight, simple trainer library for PyTorch that simplifies training loops while keeping users close to native PyTorch. It sup… | 23 | 1155 | maintenance |
| DeepGraphLearning/RecommenderSystems A Python research library collecting the authors' recommender systems code, covering sequential, feature-based (CTR), and social recommenda… | 32 | 1154 | maintenance |