Model reference · open weights
Prithvi-WxC-1.0 is an open-weight embedding model from ibm-nasa-geospatial. AxForge deploys and operates it for you on dedicated EU-owned hardware — with the licence handled where one is required.
Available as managed deployment — configured and operated for you on dedicated EU hardware, quoted per deployment.
What it is
| Released by | ibm-nasa-geospatial |
|---|---|
| Type | Embedding models |
| Task | Image embed |
| Runs with | terratorch |
| Released | 2024-09-20 |
| Popularity | 1k downloads / month |
| Licence | Commercial licence needed |
About
NASA and IBM have teamed up to create an AI Foundation Model for Weather and Climate - Prithvi WxC, using MERRA-2 data. By embracing the principles of open science, both organizations are actively contributing to the global mission of promoting knowledge sharing and accelerating innovations in addressing critical environmental challenges. With Hugging Face’s platform, they simplify model training and deployment, making it accessible for open science users, startups, and enterprises on multi-cloud AI platforms like watsonx. Additionally, Hugging Face enables easy sharing of the pipelines of the model family, which our team calls Prithvi WxC, within the community, fostering global collaboration and engagement. Prithvi WxC has been fine-tuned for several tasks including climate downscaling (released as part of the enterprise-grade IBM granite family of models) and gravity wave parameterization. More details on Prithvi WxC can be found in the joint IBM NASA technical paper.
Current downstreams tasks of PrithviWxC are (please feel free to submit a PR if you want to add yours): Gravity Wave
More information:Paper, Code, Model V1 PRs: prithvi-weather-climate-foundation-model (NASA), weather-climate-foundation-model (IBM) nasa-ibm-weather-climate-foundation-model (NASA)
Prithvi WxC is a 2.3 billion parameter model trained on 160 different variables from MERRA-2 data. It has been pretrained on both forecasting and masked reconstruction objectives. I.e.~the model is capable of reconstructing atmospheric state from partial information as well as propagating state into the future. The model takes data from two timestamps as input and generates a single, possibly future, timestamp as output. Currently Prithvi WxC comes in two flavors:
prithvi.wxc.2300m.v1 has been pretrained with a 50% masking ratio. The time delta between input timestamps is variable as is the forecast lead time.
During pretraining, the input delta was chosen from [-3, -6, -9, -12] hours while the forecast lead time was chosen from [0, 6, 12, 24] hours. We recommend using
prithvi.wxc.2300m.v1 for generic use cases that do not focus on forecasting.prithvi.wxc.rollout.2300m.v1 has been through further training cycles to be optimzed for autoregressive rollout. Here, we restricted the input delta
as well as the lead time to 6 hours. We recommend using prithvi.wxc.rollout.2300m.v1 for forecasting applications.If you use this work, consider citing our paper
@misc{schmude2024prithviwxcfoundationmodel,
title={Prithvi WxC: Foundation Model for Weather and Climate},
author={Johannes Schmude and Sujit Roy and Will Trojak and Johannes Jakubik and Daniel Salles Civitarese and Shraddha Singh and Julian Kuehnert and Kumar Ankur and Aman Gupta and Christopher E Phillips and Romeo Kienzler and Daniela Szwarcman and Vishal Gaur and Rajat Shinde and Rohit Lal and Arlindo Da Silva and Jorge Luis Guevara Diaz and Anne Jones and Simon Pfreundschuh and Amy Lin and Aditi Sheshadri and Udaysankar Nair and Valentine Anantharaj and Hendrik Hamann and Campbell Watson and Manil Maskey and Tsengdar J Lee and Juan Bernabe Moreno and Rahul Ramachandran},
year={2024},
eprint={2409.13598},
archivePrefix={arXiv},
primaryClass={cs.LG},
url={https://arxiv.org/abs/2409.13598},
}
From the published model card. Full card on the HuggingFace links in the sidebar.
How it works
Using it via the API
Once AxForge deploys prithvi-wxc-1-0 for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (prithvi-wxc-1-0 below is illustrative; you get the exact model name on deployment.)
$ curl -sS https://api.axforge.ai/v1/embeddings \
-H "Authorization: Bearer $AXFORGE_API_KEY" \
-H "Content-Type: application/json" \
-d '{"model":"prithvi-wxc-1-0","input":"text to embed"}'
Create an account — your API key is available in the console. 3M free tokens every 30 days with every new account.