{"id":300840,"name":"Prithvi WxC","description":"Foundation model for weather and climate, that employs an encoder-decoder-based architecture, incorporating concepts from various recent transformer models to effectively capture both regional and global dependencies in the input data.","url":"https://github.com/nasa-impact/prithvi-wxc","last_synced_at":"2026-04-30T17:00:32.202Z","repository":{"id":257801977,"uuid":"849978617","full_name":"NASA-IMPACT/Prithvi-WxC","owner":"NASA-IMPACT","description":"Implementation of the Prithvi WxC Foundation Model and Downstream Tasks","archived":false,"fork":false,"pushed_at":"2026-02-05T13:14:14.000Z","size":63595,"stargazers_count":184,"open_issues_count":11,"forks_count":44,"subscribers_count":6,"default_branch":"main","last_synced_at":"2026-04-26T15:02:18.678Z","etag":null,"topics":[],"latest_commit_sha":null,"homepage":null,"language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/NASA-IMPACT.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null,"zenodo":null,"notice":null,"maintainers":null,"copyright":null,"agents":null,"dco":null,"cla":null}},"created_at":"2024-08-30T16:23:06.000Z","updated_at":"2026-04-24T06:27:41.000Z","dependencies_parsed_at":"2025-03-31T17:02:20.690Z","dependency_job_id":null,"html_url":"https://github.com/NASA-IMPACT/Prithvi-WxC","commit_stats":{"total_commits":49,"total_committers":9,"mean_commits":5.444444444444445,"dds":0.5714285714285714,"last_synced_commit":"ecfd69b2e94b6505d3accbdecf095b92fd18411e"},"previous_names":["nasa-impact/prithvi-wxc"],"tags_count":3,"template":false,"template_full_name":null,"purl":"pkg:github/NASA-IMPACT/Prithvi-WxC","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/NASA-IMPACT%2FPrithvi-WxC","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/NASA-IMPACT%2FPrithvi-WxC/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/NASA-IMPACT%2FPrithvi-WxC/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/NASA-IMPACT%2FPrithvi-WxC/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/NASA-IMPACT","download_url":"https://codeload.github.com/NASA-IMPACT/Prithvi-WxC/tar.gz/refs/heads/main","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/NASA-IMPACT%2FPrithvi-WxC/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":286080680,"owners_count":32470879,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2026-04-30T13:12:12.517Z","status":"ssl_error","status_checked_at":"2026-04-30T13:12:06.837Z","response_time":57,"last_error":"SSL_read: unexpected eof while reading","robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":false,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"owner":{"login":"NASA-IMPACT","name":"Inter Agency Implementation and Advanced Concepts","uuid":"22798984","kind":"organization","description":"","email":"esds.dsig@gmail.com","website":null,"location":null,"twitter":null,"company":null,"icon_url":"https://avatars.githubusercontent.com/u/22798984?v=4","repositories_count":88,"last_synced_at":"2023-03-04T07:55:15.502Z","metadata":{"has_sponsors_listing":false},"html_url":"https://github.com/NASA-IMPACT","funding_links":[],"total_stars":null,"followers":null,"following":null,"created_at":"2022-11-14T15:35:50.518Z","updated_at":"2023-03-04T07:55:15.538Z","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/NASA-IMPACT","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/NASA-IMPACT/repositories"},"packages":[],"commits":{"id":1866852,"full_name":"nasa-impact/prithvi-wxc","default_branch":"main","total_commits":96,"total_committers":10,"total_bot_commits":0,"total_bot_committers":0,"mean_commits":9.6,"dds":0.5729166666666667,"past_year_total_commits":9,"past_year_total_committers":4,"past_year_total_bot_commits":0,"past_year_total_bot_committers":0,"past_year_mean_commits":2.25,"past_year_dds":0.33333333333333337,"last_synced_at":"2026-04-28T13:21:40.135Z","last_synced_commit":"79dabfcd17abe77e2d5c696707c0164a04f2ec01","created_at":"2024-09-27T00:14:06.168Z","updated_at":"2026-04-28T13:21:14.938Z","committers":[{"name":"Johannes Schmude","email":"Johannes.Schmude@ibm.com","login":"johannesschmude","count":41},{"name":"Will Trojak","email":"w.trojak@ibm.com","login":"WillTrojak","count":22},{"name":"Romeo Kienzler","email":"romeo.kienzler1@ibm.com","login":"romeokienzler","count":11},{"name":"Lelouch vi' Britania","email":"sujitroy1994@hotmail.com","login":"thesujitroy","count":7},{"name":"Simon Pfreundschuh","email":"s.pfreundschuh@gmail.com","login":"simonpf","count":6},{"name":"Daniel Salles Civitarese","email":"1186827+dancivitarese","login":"dancivitarese","count":5},{"name":"lino","email":"40688435+michelebanfi","login":"michelebanfi","count":1},{"name":"Julian Kuehnert","email":"Jubeku","login":"Jubeku","count":1},{"name":"Johannes Jakubik","email":"127425365+jhnnsjkbk1","login":"jhnnsjkbk1","count":1},{"name":"Eloisa Bentivegna","email":"eloisa.bentivegna@ibm.com","login":null,"count":1}],"past_year_committers":[{"name":"Simon Pfreundschuh","email":"s.pfreundschuh@gmail.com","login":"simonpf","count":6},{"name":"Romeo Kienzler","email":"5694071+romeokienzler","login":"romeokienzler","count":1},{"name":"Johannes Jakubik","email":"127425365+jhnnsjkbk1","login":"jhnnsjkbk1","count":1},{"name":"Eloisa Bentivegna","email":"eloisa.bentivegna@ibm.com","login":null,"count":1}],"commits_url":"https://commits.ecosyste.ms/api/v1/hosts/GitHub/repositories/nasa-impact%2Fprithvi-wxc/commits","host":{"name":"GitHub","url":"https://github.com","kind":"github","last_synced_at":"2026-04-28T00:00:07.369Z","repositories_count":6221076,"commits_count":900092367,"contributors_count":34897309,"owners_count":1146655,"icon_url":"https://github.com/github.png","host_url":"https://commits.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://commits.ecosyste.ms/api/v1/hosts/GitHub/repositories"}},"issues_stats":{"full_name":"nasa-impact/prithvi-wxc","html_url":"https://github.com/nasa-impact/prithvi-wxc","last_synced_at":"2026-04-26T15:00:31.275Z","status":"error","issues_count":15,"pull_requests_count":51,"avg_time_to_close_issue":791136.0,"avg_time_to_close_pull_request":56921.90476190476,"issues_closed_count":7,"pull_requests_closed_count":42,"pull_request_authors_count":10,"issue_authors_count":10,"avg_comments_per_issue":0.4,"avg_comments_per_pull_request":0.27450980392156865,"merged_pull_requests_count":39,"bot_issues_count":0,"bot_pull_requests_count":0,"past_year_issues_count":8,"past_year_pull_requests_count":12,"past_year_avg_time_to_close_issue":150386.0,"past_year_avg_time_to_close_pull_request":32046.777777777777,"past_year_issues_closed_count":2,"past_year_pull_requests_closed_count":9,"past_year_pull_request_authors_count":6,"past_year_issue_authors_count":6,"past_year_avg_comments_per_issue":0.125,"past_year_avg_comments_per_pull_request":0.08333333333333333,"past_year_bot_issues_count":0,"past_year_bot_pull_requests_count":0,"past_year_merged_pull_requests_count":9,"created_at":"2024-09-27T00:14:06.431Z","updated_at":"2026-04-26T15:00:31.275Z","repository_url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/repositories/nasa-impact%2Fprithvi-wxc","issues_url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/repositories/nasa-impact%2Fprithvi-wxc/issues","issue_labels_count":{"bug":1},"pull_request_labels_count":{},"issue_author_associations_count":{"NONE":10,"COLLABORATOR":5},"pull_request_author_associations_count":{"COLLABORATOR":44,"NONE":5,"CONTRIBUTOR":2},"issue_authors":{"WillTrojak":5,"ShileiCao":2,"rubencart":1,"manmeet3591":1,"taylorfturner":1,"gajeshladhar":1,"sdash77":1,"dun280":1,"whpy":1,"seangtkelley":1},"pull_request_authors":{"WillTrojak":23,"romeokienzler":7,"dancivitarese":6,"johannesschmude":5,"jhnnsjkbk1":2,"michelebanfi":2,"simonpf":2,"take2rohit":2,"Jubeku":1,"eloisabentivegna":1},"host":{"name":"GitHub","url":"https://github.com","kind":"github","last_synced_at":"2026-04-30T00:00:06.639Z","repositories_count":14451099,"issues_count":34192146,"pull_requests_count":112144725,"authors_count":11249630,"icon_url":"https://github.com/github.png","host_url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/repositories","owners_url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/owners","authors_url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/authors"},"past_year_issue_labels_count":{},"past_year_pull_request_labels_count":{},"past_year_issue_author_associations_count":{"NONE":2},"past_year_pull_request_author_associations_count":{"COLLABORATOR":3,"NONE":2,"CONTRIBUTOR":1},"past_year_issue_authors":{"dun280":1,"manmeet3591":1},"past_year_pull_request_authors":{"jhnnsjkbk1":2,"simonpf":2,"eloisabentivegna":1,"romeokienzler":1},"maintainers":[{"login":"WillTrojak","count":28,"url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/authors/WillTrojak"},{"login":"romeokienzler","count":7,"url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/authors/romeokienzler"},{"login":"dancivitarese","count":6,"url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/authors/dancivitarese"},{"login":"johannesschmude","count":5,"url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/authors/johannesschmude"},{"login":"jhnnsjkbk1","count":2,"url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/authors/jhnnsjkbk1"},{"login":"Jubeku","count":1,"url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/authors/Jubeku"}],"active_maintainers":[{"login":"jhnnsjkbk1","count":2,"url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/authors/jhnnsjkbk1"},{"login":"romeokienzler","count":1,"url":"https://issues.ecosyste.ms/api/v1/hosts/GitHub/authors/romeokienzler"}]},"events":{"total":{"ReleaseEvent":3,"DeleteEvent":5,"MemberEvent":1,"PullRequestEvent":23,"ForkEvent":20,"IssuesEvent":16,"WatchEvent":63,"IssueCommentEvent":10,"PushEvent":16,"PullRequestReviewCommentEvent":2,"PullRequestReviewEvent":6,"CreateEvent":8},"last_year":{"DeleteEvent":1,"MemberEvent":1,"PullRequestEvent":5,"ForkEvent":6,"IssuesEvent":1,"WatchEvent":20,"PushEvent":3,"PullRequestReviewEvent":1,"CreateEvent":1}},"keywords":[],"dependencies":[{"ecosystem":"pypi","filepath":"docs/requirements.txt","sha":null,"kind":"manifest","created_at":"2024-09-28T00:30:19.376Z","updated_at":"2024-09-28T00:30:19.376Z","repository_link":"https://github.com/NASA-IMPACT/Prithvi-WxC/blob/main/docs/requirements.txt","dependencies":[{"id":20208284056,"package_name":"Sphinx","ecosystem":"pypi","requirements":"\u003e=8.0.2","direct":true,"kind":"runtime","optional":false},{"id":20208284057,"package_name":"graphviz","ecosystem":"pypi","requirements":"*","direct":true,"kind":"runtime","optional":false},{"id":20208284058,"package_name":"mkdocs","ecosystem":"pypi","requirements":"*","direct":true,"kind":"runtime","optional":false},{"id":20208284059,"package_name":"sphinx-rtd-theme","ecosystem":"pypi","requirements":"*","direct":true,"kind":"runtime","optional":false},{"id":20208284060,"package_name":"sphinxcontrib-napoleon","ecosystem":"pypi","requirements":"*","direct":true,"kind":"runtime","optional":false},{"id":20208284061,"package_name":"pytools","ecosystem":"pypi","requirements":"\u003e=2016.2.1","direct":true,"kind":"runtime","optional":false}]},{"ecosystem":"pypi","filepath":"pyproject.toml","sha":null,"kind":"manifest","created_at":"2024-09-28T00:30:21.279Z","updated_at":"2024-09-28T00:30:21.279Z","repository_link":"https://github.com/NASA-IMPACT/Prithvi-WxC/blob/main/pyproject.toml","dependencies":[{"id":20208284062,"package_name":"h5py","ecosystem":"pypi","requirements":"~= 3.3","direct":true,"kind":"runtime","optional":false},{"id":20208284063,"package_name":"numpy","ecosystem":"pypi","requirements":"~= 1.26","direct":true,"kind":"runtime","optional":false},{"id":20208284064,"package_name":"pandas","ecosystem":"pypi","requirements":"~= 2.2","direct":true,"kind":"runtime","optional":false},{"id":20208284065,"package_name":"torch","ecosystem":"pypi","requirements":"\u003e= 2.2","direct":true,"kind":"runtime","optional":false}]}],"score":7.575584651557793,"created_at":"2024-09-27T00:14:03.888Z","updated_at":"2026-04-30T17:00:32.211Z","avatar_url":"https://github.com/NASA-IMPACT.png","language":"Python","category":"Atmosphere","sub_category":"Meteorological Observation and Forecast","monthly_downloads":0,"total_dependent_repos":0,"total_dependent_packages":0,"readme":"# Prithvi WxC: Foundation model for weather and climate\n\nThis repository contains the code of the Prithvi WxC foundation model as well as basic zero-shot examples for testing and illustration. For fine-tuning applications please refer to task-specific repositories listed [below](https://github.com/NASA-IMPACT/Prithvi-WxC?tab=readme-ov-file#fine-tuning-applications).\n\n## Updates\n\n### March 25, 2025\n\nThe previous version of this repo contained a number of bugs that led to incorrect model outputs and worse performance than in our paper. We just addressed these issues. In particular, there is validation code below that lets you verify whether your particular platform and version of the code obtains results comparable to ours. (See step 3 under [Getting started](#getting-started).)\n\n## Architecture overview: A scalable and flexible vision transformer\n\nPrithvi WxC is at its core a scalable 2D vision transformer. The architecture is designed to allow for memory-efficient masked pretraining. It draws inspiration from both Hiera, MaxViT and SWIN transformers. Inputs, structured into windows, take the shape (batch, windows, tokens, features). We alternate between **local attention** (within a window) and **global attention** (across windows). This is implemented by transposing dimensions between transformer layers. Attention acts on the third dimension, the second being part of the batch. When data becomes dense -- i.e. in the absence of masking -- it is possible to add SWIN-like shifts to the local attention layers. See the figure for illustration:\n\n![arch_main](docs/arch_main.png)\n\n## Fine-tuning applications\n\nWe have fine-tuned the model to a number of downstream tasks. See the paper as well as the respective repository for details.\n\n| Application                  | Dataset     | Repository                         |\n| ---------------------------- | ----------- | ---------------------------------- |\n| Downscaling                  | MERRA-2     | https://github.com/IBM/granite-wxc |\n| Downscaling                  | EURO-CORDEX | https://github.com/IBM/granite-wxc |\n| Gravity wave parametrization | ERA5        | https://github.com/NASA-IMPACT/gravity-wave-finetuning |\n\nBeyond these there are zero-shot applications in masked reconstruction and forecasting.\n\n## Getting started\n\n1. Create a virtual environment\n2. Clone this repository and install Prithvi WxC as a module\n   ```\n   git clone https://github.com/NASA-IMPACT/Prithvi-WxC\n   cd Prithvi-WxC\n   pip install '.[examples]'\n   ```\n3. Validate that the model behaves as expected. For that run\n   ```\n   python -m validation.validate_prithvi_wxc -c validation/config.yaml\n   ```\n4. Run one of the notebooks in the `examples` directory:\n   - [Basic inference](examples/PrithviWxC_inference.ipynb)\n   - [Rollout inference](examples/PrithviWxC_rollout.ipynb)\n   \n   These notebooks will download model weights as well as sample data for basic illustration from [Hugging Face](https://huggingface.co/Prithvi-WxC).\n\n## Pretrained models\n\nPrithvi WxC is a very flexible model. It has been pretrained on a pretext task blending masked reconstruction and forecasting so that it can be used for both zero-hours ahead as well as forecasting applications. Moreover, the masking pattern makes it suitable for both global and regional applications. There are currently two pretrained base models as well as several [fine-tuning applications](https://github.com/NASA-IMPACT/Prithvi-WxC?tab=readme-ov-file#fine-tuning-applications).\n\n| Model                        | Details                                                                                                   | Weights                                                 |\n| ---------------------------- | --------------------------------------------------------------------------------------------------------- | ------------------------------------------------------- |\n| prithvi.wxc.2300m.v1         | Pretrained 2.3B parameter model. Flexible input and lead time. For general and 0-hour ahead applications. | https://huggingface.co/Prithvi-WxC/prithvi.wxc.2300m.v1 |\n| prithvi.wxc.rollout.2300m.v1 | Pretrained 2.3B parameter model. Input and lead time fixed to 6h. For forecasting applications.           | https://huggingface.co/Prithvi-WxC/prithvi.wxc.rollout.2300m.v1 |\n\n\n## Data\n\nPrithvi WxC used data from the MERRA-2 reanalysis for pretraining. In particular, the model uses a climatology computed from MERRA-2 data. The climatology, too, is [available via Hugging Face](https://huggingface.co/Prithvi-WxC/prithvi.wxc.2300m.v1/tree/main/climatology). See the paper for details on variables choosen and the methodology behind the climatology.\n\n\n## Citation\nIf you use this work, consider citing our paper\n\n```\n@misc{schmude2024prithviwxcfoundationmodel,\n      title={Prithvi WxC: Foundation Model for Weather and Climate}, \n      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},\n      year={2024},\n      eprint={2409.13598},\n      archivePrefix={arXiv},\n      primaryClass={cs.LG},\n      url={https://arxiv.org/abs/2409.13598}, \n}\n```\n","funding_links":[],"readme_doi_urls":[],"works":{},"citation_counts":{},"total_citations":0,"keywords_from_contributors":[],"project_url":"https://ost.ecosyste.ms/api/v1/projects/300840","html_url":"https://ost.ecosyste.ms/projects/300840"}