Desktop appSoftware development
ml_forge
Build, train, run inference, and export image-classification models with a visual node editor.
- Stars
- 417
- Forks
- 50
- Updated
- Updated Jul 11, 2026
MIT
Build, train, run inference, and export image-classification models with a visual node editor.
ML Forge is a visual PyTorch pipeline editor for building image-classification models without writing the training workflow by hand. Users drag data-preparation, model, and training nodes onto a canvas, connect them, and let the application generate and run the code. It shows live loss curves, saves checkpoints, supports inference with the trained model, and exports projects as readable PyTorch. Python 3.10 or newer is required, and PyTorch plus torchvision must be installed separately before training.
Resource types
Use cases
Runtime
Audience
Edit, navigate, debug, and extend code with the open-source core behind VS Code.
Switch providers and sync MCP, prompts, skills, and sessions across AI coding tools.
Public GitHub facts last synced Jul 14, 2026.
Use built-in or third-party coding agents in a terminal-based development environment
Install, edit, execute, and test code through a desktop app, CLI, or API.