function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| Kosinkadink/ComfyUI-Advanced-ControlNet A set of ComfyUI custom nodes providing advanced ControlNet scheduling, weighting, and masking for Stable Diffusion workflows. It supports … | 71 | 1007 | active |
| pytorch/ELF ELF is an extensive, lightweight, and flexible C++/Python platform for game research, best known for ELF OpenGo, a full AlphaGoZero/AlphaZe… | 10 | 3411 | maintenance |
| facebookresearch/fairscale FairScale is a PyTorch extension library providing composable modules and APIs for high-performance, large-scale distributed training, incl… | 10 | 3406 | maintenance |
| meetps/pytorch-semseg A PyTorch library implementing popular semantic segmentation architectures such as FCN, U-Net, SegNet, PSPNet, ICNet, FRRN, and LinkNet, wi… | 23 | 3402 | maintenance |
| minimaxir/gpt-2-simple A Python package that simplifies fine-tuning OpenAI's GPT-2 text-generation model (124M/355M) on custom text and generating text from the r… | 23 | 3400 | maintenance |
| RIFE RIFE is a deep learning model for real-time video frame interpolation, estimating intermediate flow between frames to generate smooth slow-… | 77 | 1004 | active |
| huawei-noah/noah-research A collection of research code subprojects released by Huawei Noah's Ark Lab, each in its own directory. It is not an official Huawei produc… | 76 | 1004 | active |
| TinyLLaVA/TinyLLaVA_Factory TinyLLaVA Factory is an open-source modular PyTorch/HuggingFace codebase for training small-scale large multimodal models (LMMs) that combi… | 68 | 1004 | active |
| SciSharp/TensorFlow.NET TensorFlow.NET provides .NET Standard bindings for Google's TensorFlow, aiming to implement the complete TensorFlow API in C# and F#. It in… | 24 | 3397 | maintenance |
| PrithivirajDamodaran/FlashRank FlashRank is an ultra-light, fast Python library for re-ranking search and retrieval results using SoTA cross-encoders and listwise LLM-bas… | 58 | 1003 | active |
| ZJUI-AI4H/Hulu-Med Hulu-Med is a family of open-source transparent generalist medical vision-language models ranging from 4B to 235B parameters, covering text… | 62 | 1001 | active |
| pjlab-sys4nlp/llama-moe LLaMA-MoE is a Python toolkit and model series for building Mixture-of-Experts (MoE) language models from dense LLaMA models via expert con… | 19 | 1001 | active |
| catalyst-team/catalyst Catalyst is a high-level PyTorch framework for deep learning research and development, focused on reproducibility, rapid experimentation, a… | 64 | 3382 | maintenance |
| mit-han-lab/efficientvit A collection of efficient vision foundation models from MIT Han Lab, including EfficientViT backbones for perception, EfficientViT-SAM for … | 48 | 3358 | maintenance |
| eladrich/pixel2style2pixel Official PyTorch implementation of pixel2style2pixel (pSp), a StyleGAN encoder from CVPR 2021 that maps real images directly into the W+ la… | 32 | 3350 | maintenance |
| tianzhi0549/FCOS Official PyTorch implementation of FCOS, a fully convolutional one-stage, anchor-free object detector published at ICCV 2019. It provides t… | 32 | 3345 | maintenance |
| tamarott/SinGAN Official PyTorch implementation of SinGAN, an ICCV 2019 best-paper generative model trained on a single natural image. It learns patch stat… | 32 | 3344 | maintenance |
| hamuchiwa/AutoRCCar An open-source project that turns a hobby RC car into an autonomous self-driving vehicle using a Raspberry Pi, Arduino, camera, and ultraso… | 32 | 3342 | maintenance |
| NVlabs/eg3d Official PyTorch implementation of EG3D, an efficient geometry-aware 3D generative adversarial network from NVIDIA Research (CVPR 2022). It… | 32 | 3337 | maintenance |
| HRNet/HRNet-Semantic-Segmentation Official PyTorch implementation of HRNet (High-Resolution Network) and the Segmentation Transformer (OCR) approach for semantic segmentatio… | 32 | 3330 | maintenance |
| run-youngjoo/SC-FEGAN SC-FEGAN is a GUI application that uses a generative adversarial network (SN-patchGAN discriminator with a U-Net generator) to edit face im… | 32 | 3329 | maintenance |
| JIA-Lab-research/MGM Official PyTorch implementation of Mini-Gemini, a multimodal vision-language model framework built on LLaVA that supports dense and MoE LLM… | 25 | 3326 | maintenance |
| pytorch/glow Glow is a machine learning compiler and execution engine for neural network hardware accelerators, developed under the PyTorch project. It … | 10 | 3319 | maintenance |
| PixArt-alpha/PixArt-alpha PixArt-α is a Transformer-based text-to-image diffusion model with PyTorch model definitions, pre-trained weights, and inference/training c… | 27 | 3304 | maintenance |
| open-mmlab/mmselfsup OpenMMLab's PyTorch-based toolbox and benchmark for self-supervised and unsupervised visual representation learning. It provides implementa… | 23 | 3303 | maintenance |
| resemble-ai/Resemblyzer Resemblyzer is a Python package that uses a deep learning voice encoder to convert speech audio into 256-dimensional voice embeddings. Thes… | 23 | 3302 | maintenance |
| mit-biomimetics/Cheetah-Software MIT Biomimetics' software for the Cheetah 3 and Mini Cheetah quadruped robots, containing a dynamics/common library, robot control programs… | 32 | 3299 | maintenance |
| bytedance/lightseq LightSeq is a high-performance CUDA-based library for training and inference of sequence models like Transformer, BERT, GPT, and BART, with… | 10 | 3295 | maintenance |
| tradytics/eiten Eiten is an open-source Python toolkit for building statistical and algorithmic investment portfolios, implementing strategies like Eigen P… | 32 | 3293 | maintenance |
| facebookresearch/vissl VISSL is Facebook AI Research's extensible, modular and scalable PyTorch library for state-of-the-art self-supervised learning with images.… | 10 | 3293 | maintenance |
| NVIDIA/flownet2-pytorch A PyTorch implementation of FlowNet 2.0 for deep-learning-based optical flow estimation, released by NVIDIA. It provides multiple network a… | 66 | 3292 | maintenance |
| google-deepmind/dm-haiku Haiku is a simple neural network library for JAX from Google DeepMind, providing an object-oriented module abstraction (hk.Module) and a fu… | 89 | 3278 | maintenance |
| google-research/albert Official TensorFlow implementation and pretrained checkpoints of ALBERT, a lite version of BERT for self-supervised learning of language re… | 10 | 3278 | maintenance |
| zhanghang1989/ResNeSt ResNeSt is a PyTorch implementation of the Split-Attention Network, a ResNet variant that applies channel-wise attention across network bra… | 23 | 3261 | maintenance |
| guoguibing/librec LibRec is a Java library for recommender systems implementing over 70 state-of-the-art recommendation algorithms. It supports the two class… | 23 | 3256 | maintenance |
| anandpawara/Real_Time_Image_Animation A real-time Python application that animates a still image (e.g., a portrait) using facial motion from a live camera or video file, built o… | 32 | 3247 | maintenance |
| martinarjovsky/WassersteinGAN Reference PyTorch implementation of the Wasserstein GAN paper, providing training scripts for DCGAN and MLP architectures on datasets like … | 32 | 3244 | maintenance |
| facebookresearch/MUSE MUSE is a Python library from Facebook AI Research for creating and aligning multilingual word embeddings, using both supervised (bilingual… | 10 | 3244 | maintenance |
| google/model_search Model Search is a Google AutoML framework that implements neural architecture search algorithms at scale to find optimal DNN architectures … | 10 | 3238 | maintenance |
| google/brax Brax is a fast, fully differentiable rigidbody physics engine written in JAX, designed for massively parallel simulation on accelerator har… | 85 | 3229 | maintenance |
| JoePenna/Dreambooth-Stable-Diffusion A Jupyter Notebook-based implementation of Dreambooth fine-tuning for Stable Diffusion, adapted from XavierXiao's repo with tweaks for trai… | 32 | 3212 | maintenance |
| mkocabas/VIBE Official PyTorch implementation of VIBE (CVPR 2020), a video-based method for 3D human body pose and shape estimation that predicts SMPL bo… | 23 | 3212 | maintenance |
| tomgoldstein/loss-landscape PyTorch code for visualizing the loss landscape of neural networks, from the NIPS 2018 paper 'Visualizing the Loss Landscape of Neural Nets… | 32 | 3201 | maintenance |
| eureka-research/Eureka Eureka is a research library from NVIDIA that uses coding LLMs like GPT-4 to automatically generate reward functions for reinforcement lear… | 27 | 3192 | maintenance |
| tinyvision/DAMO-YOLO DAMO-YOLO is a fast and accurate object detection framework built on PyTorch, featuring NAS-searched backbones, RepGFPN, a lightweight Zero… | 32 | 3187 | maintenance |
| yusugomori/DeepLearning A multi-language (Python, C, C++, Java, Scala, Go) educational implementation of classic deep learning algorithms such as Deep Belief Nets,… | 32 | 3180 | maintenance |
| adambielski/siamese-triplet A PyTorch library implementing siamese and triplet networks for learning image embeddings, with online pair/triplet mining strategies. It p… | 32 | 3173 | maintenance |
| jettify/pytorch-optimizer torch-optimizer is a Python library providing a collection of additional optimization algorithms for PyTorch, drop-in compatible with the t… | 23 | 3171 | maintenance |
| cemoody/lda2vec A Python library implementing lda2vec, a hybrid topic model that combines word2vec word embeddings with LDA-style interpretable document to… | 32 | 3169 | maintenance |
| farizrahman4u/seq2seq A sequence-to-sequence learning add-on library for Keras, providing modular encoder-decoder layers and ready-made Seq2Seq models. It suppor… | 32 | 3169 | maintenance |
| huawei-noah/Pretrained-Language-Model A collection of pretrained language models and optimization techniques from Huawei Noah's Ark Lab, including PanGu-α (200B-parameter Chines… | 32 | 3166 | maintenance |
| google-research/frame-interpolation FILM is the official TensorFlow 2 implementation of a state-of-the-art frame interpolation neural network from Google Research, presented a… | 10 | 3151 | maintenance |
| project-baize/baize-chatbot Baize is an open-source chat model built on LLaMA using LoRA parameter-efficient tuning, trained on 100k self-chat dialogs generated by Cha… | 21 | 3148 | maintenance |
| migueldeicaza/TensorFlowSharp TensorFlowSharp provides .NET bindings to the TensorFlow C API, exposing a strongly-typed low-level API for C# and F#. It is designed mainl… | 10 | 3145 | maintenance |
| DAMO-NLP-SG/Video-LLaMA Video-LLaMA is an instruction-tuned audio-visual language model that extends LLaMA with video and audio understanding via cross-modal pretr… | 29 | 3141 | maintenance |
| microsoft/torchscale A PyTorch library from Microsoft implementing foundation Transformer architectures such as DeepNet, Magneto, RetNet, LongNet, BitNet, and X… | 32 | 3139 | maintenance |
| FudanNLP/fastNLP fastNLP is a lightweight, modularized and extensible NLP framework in Python that reduces engineering boilerplate such as data processing l… | 23 | 3139 | maintenance |
| open-mmlab/mmdeploy MMDeploy is the OpenMMLab model deployment framework that converts PyTorch-based OpenMMLab models (mmdetection, mmsegmentation, etc.) into … | 23 | 3139 | maintenance |
| google-deepmind/trfl TRFL is a Python library built on TensorFlow that provides building-block loss operations (e.g., Q-learning, TD learning, distributional RL… | 32 | 3130 | maintenance |
| open-mmlab/mmskeleton MMSkeleton is an OpenMMLAB toolbox for skeleton-based human understanding, built on PyTorch. It supports 2D pose estimation, skeleton-based… | 23 | 3129 | maintenance |
| diegoantognini/pyGAT A PyTorch implementation of the Graph Attention Network (GAT) model from Veličković et al. (2017), including a sparse-matrix variant. It re… | 32 | 3122 | maintenance |
| Habitat AI Habitat is a high-performance, physics-enabled 3D simulation platform for Embodied AI research, consisting of Habitat-Sim (a fast 3D sim… | 74 | 3118 | maintenance |
| leggedrobotics/legged_gym A Python library providing Isaac Gym environments for training legged robots (e.g., ANYmal) to walk on rough terrain with reinforcement lea… | 43 | 3112 | maintenance |
| dbiir/UER-py UER-py is a PyTorch framework for pre-training transformer language models (BERT, GPT-2, T5, ELMo, etc.) and fine-tuning them on downstream… | 32 | 3112 | maintenance |
| lucasjinreal/yolov7_d2 A detectron2-based implementation of YOLOv7 that extends YOLO-style detection to instance segmentation, keypoint detection, and multi-head … | 23 | 3108 | maintenance |
| cysmith/neural-style-tf A TensorFlow implementation of neural style transfer based on Gatys et al.'s convolutional neural network approach, with support for video … | 32 | 3104 | maintenance |
| salesforce/CodeT5 Official research release of CodeT5 and CodeT5+ open code large language models from Salesforce Research for code understanding and generat… | 10 | 3093 | maintenance |
| aksnzhy/xlearn xLearn is a high-performance C++ machine learning package implementing linear models (LR), factorization machines (FM), and field-aware fac… | 23 | 3089 | maintenance |
| tusen-ai/simpledet SimpleDet is a Python framework built on MXNet for object detection and instance recognition. It provides state-of-the-art detection models… | 32 | 3085 | maintenance |
| shogun-toolbox/shogun Shogun is a C++ machine learning toolbox offering unified, efficient implementations of many ML algorithms with automatically generated bin… | 23 | 3080 | maintenance |
| qwopqwop200/GPTQ-for-LLaMa A Python library that applies GPTQ 4-bit weight quantization to LLaMA large language models, drastically reducing memory usage and checkpoi… | 30 | 3072 | maintenance |
| Tramac/awesome-semantic-segmentation-pytorch A PyTorch library providing concise, modifiable reference implementations of many semantic segmentation models such as FCN, PSPNet, DeepLab… | 32 | 3070 | maintenance |
| argman/EAST A TensorFlow re-implementation of the EAST (Efficient and Accurate Scene Text Detector) deep learning model for detecting text in natural s… | 32 | 3059 | maintenance |
| stellargraph/stellargraph StellarGraph is a Python library for machine learning on graphs and networks, offering state-of-the-art graph neural network algorithms suc… | 23 | 3059 | maintenance |
| cvlab-columbia/zero123 Zero-1-to-3 is a research codebase and pretrained diffusion model from Columbia CVLab that changes the camera viewpoint of an object from a… | 30 | 3058 | maintenance |
| rinongal/textual_inversion Official implementation of the Textual Inversion paper, which learns new word embeddings in a frozen text-to-image (Latent Diffusion) model… | 32 | 3055 | maintenance |
| avinashpaliwal/Super-SloMo A PyTorch implementation of the Super SloMo paper for high-quality video frame interpolation, generating multiple intermediate frames to co… | 10 | 3024 | maintenance |
| microsoft/human-pose-estimation.pytorch Official PyTorch implementation of the ECCV 2018 paper 'Simple Baselines for Human Pose Estimation and Tracking' from Microsoft. It provide… | 10 | 3008 | maintenance |
| weld-project/weld Weld is a language and high-performance runtime that optimizes data-intensive analytics workflows by expressing computations in a common in… | 57 | 3006 | maintenance |
| CodeRayZhang/Movie_Recommend A full-stack movie recommendation system built on Spark, including a Scrapy crawler, an SSM-based movie website, an admin backend, and a Sp… | 32 | 3006 | maintenance |
| divamgupta/image-segmentation-keras A Keras library implementing popular deep learning semantic image segmentation models including SegNet, FCN, U-Net, and PSPNet. It provides… | 23 | 3003 | maintenance |
| jfzhang95/pytorch-deeplab-xception A PyTorch implementation of the DeepLab v3+ semantic segmentation model with support for multiple backbones (Xception, ResNet, MobileNet, D… | 32 | 3000 | maintenance |
| BayesWitnesses/m2cgen m2cgen is a lightweight Python library that transpiles trained statistical machine learning models into native code in languages such as Py… | 23 | 3000 | maintenance |
| google-research/t5x T5X is a modular, composable framework built on JAX and Flax for high-performance training, evaluation, and inference of sequence models at… | 75 | 2998 | maintenance |
| yangjianxin1/GPT2-chitchat A GPT2-based Chinese chitchat dialogue model project built on HuggingFace transformers, including training, preprocessing, and interactive … | 32 | 2996 | maintenance |
| keithito/tacotron An unofficial open-source TensorFlow implementation of Google's Tacotron end-to-end neural text-to-speech model, with a pre-trained model a… | 23 | 2995 | maintenance |
| libffcv/ffcv FFCV is a fast data loading system for PyTorch that dramatically increases data throughput in model training by replacing standard data loa… | 23 | 2992 | maintenance |
| biubug6/Pytorch_Retinaface A PyTorch implementation of the RetinaFace single-stage face detection model, supporting mobilenet0.25 and resnet50 backbones with pretrain… | 32 | 2977 | maintenance |
| replicate/scribble-diffusion Scribble Diffusion is an open-source Next.js web application that turns rough sketches into refined images using the ControlNet scribble mo… | 48 | 2976 | maintenance |
| Tencent/FaceDetection-DSFD DSFD (Dual Shot Face Detector) is Tencent Youtu's high-accuracy face detection network, released with PyTorch inference code and pretrained… | 56 | 2971 | maintenance |
| johnma2006/mamba-minimal A single-file, readable PyTorch implementation of the Mamba selective state space model architecture. It produces numerically equivalent ou… | 26 | 2967 | maintenance |
| Cartucho/mAP A Python library and script that computes mean Average Precision (mAP) for object detection models, adapted from the official PASCAL VOC 20… | 23 | 2966 | maintenance |
| NVIDIA/MinkowskiEngine Minkowski Engine is an auto-differentiation neural network library for high-dimensional sparse tensors, built on PyTorch with CUDA accelera… | 23 | 2958 | maintenance |
| FreedomIntelligence/LLMZoo LLM Zoo is a project providing data, models, and evaluation benchmarks for large language models, including the multilingual Phoenix and Ch… | 30 | 2934 | maintenance |
| pytorch/audio TorchAudio is PyTorch's audio library providing data manipulation, transforms, and dataset loaders for audio and speech machine learning. I… | 87 | 2933 | maintenance |
| rail-berkeley/rlkit RLkit is a PyTorch-based reinforcement learning framework and algorithm collection from UC Berkeley RAIL. It provides reference implementat… | 32 | 2932 | maintenance |
| Handtrack Handtrack.js is a JavaScript library for prototyping realtime hand detection directly in the browser, framing handtracking as an object det… | 23 | 2930 | maintenance |
| ajbrock/BigGAN-PyTorch A PyTorch implementation of BigGAN for large-scale high-fidelity image synthesis, written by the paper's first author. It supports multi-GP… | 32 | 2923 | maintenance |
| facebookresearch/XLM PyTorch implementation of Cross-lingual Language Model Pretraining (XLM) from Facebook AI Research, covering MLM, CLM, and TLM objectives p… | 10 | 2920 | maintenance |
| Tencent/PhoenixGo PhoenixGo is a Go (weiqi/baduk) AI engine from Tencent that implements the AlphaGo Zero approach of learning the game without human knowled… | 23 | 2917 | maintenance |