We propose PROSE-PDE algorithm, a extension of PROSE, training the map from multimodal inputs to multimodal outputs.

We provide several extrapolation studies to highlight the ability of PROSE-PDE in generalizing physical features through the robust training to unseen operators.
Paper Github