Implementation of the paper: "Aurora: A Foundation Model of the Atmosphere" in PyTorch: Paper link
pip3 install aurora-torch
MIT
@misc{bodnar2024aurora,
title={Aurora: A Foundation Model of the Atmosphere},
author={Cristian Bodnar and Wessel P. Bruinsma and Ana Lucic and Megan Stanley and Johannes Brandstetter and Patrick Garvan and Maik Riechert and Jonathan Weyn and Haiyu Dong and Anna Vaughan and Jayesh K. Gupta and Kit Tambiratnam and Alex Archibald and Elizabeth Heider and Max Welling and Richard E. Turner and Paris Perdikaris},
year={2024},
eprint={2405.13063},
archivePrefix={arXiv},
primaryClass={physics.ao-ph}
}