Research & Papers

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

A new web tool lets engineers interactively simulate chip temperatures using a browser-based Physics-Informed Neural Network.

Deep Dive

Quantyze Labs has released an interactive web application that demonstrates a significant step toward making scientific AI a practical, real-time engineering tool. The application uses a custom Physics-Informed Neural Network (PINN) to solve the 2D heat equation, simulating the thermal behavior of two chips on a circuit board. By training the model in Python using the DeepXDE library with a PyTorch backend and then exporting it to the high-performance ONNX format, the team enabled the entire simulation to run client-side in a browser via a Blazor WebAssembly frontend, hosted on Azure.

This approach allows engineers and designers to interact directly with the AI model. Users can dynamically adjust key parameters like individual chip power output and ambient temperature, and the application generates an updated temperature heatmap and identifies hotspot temperatures in real-time. The project's core innovation is its shift from static research notebooks to an accessible, interactive tool, showcasing how PINNs can be deployed for immediate practical analysis. The developer notes ongoing work to improve boundary condition handling and accuracy for more complex board layouts, indicating a path toward simulating intricate real-world scenarios.

Key Points
  • Uses a Physics-Informed Neural Network (PINN) trained with DeepXDE/PyTorch to solve the 2D heat equation for thermal simulation.
  • The trained model is exported to ONNX and runs fully client-side in a Blazor WebAssembly app, requiring no server-side computation.
  • Provides an interactive interface for engineers to vary chip power and ambient temperature, generating real-time heatmaps and hotspot data.

Why It Matters

This demos a practical pipeline for deploying complex scientific AI models as accessible web tools, moving from academic research to immediate engineering utility.