README.md

---
license: mit
---

# PROSE-FD: A Multimodal PDE Foundation Model for Learning Multiple Operators for Forecasting Fluid Dynamics

This folder contains model weights for the PROSE-FD model in paper [PROSE-FD: A Multimodal PDE Foundation Model for Learning Multiple Operators for Forecasting Fluid Dynamics](https://arxiv.org/abs/2409.09811). The code is included in: https://github.com/felix-lyx/prose.

![model](model.png)

Note: The keys for the weights contain ```module._orig_mod.``` prefixes (coming from DDP and torch.compile). This is automatically taken care of in our code, but worth noting. 

## Citation

```
@article{liu2024prose_fd,
  title={{PROSE-FD}: A Multimodal PDE Foundation Model for Learning Multiple Operators for Forecasting Fluid Dynamics},
  author={Liu, Yuxuan and Sun, Jingmin and He, Xinjie and Pinney, Griffin and Zhang, Zecheng and Schaeffer, Hayden},
  journal={arXiv preprint arXiv:2409.09811},
  year={2024}
}
```