Topics / #model-training

Model-training projects on GitHub

Training machine learning models on datasets.

As of , the most-starred Model-training projects on GitHub are yolov5 (58.2K stars), Retrieval-based-Voice-Conversion-WebUI (38.8K stars) and open-r1 (26.5K stars). GitHub.Locker tracks 28 Model-training repositories with 307.4K stars combined. The fastest-growing on October 16 was Retrieval-based-Voice-Conversion-WebUI, with +23 stars in a day.

28Projects
307.4KStars in total
+163Stars in a day

Hot right now

Most stars gained in a day · Oct 16

The all-time top

28 projects · by stars

  1. #1 · ultralytics/yolov5

    yolov5

    YOLOv5 is a fast, accurate, and easy-to-use computer vision model from Ultralytics for object detection, instance segmentation, and image classification, built on PyTorch.

    58.2K
  2. #2 · RVC-Project/Retrieval-based-Voice-Conversion-WebUI

    Retrieval-based-Voice-Conversion-WebUI

    RVC WebUI is a web-based tool for training and running voice conversion (voice-changing) models, letting you clone a target voice's timbre using as little as about 10 minutes of audio.

    38.8K
  3. #3 · huggingface/open-r1

    open-r1

    Open R1 is a fully open effort to reproduce the training pipeline behind DeepSeek-R1, providing scripts to run SFT and GRPO training, generate synthetic reasoning data, and evaluate the resulting reasoning models.

    26.5K
  1. 4

    examples

    24.1Kpytorch/examples · Python

    A curated set of short, standalone PyTorch example projects covering computer vision, text, and reinforcement learning, each meant to be emulated in your own work.

  2. 5

    Megatron-LM

    18.2KNVIDIA/Megatron-LM · Python

    NVIDIA's open-source toolkit for training large transformer (LLM) models at scale. It comes as Megatron-LM, a reference setup with ready training scripts, and Megatron Core, a composable library of GPU-optimized building blocks for training models across thousands of GPUs.

  3. 6

    vision

    18Kpytorch/vision · Python

    Torchvision is PyTorch's computer vision package: popular datasets, ready-made model architectures, and common image transformations for vision work.

  4. 7

    StableLM

    15.7KStability-AI/StableLM · Jupyter Notebook

    StableLM is Stability AI's family of open-source language models, including the StableLM-3B-4E1T model, StableLM-Alpha v1/v2 base and tuned models, and StableVicuna, an RLHF-tuned chatbot built on Vicuna-13B.

  5. 8

    DiffSynth-Studio

    13.3Kmodelscope/DiffSynth-Studio · Python

    DiffSynth-Studio is an open-source Diffusion model engine built and maintained by the ModelScope Community, covering image, video, and audio generation plus image quality metrics, with support for both inference and training.

  6. 9

    ai-toolkit

    12.3Kostris/ai-toolkit · Python

    AI Toolkit is an all-in-one training suite for finetuning diffusion models, covering image, video, and audio generation models, with support for the latest models like FLUX, Qwen-Image, Wan 2.1/2.2, and Ace-Step on consumer GPUs.

  7. 10

    train-llm-from-scratch

    12.2KFareedKhan-dev/train-llm-from-scratch · Python

    A from-scratch tutorial and codebase for training your own LLM end to end: tokenizing raw text, pretraining a GPT-style Transformer, and then post-training it with SFT, a reward model, DPO/ORPO/KTO, PPO and GRPO into an aligned, reasoning-style assistant.

  8. 11

    allennlp

    11.9Kallenai/allennlp · Python

    AllenNLP is an open-source NLP research library built on PyTorch for developing state-of-the-art deep learning models across a wide range of linguistic tasks.

  9. 12

    dolly

    10.8Kdatabrickslabs/dolly · Python

    Dolly is Databricks' instruction-following large language model, trained on the Databricks Machine Learning Platform and licensed for commercial use; the flagship dolly-v2-12b derives from EleutherAI's Pythia-12b.

  10. 13

    yolov3

    10.6Kultralytics/yolov3 · Python

    Ultralytics' PyTorch implementation of the YOLOv3 real-time object detector, packaging three model variants - YOLOv3, YOLOv3-SPP and YOLOv3-tiny - with training, validation, inference and multi-format export tooling.

  11. 14

    reef

    7.6KHuman-Agent-Society/reef · Python

    Reef is open-source infrastructure for continually self-improving agents. It wires together agent inference, feedback collection, learning, and versioned delivery so an agent keeps getting better from how you interact with it.

  12. 15

    ANE

    7.3Kmaderix/ANE · Objective-C

    ANE is a research project that trains neural networks directly on Apple's Neural Engine using reverse-engineered private APIs, instead of CoreML's inference-only path.

  13. 16

    DeepSpec

    7.2Kdeepseek-ai/DeepSpec · Python

    DeepSpec is DeepSeek's full-stack Python codebase for training and evaluating draft models for speculative decoding — the technique where a small draft model accelerates a larger target LLM.

  14. 17

    higgsfield

    6.1Khiggsfield-ai/higgsfield · Jupyter Notebook

    Higgsfield is an open-source, fault-tolerant GPU orchestration system and machine learning framework for training huge models, from billions to trillions of parameters, such as LLMs.

  15. 18

    Attention-Residuals

    3.5KMoonshotAI/Attention-Residuals

    AttnRes (Attention Residuals) is a drop-in replacement for standard residual connections in Transformers, letting each layer use learned, input-dependent attention over all previous layer outputs instead of simple additive accumulation.

  16. 19

    mini-AGI

    1.2Kvolotat/mini-AGI · Python

    mini-AGI is a continual learning, byte-level language model that assembles its own architecture, trains from scratch on a single 8GB VRAM GPU, and keeps learning from everything it reads, without a separate fine-tuning phase.

  17. 20

    Valen

    843Liuziyu77/Valen · Jupyter Notebook

    Valen (万澜) is a multimodal decision model that adds visual perception to System One-style decision-making: it scores candidate actions against text, images and video using a task instruction, instead of generating free-form answers.

  18. 21

    Intern-Decision

    811InternLM/Intern-Decision · Python

    Intern-Decision turns a state, optional images, and multiple typed questions into decisions with probabilities, by fine-tuning the language backbone of Qwen3.5 while freezing its vision tower and projector.

  19. 22

    SolarWM

    777Junchao-cs/SolarWM · Python

    SolarWM is a fully open foundation for building interactive video world models, covering data preparation, scalable multi-backbone training, and long-horizon inference.

  20. 23

    Meshy

    387OpenBMB/Meshy · Python

    Meshy is an open-source framework for running reinforcement learning (RL) training jobs where every role (inference, training, rollout) runs as an independent service, communicating through a shared TransferQueue data plane. It is built on SGLang and torchtitan.

  21. 24

    nnUNet_att_position_correction

    354qpuchen/nnUNet_att_position_correction · Python

    This is the official PyTorch code for the MICCAI 2023 STS Challenge (3D tooth segmentation from CBCT scans) by team sdkxd, which extends nn-UNet with axial attention and a positional correction module to clean up abnormal segmentation points.

  22. 25

    LittleBit

    283SamsungLabs/LittleBit · Python

    LittleBit is the official research code for compressing large language models to sub-1-bit precision (down to 0.1 bits per weight) by factorizing each weight matrix into low-rank latent factors and binarizing them, with magnitude restored through learned scales. It also includes LittleBit-2, a follow-up that improves initialization via latent geometry alignment.

  23. 26

    video-prediction-policy-2

    249roboterax/video-prediction-policy-2 · Python

    Video Prediction Policy 2 (VPP2) is a world-action model (WAM) for robot manipulation that predicts future video frames and converts them into robot actions, with strong zero-shot generalization across tasks.

  24. 27

    pointdit

    233google-research/pointdit · Python

    PointDiT is a research codebase for a minimalist pixel-space Diffusion Transformer that estimates 3D geometry (point maps) from a single monocular image.

  25. 28

    UniVR

    178FlagOpen/UniVR · Python

    UniVR is an end-to-end SFT and Reinforcement Learning (GRPO) training framework built specifically for Emu3.5 series unified generative models, letting you train visual reasoning tasks by swapping in your own data loader and reward function.

Often tagged together