This is the enhanced version of the original source code for the paper Poseidon: Efficient Foundation Models for PDEs. This repository also acts as a package if you want to use the enhanced models in your code.
Use the pretrained models and pretraining dataset in our assortment on the 🤗 Hub – Pretrained Models and Pretraining Datasets. All datasets corresponding to downstream tasks can be downloaded from the same collection on the 🤗 Hub – Downstream Tasks as well. To use them, follow the corresponding sections below.
[...CONTENT OMITTED FOR BREVITY...]
If you use our models, code, or datasets, please consider citing our paper, details below:
@misc{herde2024poseidon,
title={Poseidon: Efficient Foundation Models for PDEs},
author={Maximilian Herde and Bogdan Raonić and Tobias Rohner and Roger Käppeli and Roberto Molinaro and Emmanuel de Bézenac and Siddhartha Mishra},
year={2024},
eprint={2405.19101},
archivePrefix={arXiv},
primaryClass={cs.LG}
}