search

NVIDIA / earth2studio

Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.
https://nvidia.github.io/earth2studio/
Apache License 2.0
84 stars 29 forks source link

🚀[FEA]: Will Earth2Studio support training of AI weather models？ #145

Open LevineHuang opened 1 week ago

LevineHuang commented 1 week ago

Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request

Critical (currently preventing usage)

Please provide a clear description of problem you would like to solve.

Will Earth2Studio support training of AI weather models？

NickGeneva commented 1 week ago

Hi @LevineHuang

At the moment Earth2Studio is a inference focused package. Training of weather AI models is something we leave to other repos, whether modulus, makani or the research repo's that are tied to a particular model. I do know some of the utils inside Earth2Studio can be potentially useful for training (such as fetching data).

So some components could be improved based on peoples needs. Did you have something particular in mind?