function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| 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 |
| tinyvision/SOLIDER SOLIDER is a semantic-controllable self-supervised learning framework that learns general human representations from massive unlabeled huma… | 32 | 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 |
| AITTSMD/MTCNN-Tensorflow A TensorFlow reproduction of MTCNN (Multi-task Cascaded Convolutional Networks) for joint face detection and facial landmark alignment. It … | 32 | 1502 | 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 |
| luuuyi/CBAM.PyTorch A non-official PyTorch re-implementation of the CBAM (Convolutional Block Attention Module) paper from ECCV 2018. It provides channel and s… | 32 | 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 |
| filipradenovic/cnnimageretrieval-pytorch A PyTorch toolbox for training and evaluating convolutional neural networks for image retrieval, implementing the authors' TPAMI 2018 and E… | 23 | 1496 | 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 | 1496 | maintenance |
| DeepVoltaire/AutoAugment An unofficial Python implementation of the AutoAugment data augmentation policies learned for ImageNet, CIFAR-10, and SVHN, built on Pillow… | 32 | 1493 | maintenance |
| king-yyf/CMeKG_tools A collection of pretrained models and Python code for the Chinese Medical Knowledge Graph (CMeKG), providing medical text segmentation, med… | 32 | 1492 | 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 |
| thu-ml/unidiffuser UniDiffuser is a unified diffusion framework that fits marginal, conditional, and joint distributions across image and text modalities in a… | 30 | 1487 | maintenance |
| cvg/pixel-perfect-sfm pixsfm is a Python package with a C++ core that improves Structure-from-Motion and visual localization accuracy by refining keypoints, came… | 23 | 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 | 1485 | maintenance |
| AIRLegend/aitrack AITrack is a free 6DoF head tracking application that uses a webcam and neural networks to estimate head position and rotation. It streams … | 23 | 1485 | 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 |
| 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 |
| karpathy/reinforcejs REINFORCEjs is a JavaScript library implementing common reinforcement learning algorithms including Dynamic Programming, Temporal Differenc… | 32 | 1481 | 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 |
| YuanGongND/ast Official PyTorch implementation of the Audio Spectrogram Transformer (AST) from the Interspeech 2021 paper, which applies a Vision Transfor… | 32 | 1474 | 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 |
| marian-nmt/marian Marian is an efficient Neural Machine Translation framework written in pure C++ with minimal dependencies, supporting fast multi-GPU traini… | 23 | 1471 | 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 |
| 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 | 1467 | maintenance |
| lxztju/pytorch_classification A complete PyTorch image classification codebase built on torchvision, covering training, prediction, TTA, model ensembling, knowledge dist… | 32 | 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 |
| CStanKonrad/long_llama LongLLaMA is a large language model handling long contexts of 256k tokens or more, built on OpenLLaMA and fine-tuned with the Focused Trans… | 29 | 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 | 1464 | 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 |
| szagoruyko/attention-transfer PyTorch reference implementation of the ICLR 2017 paper 'Paying More Attention to Attention', which improves convolutional neural networks … | 32 | 1463 | maintenance |
| datitran/face2face-demo A pix2pix demo that learns from facial landmarks and translates them into a trained target face, with a real-time webcam application. It in… | 32 | 1461 | 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 | 1461 | 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 |
| wvangansbeke/Unsupervised-Classification PyTorch implementation of SCAN (ECCV 2020), a two-step method for unsupervised image classification that combines self-supervised represent… | 32 | 1457 | maintenance |
| buaacyw/GaussianEditor GaussianEditor is a research application for fast, controllable 3D scene editing built on Gaussian Splatting, released as CVPR 2024 code. I… | 27 | 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 | 1455 | 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 |
| xuannianz/EfficientDet A Keras/TensorFlow implementation of the EfficientDet object detection model with pretrained COCO and ImageNet weights. It supports trainin… | 32 | 1452 | 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 |
| pytorch/gloo Gloo is a C++ collective communications library providing primitives like barrier, broadcast, and allreduce for multi-machine machine learn… | 76 | 1449 | 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 |
| 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 |
| 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 |
| hysts/pytorch_image_classification A PyTorch library implementing many image classification architectures (ResNet, DenseNet, WRN, PyramidNet, SENet, etc.) and augmentation te… | 10 | 1448 | 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 |
| ZHO-ZHO-ZHO/ComfyUI-InstantID An unofficial ComfyUI custom node implementation of InstantID for identity-preserving image generation with Stable Diffusion XL. It provide… | 26 | 1444 | maintenance |
| ZPdesu/Barbershop Official PyTorch implementation of Barbershop (SIGGRAPH Asia 2021), a GAN-inversion-based method for compositing images using segmentation … | 32 | 1440 | maintenance |
| rail-berkeley/softlearning Softlearning is a deep reinforcement learning toolbox for training maximum entropy policies in continuous domains, and the official impleme… | 32 | 1440 | 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 | 1440 | 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 | 1439 | 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 |
| wenhaochai/StableVideo StableVideo is the official ICCV 2023 implementation of text-driven, consistency-aware diffusion-based video editing built on ControlNet an… | 31 | 1439 | 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 |
| mchong6/JoJoGAN JoJoGAN is the official PyTorch implementation of a one-shot face stylization method that finetunes a pretrained StyleGAN using GAN inversi… | 32 | 1438 | 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 |
| 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 |
| 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 |
| xuebinqin/BASNet BASNet is the official PyTorch implementation of the CVPR 2019 paper 'BASNet: Boundary-Aware Salient Object Detection', a deep learning mod… | 32 | 1436 | 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 |
| 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 | 1433 | 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 | 1432 | maintenance |
| vlfeat/matconvnet MatConvNet is a MATLAB toolbox implementing convolutional neural networks (CNNs) for computer vision applications. It supports training and… | 32 | 1431 | 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 | 1431 | 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 | 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 | 1429 | 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 |
| KaiyangZhou/Dassl.pytorch Dassl is a PyTorch toolbox for research in domain adaptation, domain generalization, and semi-supervised learning. It provides modular comp… | 32 | 1429 | maintenance |
| facebookresearch/diplomacy_cicero Research code and model checkpoints for Cicero and Diplodocus, AI agents that play the board game Diplomacy at human level by combining lan… | 10 | 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 |
| logpai/loglizer Loglizer is a machine learning toolkit for log-based anomaly detection, implementing classic supervised and unsupervised models (LR, Decisi… | 32 | 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 |
| 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 | 1425 | 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 |
| gradslam/gradslam gradslam is a fully differentiable dense SLAM library built on PyTorch, providing differentiable building blocks such as nonlinear least sq… | 23 | 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 |
| 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 | 1420 | 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 |
| 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 |
| cdpierse/transformers-interpret A Python library providing model explainability for Hugging Face Transformers models, built on Captum. It explains text classification, que… | 23 | 1417 | maintenance |
| BangguWu/ECANet Official PyTorch implementation of ECA-Net, an efficient channel attention module for deep convolutional neural networks published at CVPR … | 32 | 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 | 1413 | 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 | 1413 | 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 | 1410 | maintenance |
| yu-changqian/TorchSeg A fast, modular PyTorch reference implementation for training and evaluating semantic segmentation models such as FCN, DFN, BiSeNet, PSPNet… | 23 | 1410 | 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 | 1410 | maintenance |