function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| chrischoy/3D-R2N2 3D-R2N2 is a PyTorch-based implementation of a recurrent neural network that reconstructs voxelized 3D models of objects from one or multip… | 32 | 1409 | maintenance |
| Ceruleanacg/Personae Personae is a Python library implementing deep reinforcement learning (DDPG, Double DQN, Dueling DQN, Policy Gradient) and supervised learn… | 32 | 1409 | maintenance |
| tonylins/pytorch-mobilenet-v2 A PyTorch implementation of the MobileNetV2 architecture with automatic download of ImageNet pretrained weights. It reproduces the official… | 32 | 1406 | maintenance |
| advboxes/AdvBox AdvBox is a Baidu open-source toolbox for generating adversarial examples that fool neural networks across frameworks like PaddlePaddle, Py… | 32 | 1404 | maintenance |
| dbolya/tomesd ToMe for SD is a Python/PyTorch library that speeds up Stable Diffusion image generation by merging redundant tokens in the underlying tran… | 21 | 1404 | maintenance |
| facebookresearch/end-to-end-negotiator A PyTorch research codebase from Facebook AI Research implementing end-to-end learning for negotiation dialogues, including models for hier… | 10 | 1400 | maintenance |
| gotzmann/llama.go A pure-Go reimplementation of llama.cpp-style LLM inference, running LLaMA-family models on CPU without C++ dependencies. It includes tenso… | 21 | 1399 | maintenance |
| TorchCraft/TorchCraft TorchCraft is a bridge library connecting the Torch machine learning framework to StarCraft: Brood War via a BWAPI module that streams game… | 10 | 1398 | maintenance |
| opendilab/DI-star DI-star is a large-scale distributed training platform for building StarCraft II game AI, including supervised and reinforcement learning t… | 37 | 1393 | maintenance |
| swz30/MPRNet MPRNet is a PyTorch implementation of a multi-stage progressive image restoration network published at CVPR 2021. It provides pretrained mo… | 32 | 1393 | maintenance |
| JasperSnoek/spearmint Spearmint is a Python package for Bayesian optimization of machine learning algorithms, iteratively tuning parameters to minimize an object… | 32 | 1392 | maintenance |
| innnky/emotional-vits Emotional VITS is a voice synthesis (TTS) model based on VITS that enables emotion-controllable speech generation without requiring manual … | 32 | 1391 | maintenance |
| Maluuba/nlg-eval A Python library and CLI tool that computes unsupervised automated metrics (BLEU, METEOR, ROUGE, CIDEr, SPICE, SkipThought, and embedding-b… | 23 | 1391 | maintenance |
| 198808xc/Pangu-Weather Official implementation of Pangu-Weather, a deep learning model for fast, accurate medium-range global weather forecasting using 3D neural … | 22 | 1390 | maintenance |
| facebookresearch/LAMA LAMA is a research probe for analyzing the factual and commonsense knowledge stored in pretrained language models such as BERT, RoBERTa, GP… | 10 | 1390 | maintenance |
| yao8839836/text_gcn The official implementation of Text GCN, a Graph Convolutional Network approach for text classification published at AAAI 2019. It builds w… | 32 | 1389 | maintenance |
| yinboc/liif LIIF is the official PyTorch implementation of the CVPR 2021 paper 'Learning Continuous Image Representation with Local Implicit Image Func… | 32 | 1389 | maintenance |
| doonny/PipeCNN PipeCNN is an OpenCL-based FPGA accelerator for large-scale convolutional neural network inference, written in C with pipelined kernels. It… | 32 | 1387 | maintenance |
| chrisdonahue/wavegan Official TensorFlow implementation of WaveGAN, a generative adversarial network that learns to synthesize raw audio waveforms from examples… | 32 | 1383 | maintenance |
| DeepGraphLearning/KnowledgeGraphEmbedding A PyTorch toolkit implementing the RotatE knowledge graph embedding model along with TransE, ComplEx, DistMult, and pRotatE. It trains and … | 32 | 1383 | maintenance |
| bytedance/music_source_separation A PyTorch implementation of music source separation that splits songs into sources like vocals and accompaniment. It ships pretrained check… | 10 | 1383 | maintenance |
| myhub/tr An offline Chinese text detection and recognition OCR SDK with C++ core code and Python bindings, supporting models like CRNN, CTPN, and Pi… | 50 | 1382 | maintenance |
| zhanlaoban/EDA_NLP_for_Chinese A Python implementation of the EDA (Easy Data Augmentation) paper adapted for Chinese text corpora. It augments labeled text classification… | 32 | 1382 | maintenance |
| wouterkool/attention-learn-to-route A PyTorch implementation of the attention-based neural model from the ICLR 2019 paper 'Attention, Learn to Solve Routing Problems!', traine… | 32 | 1382 | maintenance |
| jrieke/traingenerator Traingenerator is a Streamlit-based web app that generates custom template code for machine learning projects with PyTorch and scikit-learn… | 32 | 1381 | maintenance |
| amazon-science/patchcore-inspection Official implementation of PatchCore, a deep-learning method for industrial image anomaly detection and localization from Roth et al. (2021… | 32 | 1379 | maintenance |
| PSPNet Reference implementation of the Pyramid Scene Parsing Network (PSPNet), a CVPR 2017 semantic segmentation model that won the ImageNet Scene… | 32 | 1378 | maintenance |
| tensorflow/ecosystem A collection of templates and connectors for integrating TensorFlow with other open-source frameworks such as Kubernetes, Kubeflow, Spark, … | 10 | 1377 | maintenance |
| takuya-takeuchi/FaceRecognitionDotNet A C# port of the popular face_recognition library providing a simple facial recognition API for .NET. It supports face detection, recogniti… | 23 | 1376 | maintenance |
| tinkoff-ai/CORL CORL is a research-oriented Python library providing thoroughly benchmarked single-file implementations of state-of-the-art offline and off… | 10 | 1373 | maintenance |
| tensorboy/pytorch_Realtime_Multi-Person_Pose_Estimation A PyTorch implementation of the CVPR'17 Realtime Multi-Person 2D Pose Estimation (OpenPose/rtpose) model. It provides pretrained weights, d… | 32 | 1371 | maintenance |
| Cysu/open-reid Open-ReID is a lightweight Python library for person re-identification built on PyTorch, providing a uniform dataset interface, models, and… | 32 | 1370 | maintenance |
| zalandoresearch/pytorch-ts PyTorchTS is a PyTorch-based probabilistic time series forecasting library that uses GluonTS as its backend API for loading, transforming, … | 23 | 1370 | maintenance |
| nv-tlabs/lift-splat-shoot PyTorch implementation of Lift-Splat-Shoot (ECCV 2020), an end-to-end model that converts images from arbitrary multi-camera rigs into a bi… | 32 | 1369 | maintenance |
| open-mmlab/mmfashion MMFashion is an open-source PyTorch-based toolbox for visual fashion analysis from the OpenMMLab project. It provides modular implementatio… | 32 | 1369 | maintenance |
| akamaster/pytorch_resnet_cifar10 A PyTorch implementation of ResNet architectures (ResNet20 through ResNet1202) for CIFAR10/CIFAR100 that faithfully matches the original pa… | 32 | 1368 | maintenance |
| tensor-compiler/taco TACO (The Tensor Algebra Compiler) is a C++ library that computes sparse and dense tensor algebra and linear algebra expressions, generatin… | 30 | 1367 | maintenance |
| mdeff/cnn_graph Reference implementation of the NIPS 2016 paper 'Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering' (ChebNet),… | 32 | 1366 | maintenance |
| JonasGeiping/cramming A research framework for pretraining BERT-type language models from scratch on a single consumer GPU in one day, replicating the paper 'Cra… | 22 | 1366 | maintenance |
| hiyouga/FastEdit FastEdit is a Python library and CLI tool for editing large language models, injecting fresh or customized factual knowledge into pretraine… | 19 | 1366 | maintenance |
| BorealisAI/advertorch AdverTorch is a Python toolbox for adversarial robustness research built on PyTorch. It provides modules for generating adversarial perturb… | 32 | 1364 | maintenance |
| wzzheng/TPVFormer TPVFormer is a CVPR 2023 research implementation of a tri-perspective view transformer for vision-based 3D semantic occupancy prediction, s… | 32 | 1364 | maintenance |
| ali-vilab/ACE_plus ACE++ is a Python library and model release from Alibaba's Tongyi Lab for instruction-based image creation and editing via context-aware co… | 28 | 1364 | maintenance |
| taoxugit/AttnGAN A PyTorch implementation of AttnGAN, a fine-grained text-to-image generative adversarial network with multi-stage attention mechanisms from… | 32 | 1363 | maintenance |
| keirp/automatic_prompt_engineer Automatic Prompt Engineer (APE) is a Python library implementing the research method from 'Large Language Models Are Human-Level Prompt Eng… | 32 | 1363 | maintenance |
| sail-sg/poolformer PoolFormer is a PyTorch implementation of the CVPR 2022 paper 'MetaFormer Is Actually What You Need for Vision', which replaces attention i… | 23 | 1363 | maintenance |
| nvidia-cosmos/cosmos-predict2.5 NVIDIA Cosmos-Predict2.5 is a family of world foundation models (WFMs) that generate video predictions of future world states for physical … | 72 | 1362 | maintenance |
| MoyGcc/vid2avatar Vid2Avatar is the official PyTorch implementation of a CVPR 2023 method that reconstructs detailed 3D human avatars from monocular in-the-w… | 56 | 1360 | maintenance |
| fofr/cog-face-to-many A Cog-packaged ComfyUI model that transforms a face photo into 3D, pixel art, video game character, claymation, emoji, or toy styles using … | 25 | 1360 | maintenance |
| LetheSec/HuggingFace-Download-Accelerator A Python wrapper script around huggingface-cli and hf_transfer that enables high-speed downloads of HuggingFace models and datasets, with b… | 27 | 1359 | maintenance |
| mayuelala/FollowYourPose Official PyTorch implementation of Follow-Your-Pose (AAAI 2024), a pose-guided text-to-video generation model that tunes a text-to-image mo… | 21 | 1358 | maintenance |
| orobix/retina-unet A Python implementation of a U-Net convolutional neural network for segmenting blood vessels in retina fundus images. It performs binary pi… | 32 | 1355 | maintenance |
| dlunion/DBFace DBFace is a real-time, single-stage face detection model implemented in Python, offering small model sizes with high accuracy on the WiderF… | 32 | 1354 | maintenance |
| davidcelis/recommendable Recommendable is a Ruby gem that adds a like/dislike-based recommendation engine to Ruby applications using Jaccardian similarity and memor… | 23 | 1354 | maintenance |
| bumble-tech/private-detector Bumble's Private Detector is a pretrained TensorFlow image classifier based on EfficientNet-v2 that detects lewd images. The repo provides … | 23 | 1352 | maintenance |
| ShuLiu1993/PANet A PyTorch re-implementation of PANet (Path Aggregation Network), the CVPR 2018 paper that won 1st place in the COCO 2017 Instance Segmentat… | 32 | 1347 | maintenance |
| Replicable-MARL/MARLlib MARLlib is a unified multi-agent reinforcement learning library built on Ray and RLlib with PyTorch, providing implementations of many MARL… | 23 | 1345 | maintenance |
| microsoft/X-Decoder Official PyTorch implementation of X-Decoder, a generalized decoding model from CVPR 2023 that unifies pixel-level segmentation, image-leve… | 22 | 1345 | maintenance |
| Star-Clouds/CenterFace CenterFace is a lightweight (7.3MB) anchor-free face detection and alignment model that detects faces and predicts facial landmarks using a… | 32 | 1344 | maintenance |
| dorarad/gansformer GANformer is a research implementation of a generative adversarial transformer that uses a bipartite attention structure for efficient high… | 23 | 1344 | maintenance |
| PeizeSun/SparseR-CNN Sparse R-CNN is a PyTorch implementation (built on Detectron2) of the CVPR 2021 / PAMI 2023 paper 'End-to-End Object Detection with Learnab… | 23 | 1343 | maintenance |
| mailgun/talon Talon is a Python library from Mailgun that extracts reply quotations and signatures from email messages, both plain text and HTML. It comb… | 23 | 1342 | maintenance |
| HobbitLong/CMC Official PyTorch implementation of Contrastive Multiview Coding (CMC), a self-supervised visual representation learning method that contras… | 32 | 1340 | maintenance |
| kuaikuaikim/dface DFace is an open-source Python library implementing face detection and recognition with PyTorch, based on the MTCNN cascaded convolutional … | 23 | 1340 | maintenance |
| timojl/clipseg CLIPSeg is a Python implementation of the CVPR 2022 paper 'Image Segmentation Using Text and Image Prompts', enabling zero-shot segmentatio… | 32 | 1338 | maintenance |
| mdipierro/nlib A pure-Python numerical computing library (nlib.py) accompanied by the open book 'Annotated Algorithms in Python', covering linear algebra,… | 66 | 1337 | maintenance |
| bloc97/CrossAttentionControl An unofficial Jupyter Notebook implementation of the 'Prompt-to-Prompt Image Editing with Cross Attention Control' paper adapted for Stable… | 32 | 1336 | maintenance |
| ikostrikov/pytorch-a3c A PyTorch implementation of the Asynchronous Advantage Actor Critic (A3C) reinforcement learning algorithm from the DeepMind paper 'Asynchr… | 32 | 1336 | maintenance |
| torrvision/crfasrnn Reference implementation of CRF-RNN, an ICCV 2015 semantic image segmentation method that integrates conditional random fields into a neura… | 23 | 1335 | maintenance |
| SHI-Labs/Versatile-Diffusion Versatile Diffusion is the official PyTorch implementation of a unified multi-flow multimodal diffusion model that handles text-to-image, i… | 32 | 1334 | maintenance |
| ahmetozlu/tensorflow_object_counting_api An open-source framework built on TensorFlow and Keras that simplifies developing object counting systems. It supports cumulative counting,… | 23 | 1334 | maintenance |
| GuitarML/SmartGuitarAmp SmartGuitarAmp is a JUCE-based audio plugin (VST3/AU/AAX) that uses neural network models (WaveNet-style) to emulate a real tube guitar amp… | 23 | 1331 | maintenance |
| maudzung/Complex-YOLOv4-Pytorch A PyTorch implementation of Complex-YOLO, a YOLOv4-based model for real-time 3D object detection on LiDAR point clouds. It supports distrib… | 32 | 1327 | maintenance |
| FreedomIntelligence/HuatuoGPT HuatuoGPT is an open Chinese medical large language model (7B and 13B variants) fine-tuned on a hybrid of distilled and real-world medical … | 30 | 1327 | maintenance |
| kaz-Anova/StackNet StackNet is a Java-based meta-modelling framework that implements Wolpert's stacked generalization across multiple layers, resembling a fee… | 32 | 1326 | maintenance |
| google/sg2im A PyTorch research implementation of the CVPR 2018 paper 'Image Generation from Scene Graphs' by Johnson et al. It converts a structured sc… | 10 | 1325 | maintenance |
| CharlesShang/DCNv2 A PyTorch implementation of Deformable Convolutional Networks v2 (DCNv2), providing CUDA/CPU operators for deformable convolution as a drop… | 71 | 1323 | maintenance |
| kabachuha/sd-webui-text2video An Automatic1111 WebUI extension that adds text-to-video generation using diffusion models like ModelScope and VideoCrafter, relying only o… | 30 | 1322 | maintenance |
| facebookresearch/moco-v3 A PyTorch implementation of MoCo v3, a self-supervised contrastive learning method for ResNet and Vision Transformer (ViT) models. It inclu… | 10 | 1322 | maintenance |
| WisconsinAIVision/yolact_edge YolactEdge is a PyTorch implementation of a real-time instance segmentation model optimized for edge devices like the NVIDIA Jetson AGX Xav… | 32 | 1321 | maintenance |
| CSTR-Edinburgh/merlin Merlin is a toolkit from the University of Edinburgh's CSTR for building deep neural network models for statistical parametric speech synth… | 23 | 1321 | maintenance |
| YelpArchive/MOE MOE (Metric Optimization Engine) is a global, black box optimization engine built by Yelp for optimizing expensive, time-consuming objectiv… | 10 | 1321 | maintenance |
| aitorzip/PyTorch-CycleGAN A clean, readable PyTorch implementation of CycleGAN for unpaired image-to-image translation using cycle-consistent adversarial networks. I… | 32 | 1319 | maintenance |
| hukkelas/DeepPrivacy DeepPrivacy is a PyTorch-based GAN that automatically anonymizes faces in images and videos by generating realistic synthetic replacements.… | 32 | 1316 | maintenance |
| PatMartin/Dex Dex is a desktop data exploration and visualization application written in Java/Groovy on top of JavaFX. It combines ETL capabilities, mach… | 32 | 1315 | maintenance |
| msracver/Deep-Feature-Flow Official MXNet implementation of Deep Feature Flow (CVPR 2017), an end-to-end framework for video recognition such as object detection and … | 32 | 1315 | maintenance |
| szagoruyko/wide-residual-networks Reference implementation of Wide Residual Networks (WRN), a ResNet variant that trades depth for width to train faster and achieve state-of… | 32 | 1315 | maintenance |
| lda-project/lda A Python library implementing latent Dirichlet allocation (LDA) topic modeling via collapsed Gibbs sampling, with a scikit-learn-style API.… | 32 | 1314 | maintenance |
| facebookresearch/svoice SVoice is a PyTorch implementation of the ICML paper 'Voice Separation with an Unknown Number of Multiple Speakers' from Facebook AI Resear… | 10 | 1314 | maintenance |
| Renovamen/Speech-Emotion-Recognition A Python library implementing speech emotion recognition with Keras/TensorFlow 2 using LSTM, CNN, SVM, and MLP models. It extracts audio fe… | 32 | 1312 | maintenance |
| LinXueyuanStdio/LaTeX_OCR_PRO A deep-learning application that converts images of math formulas (printed, handwritten, and Chinese-mixed) into LaTeX code using a Seq2Seq… | 32 | 1312 | maintenance |
| EricGuo5513/momask-codes Official PyTorch implementation of MoMask (CVPR 2024), a masked modeling framework that generates 3D human motion sequences from text promp… | 27 | 1312 | maintenance |
| ToanTech/py-apple-quadruped-robot Py-Apple Dog (Pineapple Dog) is a low-cost, open-source hardware and software project for building a DIY quadruped robot dog. It bundles fo… | 32 | 1311 | maintenance |
| d-li14/involution Official PyTorch implementation of the involution neural operator from the CVPR 2021 paper 'Involution: Inverting the Inherence of Convolut… | 32 | 1310 | maintenance |
| tysam-code/hlb-CIFAR10 A single-file PyTorch implementation that trains a neural network to 94% accuracy on CIFAR-10 in under 6.3 seconds on a single A100 GPU, fo… | 22 | 1310 | maintenance |
| NVlabs/prismer Official PyTorch implementation of Prismer, a data- and parameter-efficient vision-language model that ensembles pre-trained task-specific … | 30 | 1309 | maintenance |
| 920232796/bert_seq2seq A lightweight PyTorch framework for fine-tuning pretrained language models (BERT, RoBERTa, Nezha, GPT2, T5, BART) on Chinese NLP tasks usin… | 32 | 1308 | maintenance |
| mit-han-lab/data-efficient-gans Official implementation of Differentiable Augmentation (DiffAugment), a NeurIPS 2020 method that improves GAN training data efficiency by a… | 32 | 1308 | maintenance |
| chickenbestlover/RNN-Time-series-Anomaly-Detection A PyTorch implementation of an RNN-based time-series anomaly detector using a two-stage strategy of multi-step prediction followed by anoma… | 32 | 1305 | maintenance |
| senlinuc/caffe_ocr An experimental research project built on Caffe implementing CNN+BLSTM+CTC text recognition architectures, with modifications for LSTM, war… | 32 | 1305 | maintenance |