/UPT

Code for the paper Universal Physics Transformers

Primary LanguagePythonMIT LicenseMIT

Universal Physics Transformer (UPT)

[Project Page] [Paper (arxiv)] [Codebase Demo Video] [BibTeX]

Train your own models

Instructions to setup the codebase on your own environment are provided in SETUP_CODE, SETUP_DATA.

A video to motivate design choices of the codebase and give an overview of the codebase can be found here.

Configurations to train models can be found here.

Citation

If you like our work, please consider giving it a star ⭐ and cite us

@article{alkin2024upt,
      title={Universal Physics Transformers}, 
      author={Benedikt Alkin and Andreas Fürst and Simon Schmid and Lukas Gruber and Markus Holzleitner and Johannes Brandstetter},
      journal={arXiv preprint arXiv:2402.12365},
      year={2024}
}