domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| 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 |
| FudanNLP/fitlog fitlog (fast + git + log) is a Python tool that helps deep learning practitioners record training logs and manage experiment code, combinin… | 23 | 1512 | 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 |
| maderix/ANE A research project demonstrating backpropagation and neural network training directly on Apple's Neural Engine using reverse-engineered pri… | 47 | 7253 | experimental |
| 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 | 1508 | 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 |
| nlpyang/BertSum BertSum is the official PyTorch implementation of the paper 'Fine-tune BERT for Extractive Summarization', providing preprocessing pipeline… | 32 | 1506 | maintenance |
| tinyvision/SOLIDER SOLIDER is a semantic-controllable self-supervised learning framework that learns general human representations from massive unlabeled huma… | 32 | 1504 | maintenance |
| stevenmiller888/mind Mind is a flexible neural network library for JavaScript that runs in both Node.js and the browser. It supports vectorized training, config… | 73 | 1503 | 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 | 1503 | maintenance |
| jina-ai/finetuner Jina AI's Finetuner is a Python library for task-oriented fine-tuning of pretrained models like BERT and CLIP to produce better embeddings … | 10 | 1503 | maintenance |
| vacancy/Synchronized-BatchNorm-PyTorch A PyTorch library implementing Synchronized Batch Normalization, which computes batch statistics across all GPUs during multi-device traini… | 32 | 1502 | maintenance |
| mpezeshki/pytorch_forward_forward A PyTorch implementation of Geoffrey Hinton's forward-forward (FF) training algorithm, an alternative to back-propagation that computes gra… | 32 | 1500 | maintenance |
| Swall0w/torchstat torchstat is a lightweight neural network analyzer for PyTorch that reports parameter counts, FLOPs, multiply-adds, and memory usage per la… | 23 | 1500 | maintenance |
| facebookresearch/UnsupervisedMT The original implementation of phrase-based (PBSMT) and neural (NMT) unsupervised machine translation models from the EMNLP 2018 paper by F… | 10 | 1499 | maintenance |
| NVIDIAGameWorks/kaolin-wisp NVIDIA Kaolin Wisp is a PyTorch library and engine for neural fields research, built on NVIDIA Kaolin Core. It provides differentiable rend… | 23 | 1498 | maintenance |
| igrigorik/decisiontree A Ruby library implementing the ID3 decision tree learning algorithm for both continuous and discrete datasets, with rule pruning (C4.5-sty… | 32 | 1490 | maintenance |
| srendle/libfm libFM is a C++ library implementing factorization machines, a model that combines feature engineering with factorization models to estimate… | 32 | 1487 | maintenance |
| speedinghzl/CCNet Official PyTorch implementation of CCNet, a Criss-Cross Attention network for semantic segmentation published at ICCV 2019 and TPAMI 2020. … | 32 | 1484 | maintenance |
| chengdazhi/Deformable-Convolution-V2-PyTorch A PyTorch implementation of Deformable Convolution V2 (DCNv2) custom CUDA operators, ported from the original MXNet implementation. It prov… | 32 | 1484 | maintenance |
| patrick-kidger/torchtyping A Python library providing type annotations for PyTorch tensors' shape, dtype, layout, and dimension names, with optional runtime checking … | 41 | 1482 | maintenance |
| pathak22/noreward-rl TensorFlow implementation of the ICML 2017 paper 'Curiosity-driven Exploration by Self-supervised Prediction', training RL agents with an i… | 32 | 1482 | maintenance |
| Hironsan/anago anaGo is a Python library for sequence labeling tasks such as named entity recognition, part-of-speech tagging, and semantic role labeling,… | 23 | 1482 | maintenance |
| chainer/chainercv ChainerCV is a Python library providing tools to train and run neural networks for computer vision tasks on top of the Chainer framework. I… | 10 | 1481 | maintenance |
| karpathy/reinforcejs REINFORCEjs is a JavaScript library implementing common reinforcement learning algorithms including Dynamic Programming, Temporal Differenc… | 32 | 1480 | maintenance |
| ruotianluo/ImageCaptioning.pytorch A PyTorch research codebase for image captioning, supporting self-critical sequence training, bottom-up features, transformer captioning mo… | 32 | 1476 | maintenance |
| google-deepmind/jraph Jraph is a lightweight Graph Neural Network library built on JAX, providing a GraphsTuple data structure, graph utilities, and a zoo of for… | 10 | 1474 | maintenance |
| faustomorales/keras-ocr A Python library packaging the CRAFT text detector and a Keras CRNN text recognition model into a high-level OCR pipeline. It supports pret… | 42 | 1473 | maintenance |
| YuanGongND/ast Official PyTorch implementation of the Audio Spectrogram Transformer (AST) from the Interspeech 2021 paper, which applies a Vision Transfor… | 32 | 1472 | maintenance |
| peteanderson80/bottom-up-attention A bottom-up attention model based on Faster R-CNN with ResNet-101 trained on Visual Genome, producing features for salient image regions. T… | 32 | 1469 | maintenance |
| CSAILVision/LabelMeAnnotationTool The source code for LabelMe, a web-based image annotation tool written in JavaScript that runs on your own Apache server. It lets users dra… | 32 | 1468 | maintenance |
| marian-nmt/marian Marian is an efficient Neural Machine Translation framework written in pure C++ with minimal dependencies, supporting fast multi-GPU traini… | 23 | 1468 | maintenance |
| mmatl/pyrender Pyrender is a pure Python, glTF 2.0-compliant OpenGL renderer for physically-based rendering and visualization of 3D scenes. It includes an… | 34 | 1466 | maintenance |
| modelfoxdotdev/modelfox ModelFox is a CLI tool and platform for training, deploying, and monitoring machine learning models from CSV files. It includes client libr… | 23 | 1466 | maintenance |
| sxyu/pixel-nerf Official PyTorch implementation of pixelNeRF, a CVPR 2021 method that predicts neural radiance fields conditioned on one or few input image… | 32 | 1465 | maintenance |
| Sanster/text_renderer A Python tool that generates synthetic text images with configurable visual effects for training deep learning OCR models like CRNN. It sup… | 32 | 1464 | maintenance |
| HRNet/HigherHRNet-Human-Pose-Estimation Official PyTorch implementation of HigherHRNet, a CVPR 2020 bottom-up multi-person human pose estimation model using scale-aware high-resol… | 32 | 1463 | maintenance |
| szagoruyko/attention-transfer PyTorch reference implementation of the ICLR 2017 paper 'Paying More Attention to Attention', which improves convolutional neural networks … | 32 | 1463 | maintenance |
| buppt/ChineseNER A simple Chinese named entity recognition (NER) implementation using BiLSTM+CRF in both TensorFlow and PyTorch. It includes training script… | 32 | 1463 | maintenance |
| facebookresearch/MaskFormer MaskFormer is a PyTorch/Detectron2-based implementation of the NeurIPS 2021 paper 'Per-Pixel Classification is Not All You Need for Semanti… | 10 | 1460 | maintenance |
| una-dinosauria/3d-pose-baseline A TensorFlow implementation of a simple yet effective baseline for 3D human pose estimation from 2D keypoints, published at ICCV 2017. It i… | 32 | 1459 | maintenance |
| DLTK/DLTK DLTK is a Python neural networks toolkit built on TensorFlow for deep learning on medical images. It provides reference model implementatio… | 32 | 1458 | maintenance |
| erwincoumans/motion_imitation A Python research codebase accompanying the paper 'Learning Agile Robotic Locomotion Skills by Imitating Animals'. It provides a Gym enviro… | 32 | 1457 | maintenance |
| nerfstudio-project/nerfacc NerfAcc is a PyTorch acceleration toolbox for Neural Radiance Fields (NeRF), focused on efficient sampling in the volumetric rendering pipe… | 23 | 1456 | maintenance |
| wvangansbeke/Unsupervised-Classification PyTorch implementation of SCAN (ECCV 2020), a two-step method for unsupervised image classification that combines self-supervised represent… | 32 | 1455 | maintenance |
| abreheret/PixelAnnotationTool A C++ desktop application for quickly annotating images with pixel-wise labels using OpenCV's watershed algorithm. Users draw markers with … | 23 | 1455 | maintenance |
| xuannianz/EfficientDet A Keras/TensorFlow implementation of the EfficientDet object detection model with pretrained COCO and ImageNet weights. It supports trainin… | 32 | 1454 | maintenance |
| dmmiller612/bert-extractive-summarizer A Python library for extractive text summarization using BERT embeddings via HuggingFace transformers. It embeds sentences, clusters them, … | 23 | 1453 | maintenance |
| microsoft/NeuronBlocks NeuronBlocks is an NLP deep learning modeling toolkit from Microsoft that lets users build end-to-end neural network training and inference… | 23 | 1452 | maintenance |
| kakao/khaiii Khaiii (Kakao Hangul Analyzer III) is a data-driven Korean morphological analyzer developed by Kakao, using a CNN-based neural network mode… | 65 | 1450 | maintenance |
| microsoft/SpeechT5 Microsoft's unified-modal speech-text pre-training framework implementing SpeechT5 and related models (SpeechLM, SpeechUT, VALL-E X, WavLLM… | 32 | 1449 | maintenance |
| HuwCampbell/grenade Grenade is a composable, dependently typed deep learning library for Haskell that lets users specify neural network architectures at the ty… | 23 | 1449 | maintenance |
| bojone/attention A small Python library providing implementations of the attention mechanism from 'Attention is All You Need' for TensorFlow/Keras. It is no… | 32 | 1448 | maintenance |
| BigScience BigScience is a research project training large transformer language models (BERT, GPT-style) at scale, built on a fork of Megatron-LM inte… | 32 | 1448 | maintenance |
| pytorch/gloo Gloo is a C++ collective communications library providing primitives like barrier, broadcast, and allreduce for multi-machine machine learn… | 76 | 1447 | maintenance |
| mit-han-lab/proxylessnas ProxylessNAS is a neural architecture search (NAS) framework that directly searches CNN architectures on the target task and target hardwar… | 32 | 1447 | maintenance |
| megvii-research/ML-GCN A PyTorch implementation of ML-GCN, the CVPR 2019 paper 'Multi-Label Image Recognition with Graph Convolutional Networks'. It provides trai… | 32 | 1446 | maintenance |
| BachiLi/redner redner is a differentiable Monte Carlo ray tracer that computes exact gradients of rendered images with respect to arbitrary scene paramete… | 32 | 1444 | maintenance |
| toandaominh1997/EfficientDet.Pytorch A PyTorch implementation of EfficientDet, a scalable and efficient object detection model from a 2019 Google Research paper. It supports Ef… | 10 | 1441 | maintenance |
| liyaguang/DCRNN A TensorFlow implementation of the Diffusion Convolutional Recurrent Neural Network (DCRNN) from the ICLR 2018 paper on data-driven traffic… | 32 | 1439 | maintenance |
| scikit-learn-contrib/metric-learn metric-learn is a Python library providing efficient implementations of popular supervised, weakly-supervised, and unsupervised metric lear… | 56 | 1438 | maintenance |
| VDIGPKU/M2Det M2Det is a PyTorch implementation of a single-shot object detector based on a Multi-Level Feature Pyramid Network (MLFPN), published at AAA… | 32 | 1438 | maintenance |
| AshwinRJ/Federated-Learning-PyTorch A PyTorch implementation of the FedAvg algorithm from the paper 'Communication-Efficient Learning of Deep Networks from Decentralized Data'… | 32 | 1437 | maintenance |
| 3Top/word2vec-api A simple Python web service that exposes word embedding models (Word2Vec/GloVe) via an HTTP API built on Gensim. It serves similarity, most… | 32 | 1437 | maintenance |
| rail-berkeley/softlearning Softlearning is a deep reinforcement learning toolbox for training maximum entropy policies in continuous domains, and the official impleme… | 32 | 1437 | maintenance |
| cmusphinx/sphinx4 Sphinx-4 is a speaker-independent, continuous speech recognition library written entirely in Java, originating from CMU and industry resear… | 32 | 1437 | maintenance |
| yangyanli/PointCNN PointCNN is a deep learning framework for feature learning from 3D point clouds, applying convolution on X-transformed points to handle the… | 55 | 1434 | maintenance |
| pytorch/functorch functorch is a library providing JAX-like composable function transforms (vmap, grad, vjp, jvp) for PyTorch, enabling efficient per-sample … | 10 | 1434 | maintenance |
| hollance/CoreMLHelpers A collection of Swift types and helper functions that simplify working with Apple's Core ML framework, such as image-to-CVPixelBuffer conve… | 32 | 1433 | maintenance |
| denizyuret/Knet.jl Knet (Koç University deep learning framework) is a deep learning library implemented in Julia with GPU support and automatic differentiatio… | 23 | 1432 | maintenance |
| andyzeng/tsdf-fusion-python A lightweight Python script that fuses multiple registered RGB-D images into a projective TSDF voxel volume, from which high-quality 3D sur… | 32 | 1430 | maintenance |
| vlfeat/matconvnet MatConvNet is a MATLAB toolbox implementing convolutional neural networks (CNNs) for computer vision applications. It supports training and… | 32 | 1430 | maintenance |
| SamLynnEvans/Transformer A PyTorch implementation of the Transformer seq2seq model designed to build language translators from parallel corpora. It accompanies a tu… | 32 | 1430 | maintenance |
| qfgaohao/pytorch-ssd A PyTorch implementation of the SSD (Single Shot MultiBox Detector) object detection algorithm with MobileNetV1, MobileNetV2, and VGG backb… | 32 | 1429 | maintenance |
| OpenLMLab/MOSS-RLHF MOSS-RLHF is the open-source companion code for the paper 'Secrets of RLHF in Large Language Models Part I: PPO', providing implementations… | 29 | 1429 | maintenance |
| nianticlabs/simplerecon SimpleRecon is the reference PyTorch implementation of an ECCV 2022 paper for multi-view stereo depth estimation and 3D reconstruction that… | 41 | 1428 | maintenance |
| theAIGuysCode/yolov4-deepsort A Python implementation of multi-object tracking that combines YOLOv4 object detection with the Deep SORT tracking algorithm using TensorFl… | 32 | 1428 | maintenance |
| KaiyangZhou/Dassl.pytorch Dassl is a PyTorch toolbox for research in domain adaptation, domain generalization, and semi-supervised learning. It provides modular comp… | 32 | 1428 | maintenance |
| logpai/loglizer Loglizer is a machine learning toolkit for log-based anomaly detection, implementing classic supervised and unsupervised models (LR, Decisi… | 32 | 1428 | maintenance |
| open-mmlab/mmhuman3d MMHuman3D is an open-source PyTorch-based toolbox and benchmark for 3D human parametric models (e.g., SMPL, SMPL-X) in computer vision and … | 23 | 1426 | maintenance |
| torchgan/torchgan TorchGAN is a PyTorch-based research framework for designing and training Generative Adversarial Networks. It provides modular building blo… | 23 | 1425 | maintenance |
| google-research/disentanglement_lib disentanglement_lib is an open-source Python library for research on learning disentangled representations, supporting models like BetaVAE,… | 10 | 1425 | maintenance |
| sfzhang15/RefineDet RefineDet is a C++/Caffe implementation of the CVPR 2018 single-shot object detection model that refines anchors to combine one-stage speed… | 32 | 1424 | maintenance |
| rmokady/CLIP_prefix_caption Official implementation of ClipCap, a CLIP-based image captioning model that maps CLIP image encodings to a GPT-2 prefix to generate captio… | 32 | 1423 | maintenance |
| gradslam/gradslam gradslam is a fully differentiable dense SLAM library built on PyTorch, providing differentiable building blocks such as nonlinear least sq… | 23 | 1422 | maintenance |
| GOATmessi8/RFBNet PyTorch implementation of RFBNet, a Receptive Field Block Net object detector presented at ECCV 2018. It enhances SSD-style detectors with … | 32 | 1419 | maintenance |
| instructlab/instructlab InstructLab (ilab) is an open-source CLI and Python package for enhancing large language models through community-contributed taxonomy-base… | 10 | 1419 | maintenance |
| thunlp/ERNIE ERNIE is a PyTorch toolkit and dataset for augmenting pre-trained language models like BERT with knowledge graph entity representations, fr… | 32 | 1417 | maintenance |
| SKTBrain/KoBERT KoBERT is a Korean BERT language model pre-trained on Korean Wikipedia by SK Telecom, distributed as a Python package with PyTorch, ONNX, a… | 44 | 1416 | maintenance |
| cdpierse/transformers-interpret A Python library providing model explainability for Hugging Face Transformers models, built on Captum. It explains text classification, que… | 23 | 1416 | maintenance |
| ConnorJL/GPT2 A community Python/TensorFlow implementation of GPT-2 model training and text generation that supports both GPUs and TPUs. It includes scri… | 32 | 1412 | maintenance |
| marl/crepe CREPE is a monophonic pitch tracker based on a deep convolutional neural network that operates directly on time-domain audio waveforms, dis… | 23 | 1411 | maintenance |
| 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 | 1410 | maintenance |
| bermanmaxim/LovaszSoftmax Standalone PyTorch and TensorFlow implementations of the Lovász-Softmax and Lovász Hinge loss functions for optimizing the Jaccard index (I… | 32 | 1410 | maintenance |
| lilipads/gradient_descent_viz A cross-platform C++/Qt desktop application that interactively visualizes five popular gradient descent optimization methods (vanilla GD, m… | 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 | 1407 | 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 |
| uzh-rpg/flightmare Flightmare is a modular quadrotor simulator from the UZH Robotics and Perception Group, composed of a decoupled Unity-based rendering engin… | 23 | 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 |
| 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 |