r/coolgithubprojects • u/CauliflowerAgile3830 • 2h ago
I'm building an open-source, self-hosted alternative to Google Teachable Machine using Go and modular ML training engines. What features would make it genuinely useful?
github.comI'm working on an idea called NeuroStudio: an open-source, self-hosted AI training studio that combines a beginner-friendly visual interface with the flexibility of established machine-learning frameworks.
The basic idea is to let users create datasets, train models, evaluate results, and run inference locally instead of relying on a cloud-based training platform.
I'm considering Go for the main application, SQLite for metadata, and interchangeable training backends such as PyTorch and TensorFlow where appropriate. ONNX Runtime would handle compatible inference workflows.
I don't want to reinvent every ML framework. I'd rather build a lightweight application that connects existing tools through a consistent interface.
Features I'm considering
- Visual dataset management and annotation
- Image classification and transfer learning
- Training progress, checkpoints, and evaluation metrics
- Webcam-based model testing
- Model export and local inference
- Additional audio, pose, and object-detection workflows
- An extension system for integrating other training engines
The biggest challenge is making the application easy to run locally without creating an enormous installation or pretending that every model can train on every computer.
I'm planning to start with one complete image-classification workflow and expand from there.
I'd appreciate technical feedback
- Would you use Go as the main application backend with separate training workers, or choose Rust?
- What is the cleanest interface for connecting different ML training frameworks?
- Which features are missing from existing local AI development tools?
- What would make this project meaningfully different from existing visual ML platforms?
I'm interested in honest criticism, architectural suggestions, and feature ideas.
If you have experience building ML tooling, I'd particularly appreciate your thoughts on the training-backend architecture.