zaina-ml/ml_forge
A drag-and-drop node editor that generates and runs PyTorch training code for image classification models.
Not currently ranked — collecting fresh signals.
star history
ML Forge is a visual desktop application that provides a three-tab workflow (Data Prep, Model, Training) for building computer vision pipelines. Users drag nodes onto a canvas, connect them with wires, and the tool automatically generates PyTorch training code. It supports popular datasets (MNIST, CIFAR10, CIFAR100, FashionMNIST, ImageFolder) and provides live training visualization with loss curves, checkpoint saving, and model inference. The tool exports projects into clean, runnable PyTorch code.
Frequently asked
- What is zaina-ml/ml_forge?
- A drag-and-drop node editor that generates and runs PyTorch training code for image classification models.
- Is ml_forge open source?
- Yes — zaina-ml/ml_forge is open source, released under the MIT license.
- What language is ml_forge written in?
- zaina-ml/ml_forge is primarily written in Python.
- How popular is ml_forge?
- zaina-ml/ml_forge has 416 stars on GitHub.
- Where can I find ml_forge?
- zaina-ml/ml_forge is on GitHub at https://github.com/zaina-ml/ml_forge.