Reddit - r/MachineLearning

I built a open source neural network shape validator [P]

Overview

Built a visual editor that validates tensor shapes, counts parameters, estimates FLOPs/VRAM while you design. Catches incompatible residuals, mismatched Linear layers, all that before you waste GPU time.

Features

  • 63 operations with proper shape inference
  • Real-time validation of tensor shapes
  • Parameter counting
  • FLOPs and VRAM estimation
  • Catches incompatible residuals and mismatched Linear layers
  • Exports PyTorch code that actually runs

Links

Comments

No comments yet. Start the discussion.