Search papers, labs, and topics across Lattice.
PDEFlow is an autonomous framework that transforms user-defined ordinary and partial differential equations into neural-operator pipelines, integrating problem specification, data generation, operator training, and inference. By utilizing a stateful input graph, it efficiently converts multi-turn natural language inputs into validated specifications, generating datasets through a FEniCSx backend for subsequent training of a multi-branch Bayesian DeepONet. Experimental results demonstrate PDEFlow's capability to produce valid specifications, generate solver-backed datasets, and deliver solver-free predictions, significantly enhancing the automation of scientific and engineering workflows.
PDEFlow automates the entire process from user input to solver-free inference, enabling rapid experimentation with complex differential equations.
We present PDEFlow, an autonomous agentic framework that turns user-level ODE and PDE descriptions into solver-backed neural-operator pipelines. The workflow links problem specification, data generation, operator training, and checkpoint-based inference. A stateful input graph converts multi-turn natural-language input and user edits into validated problem specifications. The data-generation module then samples parameters, solves the configured governing-equation with FEniCSx finite-element backend, and stores the solutions as operator-ready tensors. The training and inference stages use a registry-based interface, allowing different neural operators to be trained and deployed without changing the surrounding pipeline. In the current implementation, we instantiate this interface with a multi-branch Bayesian DeepONet. Experiments on benchmark ODE and PDE tasks show that PDEFlow can construct valid specifications, generate solver-backed datasets, train neural operators across steady and transient problem classes, and provide solver-free predictions from saved checkpoints. The framework is designed for repeatable scientific and engineering workflows where many related physics configurations must be specified, simulated, learned, and queried with minimal manual intervention.
