Adapting Aurora for Basin-Scale Scalar Prediction

[Jinal4502]

Really impressed by Aurora, it looks like a fantastic step forward for Earth system modeling. For my use case, I was wondering how should I adopt it for basin-scale scalar prediction tasks (eg. River Discharge)? Is it best to use the encoder as a frozen feature extractor on gridded atmospheric inputs and pool spatial tokens before a regression head, or is fine-tuning typically necessary? Also, should inputs match the full pretraining variable stack and temporal resolution?

Thank you so much for making your amazing work available to use!