About

TensorBloom is a graphical editor for PyTorch neural networks. It lets you see your model as a graph, train it, and export it.

Desktop app built with Tauri, React, and a Python sidecar. The frontend is a node-based graph editor (React Flow). The backend handles training, export, and shape inference over JSON-RPC.

Tech Stack

• Frontend — React 19, TypeScript, React Flow, Zustand, Tailwind CSS
• Desktop — Tauri v2 (Rust)
• Backend — Python sidecar with JSON-RPC
• ML — PyTorch 2.0+, TorchVision, ONNX
• Build — Vite, Cargo

Links

• GitHub Repository
• Roadmap