[P] Built a Interactive Web for PINN Solving the 2D Heat Equation

Reddit r/MachineLearning / 3/25/2026

💬 OpinionDeveloper Stack & InfrastructureTools & Practical UsageModels & Research

Key Points

  • The author built a real-time interactive web demo that uses Physics-Informed Neural Networks (PINNs) to solve the 2D heat equation for a thermal simulation of two chips on a circuit board.
  • The trained PINN is exported to ONNX and integrated into a browser-based app so users can adjust parameters (e.g., chip power and ambient temperature) and immediately view resulting temperature heatmaps and hotspot values.
  • The system is implemented with DeepXDE (PyTorch backend) for training and Blazor WebAssembly hosted on Azure for deployment, with simulation running entirely client-side for responsiveness.
  • The project is currently focused on improving boundary condition flexibility and accuracy to support more complex board layouts.
  • Feedback is solicited from the community on the approach’s practicality and where it could be most valuable for thermal inference workflows.

Hey everyone,

I’ve been working on the idea of taking Scientific AI out of research notebooks and making it accessible as a useful real-time tool. I just finished the first interactive demo, and I’d love some feedback.

I built and trained a 2D thermal simulation engine of two chips on a circuit board using Physics-Informed Neural Networks (PINNs), to solve the 2D heat equation.

Exporting the trained model as ONNX, I build up a simple interactive web app in the browser which allows users to interact with the PINN model by varying the parameters like chip power and ambient temperature to obtain the temperature heatmap and hotspot temperatures.

The Tech Stack:

  • AI: Trained a custom PINN in Python using DeepXDE with PyTorch backend
  • Deployment: Exported to ONNX for high-performance cross-platform execution.
  • Web: Built with Blazor WebAssembly and hosted on Azure. The simulation runs entirely client-side.

Live Demo: https://www.quantyzelabs.com/thermal-inference

I'm currently working on improving the boundary condition flexibility and accuracy for more complex board layouts. I’d love to hear your feedback and where you think this approach has the most potential.

Cheers!

submitted by /u/wyzard135
[link] [comments]