function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| CSAILVision/gandissect GANDissect is a PyTorch-based toolkit for visualizing and understanding the internal neurons of generative adversarial networks, showing ho… | 32 | 1764 | maintenance |
| OpenCodeInterpreter/OpenCodeInterpreter OpenCodeInterpreter is a suite of open-source code generation models and deployment code that integrates code execution and iterative refin… | 25 | 1763 | maintenance |
| facebookresearch/votenet VoteNet is the official PyTorch implementation of the ICCV 2019 paper 'Deep Hough Voting for 3D Object Detection in Point Clouds'. It provi… | 10 | 1761 | maintenance |
| starry-sky6688/MARL-Algorithms PyTorch implementations of classic multi-agent reinforcement learning algorithms (IQL, QMIX, VDN, COMA, QTRAN, MAVEN, CommNet, DyMA-CL, G2A… | 32 | 1760 | maintenance |
| WXinlong/SOLO Official PyTorch implementation of SOLO and SOLOv2, box-free fully convolutional methods for instance segmentation published at ECCV 2020 a… | 32 | 1758 | maintenance |
| Tencent/Metis Metis is a learnware platform for AIOps from Tencent, currently open-sourcing a time-series anomaly detection learnware. It combines statis… | 10 | 1756 | maintenance |
| enzoampil/fastquant fastquant is a Python library that makes backtesting and optimizing trading strategies accessible with as few as 3 lines of code. It also p… | 32 | 1755 | maintenance |
| salesforce/ALBEF Official PyTorch implementation of ALBEF, a vision-and-language pre-training method that aligns image and text representations before fusin… | 10 | 1754 | maintenance |
| huggingface/transfer-learning-conv-ai A clean, commented PyTorch codebase for training a dialog/chatbot agent by transfer learning from OpenAI GPT/GPT-2 language models. It repr… | 32 | 1753 | maintenance |
| deepset-ai/FARM FARM is a Python framework for fine-tuning and evaluating transformer-based language models for NLP tasks, with a focus on question answeri… | 10 | 1752 | maintenance |
| libfann/fann Fast Artificial Neural Network (FANN) is a free open-source C library implementing multilayer neural networks with fully and sparsely conne… | 73 | 1751 | maintenance |
| Lightning-Universe/lightning-bolts Lightning Bolts is a toolbox of pre-built models, callbacks, and datasets that extend PyTorch Lightning for AI/ML research and production. … | 10 | 1751 | maintenance |
| VertaAI/modeldb ModelDB is an open-source system for machine learning model versioning, metadata tracking, and experiment management. It versions models al… | 23 | 1749 | maintenance |
| magenta/mt3 MT3 is a multi-instrument automatic music transcription model built on the T5X framework, converting audio recordings into multitrack MIDI.… | 73 | 1745 | maintenance |
| ScottfreeLLC/AlphaPy AlphaPy is a Python machine learning framework built on scikit-learn, pandas, Keras, XGBoost, LightGBM, and CatBoost for building classific… | 40 | 1745 | maintenance |
| tkipf/gae A TensorFlow implementation of (Variational) Graph Auto-Encoders from Kipf & Welling's 2016 paper, supporting unsupervised learning, cluste… | 32 | 1743 | maintenance |
| facebookresearch/deepcluster DeepCluster is a research codebase from Facebook AI Research implementing the 'Deep Clustering for Unsupervised Learning of Visual Features… | 10 | 1743 | maintenance |
| google-research/robotics_transformer A Python library from Google Research implementing RT-1 (Robotics Transformer), a transformer model for end-to-end robotic control with an … | 10 | 1743 | maintenance |
| dav/word2vec The original Google word2vec tool, hosted on GitHub with community patches for Mac OS X compilation and memory fixes. It efficiently trains… | 32 | 1741 | maintenance |
| openai/Video-Pre-Training OpenAI's Video PreTraining (VPT) codebase for learning Minecraft agents by watching unlabeled online videos, including behavioral cloning a… | 41 | 1740 | maintenance |
| bearpaw/pytorch-classification A PyTorch library providing a unified interface for training image classification models on CIFAR-10/100 and ImageNet. It includes implemen… | 32 | 1739 | maintenance |
| NVlabs/PWC-Net Official NVIDIA implementation of PWC-Net, a CNN for optical flow estimation using pyramid, warping, and cost volume, released with CVPR 20… | 32 | 1737 | maintenance |
| zijundeng/pytorch-semantic-segmentation A PyTorch library providing implementations of popular semantic segmentation models such as FCN, U-Net, SegNet, PSPNet, GCN, and DUC/HDC. I… | 32 | 1735 | maintenance |
| zhougr1993/DeepInterestNetwork A TensorFlow demo implementation of the Deep Interest Network (DIN) paper for click-through rate prediction, run on the Amazon electronics … | 32 | 1734 | maintenance |
| experiencor/keras-yolo2 A Keras/TensorFlow implementation of the YOLOv2 real-time object detection model with support for training on custom datasets. It offers mu… | 23 | 1733 | maintenance |
| Teachable Machine Teachable Machine is a web-based tool by Google Creative Lab that lets anyone train and use simple machine learning models (image, sound, p… | 72 | 1731 | maintenance |
| ranahanocka/MeshCNN MeshCNN is a PyTorch library implementing a convolutional neural network that operates directly on 3D triangular meshes, with mesh-specific… | 32 | 1731 | maintenance |
| natethegreate/hent-AI A Python application that automatically detects censor bars and mosaic blurs in illustrated adult content using deep learning (Mask R-CNN) … | 23 | 1731 | maintenance |
| NVIDIA/Cosmos-Tokenizer NVIDIA Cosmos Tokenizer is a suite of neural tokenizers for images and videos that convert visual data into continuous latents or discrete … | 10 | 1730 | maintenance |
| cszn/DnCNN DnCNN is the official implementation of the TIP 2017 paper 'Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising',… | 32 | 1729 | maintenance |
| Qihoo360/hbox Hbox (formerly XLearning) is a scheduling platform that runs machine learning and deep learning frameworks like TensorFlow, MXNet, Caffe, a… | 47 | 1728 | maintenance |
| google-research/torchsde A PyTorch library providing differentiable stochastic differential equation (SDE) solvers with GPU support and efficient backpropagation vi… | 10 | 1726 | maintenance |
| bubbliiiing/unet-pytorch A PyTorch implementation of the U-Net semantic segmentation model with training, prediction, and mIoU evaluation scripts. It supports multi… | 23 | 1724 | maintenance |
| Franck-Dernoncourt/NeuroNER NeuroNER is a Python/TensorFlow program that performs named-entity recognition (NER) using neural networks, with pretrained models and an e… | 23 | 1723 | maintenance |
| DeepMotionEditing/deep-motion-editing A PyTorch-based end-to-end library for editing and rendering 3D character motion, built on SIGGRAPH 2020 research. It provides deep learnin… | 32 | 1722 | maintenance |
| Lightning-Universe/lightning-flash Lightning Flash is a high-level PyTorch library built on PyTorch Lightning that provides ready-made 'recipes' for over 15 AI tasks across 7… | 10 | 1722 | maintenance |
| svip-lab/impersonator A PyTorch implementation of Liquid Gating GAN (ICCV 2019) that performs human motion imitation, appearance transfer, and novel view synthes… | 32 | 1717 | maintenance |
| allenai/scibert SciBERT is a BERT-based pretrained language model trained on a large multi-domain corpus of scientific publications, released by the Allen … | 32 | 1717 | maintenance |
| dunbar12138/pix2pix3D pix2pix3D is the official PyTorch implementation of a CVPR 2023 paper on 3D-aware conditional image synthesis. It generates 3D objects (neu… | 31 | 1715 | maintenance |
| sipeed/MaixPy-v1 MaixPy-v1 is a MicroPython port for the Kendryte K210 RISC-V AI chip, letting users program Sipeed Maix boards in Python. It provides APIs … | 23 | 1712 | maintenance |
| Maratyszcza/NNPACK NNPACK is a C99 acceleration package providing high-performance SIMD and multi-core CPU implementations of neural network layers, especiall… | 32 | 1711 | maintenance |
| aiff22/DPED DPED is a Python/TensorFlow implementation of a deep convolutional network approach that translates ordinary smartphone photos into DSLR-qu… | 49 | 1710 | maintenance |
| airaria/TextBrewer TextBrewer is a PyTorch-based knowledge distillation toolkit for NLP models. It provides an easy-to-use framework implementing state-of-the… | 23 | 1709 | maintenance |
| ethz-adrl/control-toolbox The Control Toolbox (CT) is an open-source C++ library for modelling dynamical systems, control, estimation, trajectory optimization and mo… | 23 | 1707 | maintenance |
| CPJKU/madmom Madmom is a Python audio signal processing library with a strong focus on music information retrieval (MIR) tasks. It provides reference im… | 56 | 1706 | maintenance |
| thunlp/OpenNE OpenNE is an open-source Python toolkit for network embedding (graph representation learning) that unifies training and testing interfaces … | 32 | 1705 | maintenance |
| microsoft/i-Code Microsoft's i-Code is a collection of research models and frameworks for integrative, composable multimodal AI spanning vision, language, a… | 32 | 1704 | maintenance |
| Fyusion/LLFF LLFF (Local Light Field Fusion) is a TensorFlow implementation of the SIGGRAPH 2019 paper for novel view synthesis from sparse input images… | 32 | 1703 | maintenance |
| imcaspar/gpt2-ml GPT2-ML is a TensorFlow-based GPT-2 training and inference project supporting multiple languages, with a focus on Chinese. It provides 1.5B… | 23 | 1701 | maintenance |
| NVIDIA/DeepRecommender A research library implementing deep autoencoders for collaborative filtering in recommender systems, based on the paper 'Training Deep Aut… | 10 | 1701 | maintenance |
| bjoernkarmann/project_alias Project Alias is an open-source Raspberry Pi-based device that acts as a 'parasite' on smart home assistants, letting users train custom wa… | 32 | 1700 | maintenance |
| tensorflow/addons TensorFlow Addons is a collection of extra layers, optimizers, losses, metrics, activations, callbacks, and image ops that conform to Tenso… | 10 | 1699 | maintenance |
| hanleyweng/CoreML-in-ARKit A Swift demo/template iOS app that runs CoreML object detection (Inception V3) on live ARKit camera frames and renders 3D text labels above… | 32 | 1696 | maintenance |
| VITA-Group/TransGAN Official PyTorch implementation of TransGAN, a NeurIPS 2021 paper that builds a GAN whose generator and discriminator are both pure transfo… | 32 | 1695 | maintenance |
| ELS-RD/transformer-deploy A Python tool that optimizes Hugging Face transformer models for production inference using ONNX Runtime and NVIDIA TensorRT, deploying the… | 23 | 1690 | maintenance |
| graph4ai/graph4nlp Graph4NLP is a Python library for applying Graph Neural Networks to Natural Language Processing tasks, built on DGL and PyTorch. It provide… | 23 | 1689 | maintenance |
| LoSealL/VideoSuperResolution A Python library (pip-installable as VSR) collecting reimplementation of state-of-the-art single-image and video super-resolution neural ne… | 23 | 1688 | maintenance |
| karpathy/arxiv-sanity-lite A lightweight web application that periodically polls the arXiv API for new papers and lets users tag papers of interest. It recommends sim… | 32 | 1685 | maintenance |
| microsoft/EdgeML A library from Microsoft Research India implementing resource-efficient machine learning algorithms (Bonsai, ProtoNN, FastGRNN, EMI-RNN, DR… | 23 | 1684 | maintenance |
| adobe/antialiased-cnns A PyTorch library providing antialiased CNN models and a BlurPool layer from the ICML 2019 paper 'Making Convolutional Networks Shift-Invar… | 23 | 1683 | maintenance |
| open-mmlab/mmrazor MMRazor is OpenMMLab's model compression toolbox and benchmark built on PyTorch. It provides implementations of neural architecture search,… | 23 | 1683 | maintenance |
| lucidrains/lightweight-gan A PyTorch implementation of the 'lightweight' GAN from ICLR 2021, featuring skip-layer excitation in the generator and autoencoding self-su… | 34 | 1682 | maintenance |
| StarCraft II API Blizzard's official StarCraft II Client API, providing protobuf protocol definitions and a reference C++ library for full external control … | 23 | 1682 | maintenance |
| mit-han-lab/smoothquant SmoothQuant is a training-free post-training quantization (PTQ) method that enables W8A8 (8-bit weight, 8-bit activation) quantization for … | 32 | 1681 | maintenance |
| CuriousAI/mean-teacher Reference implementations (TensorFlow and PyTorch) of the Mean Teacher semi-supervised learning method from the NIPS 2017 paper by Tarvaine… | 32 | 1678 | maintenance |
| rockyzhengwu/FoolNLTK FoolNLTK is a Python toolkit for Chinese natural language processing built on a BiLSTM model. It provides high-accuracy word segmentation, … | 32 | 1677 | maintenance |
| Lam1360/YOLOv3-model-pruning A PyTorch implementation of YOLOv3 channel pruning (network slimming) applied to hand detection on the Oxford Hand dataset. It provides spa… | 32 | 1676 | maintenance |
| argusswift/YOLOv4-pytorch A PyTorch re-implementation of YOLOv4 object detection with variants including attentive YOLOv4 (SEnet, CBAM, CoordAttention) and MobileNet… | 23 | 1676 | maintenance |
| YuliangXiu/ICON ICON is a PyTorch research implementation of a CVPR 2022 method that reconstructs detailed, animatable 3D clothed human avatars from 2D ima… | 23 | 1676 | maintenance |
| lovoo/NSFWDetector A tiny 17 kB CoreML model and Swift library that scans images for nudity/porn content, returning a confidence score. Distributed via CocoaP… | 23 | 1675 | maintenance |
| Kaixhin/Rainbow A PyTorch implementation of the Rainbow deep reinforcement learning agent, combining DQN improvements like prioritized experience replay, d… | 23 | 1673 | maintenance |
| bwapi/bwapi BWAPI is a free and open source C++ framework for interacting with Starcraft: Broodwar, letting students, researchers, and hobbyists build … | 62 | 1672 | maintenance |
| uber/manifold Manifold is a model-agnostic visual debugging tool for machine learning from Uber. It helps ML practitioners identify which subsets of data… | 25 | 1672 | maintenance |
| topepo/caret caret (Classification And Regression Training) is an R package providing a unified interface for training, tuning, and evaluating classific… | 67 | 1669 | maintenance |
| plasticityai/magnitude Magnitude is a Python package and file format for fast, memory-efficient loading and querying of vector embeddings like word2vec, GloVe, an… | 23 | 1669 | maintenance |
| charlesq34/frustum-pointnets Official TensorFlow code release for the CVPR 2018 paper 'Frustum PointNets for 3D Object Detection from RGB-D Data' by Stanford and Nuro r… | 32 | 1668 | maintenance |
| uber-research/deep-neuroevolution Distributed implementations of deep neuroevolution algorithms (ES, NS-ES, NSR-ES, DeepGA, Random Search) from Uber AI Labs research papers,… | 32 | 1666 | maintenance |
| akanazawa/hmr HMR (Human Mesh Recovery) is a TensorFlow implementation of the CVPR 2018 paper 'End-to-end Recovery of Human Shape and Pose', which regres… | 32 | 1666 | maintenance |
| datamllab/tods TODS is a full-stack automated machine learning system for outlier detection on multivariate time-series data, developed by DATA Lab at Ric… | 32 | 1666 | maintenance |
| autonomousvision/occupancy_networks Official PyTorch implementation of the CVPR 2019 paper 'Occupancy Networks: Learning 3D Reconstruction in Function Space'. It learns contin… | 32 | 1664 | maintenance |
| google/aiyprojects-raspbian Google's Python API libraries, samples, and Raspbian system images for the AIY Projects Voice Kit and Vision Kit on Raspberry Pi. It provid… | 10 | 1662 | maintenance |
| hfawaz/dl-4-tsc Companion repository for the paper 'Deep learning for time series classification: a review', providing TensorFlow 2 implementations of nine… | 32 | 1658 | maintenance |
| google-research/pegasus PEGASUS is Google Research's implementation of transformer encoder-decoder models pre-trained with the Gap Sentences Generation objective f… | 10 | 1656 | maintenance |
| rwightman/efficientdet-pytorch A PyTorch implementation of EfficientDet object detection, faithful to the original Google TensorFlow implementation with ported pretrained… | 23 | 1655 | maintenance |
| jasonwei20/eda_nlp EDA (Easy Data Augmentation) is a Python library implementing four simple text augmentation operations—synonym replacement, random insertio… | 32 | 1653 | maintenance |
| HonglinChu/SiamTrackers A PyTorch collection of Siamese-based visual object tracking models including SiamFC, SiamRPN++, SiamMask, Ocean, LightTrack, and the light… | 32 | 1652 | maintenance |
| benhamner/Metrics A multi-language library providing implementations of supervised machine learning evaluation metrics such as AUC, log loss, MAE, RMSE, and … | 32 | 1650 | maintenance |
| CyberPunkMetalHead/Binance-News-Sentiment-Bot A Python trading bot that analyzes news headlines from the top 100 crypto news feeds and places Binance buy/sell orders based on sentiment … | 32 | 1649 | maintenance |
| facebookresearch/CrypTen CrypTen is a PyTorch-based library for privacy-preserving machine learning built on secure multiparty computation. It exposes a CrypTensor … | 10 | 1649 | maintenance |
| vlfeat/vlfeat VLFeat is an open-source C library of popular computer vision algorithms specializing in image understanding and local feature extraction a… | 32 | 1647 | maintenance |
| invictus717/MetaTransformer Meta-Transformer is a research framework for unified multimodal learning that maps inputs from 12 modalities (text, images, point clouds, a… | 19 | 1647 | maintenance |
| amir-abdi/keras_to_tensorflow A Python CLI tool that converts trained Keras models (.h5 or .h5 + .) into frozen, inference-ready TensorFlow protobuf (.pb) files. It free… | 32 | 1646 | maintenance |
| kan-bayashi/ParallelWaveGAN Unofficial PyTorch implementations of non-autoregressive neural vocoders including Parallel WaveGAN, MelGAN, Multi-band MelGAN, HiFi-GAN, a… | 23 | 1646 | maintenance |
| gnobitab/RectifiedFlow Official PyTorch implementation of Rectified Flow, an ICLR 2023 Spotlight method for learning transport maps between distributions via stra… | 32 | 1645 | maintenance |
| ai-forever/ru-dalle ruDALL-E is a Python library for generating images from Russian text prompts using DALL-E-style transformer models like Malevich, Emojich, … | 23 | 1642 | maintenance |
| Coder-Yu/QRec QRec is a Python framework built on TensorFlow 1.x for quickly implementing and evaluating recommender system models. It ships with many st… | 32 | 1638 | maintenance |
| lucidrains/alphafold2 An unofficial PyTorch implementation of DeepMind's AlphaFold2 protein structure prediction network, built as a translation of the official … | 23 | 1635 | maintenance |
| autonomio/talos Talos is a Python library for automating hyperparameter experiments and model evaluation with TensorFlow (tf.keras), Keras, and PyTorch mod… | 23 | 1634 | maintenance |
| facebookresearch/consistent_depth A research library from Facebook AI Research implementing Consistent Video Depth Estimation (SIGGRAPH 2020). It reconstructs dense, flicker… | 10 | 1634 | maintenance |
| tensorflow/mesh Mesh TensorFlow is a Python library and embedded language for specifying distributed tensor computations, letting users define how model di… | 10 | 1630 | maintenance |