Back to browse
GitHub Repository

A visual-based graph node editor for training computer vision models.

414 starsPython

MLForge – A visual graph editor for building PyTorch models

by zaina-ml·Apr 11, 2026·1 point·0 comments

AI Analysis

●●SolidCozyNiche Gem

Infers layer shapes from connections and exports standard PyTorch scripts.

Strengths
  • Auto-fills in_features and in_channels automatically based on upstream layer connections.
  • Exports standard PyTorch scripts instead of locking you into proprietary runtime.
  • Three-tab workflow separates data prep, architecture, and training logic cleanly.
Weaknesses
  • Visual graphs struggle with complex dynamic logic compared to standard code.
  • PyTorch must be preinstalled manually, adding friction to the quick start.
Category
Target Audience

Deep learning beginners, educators, visual prototypers

Similar To

Orange · KNIME · Google Vertex AI

Similar Projects