NVIDIA / modulus-makani
Massively parallel training of machine-learning based weather and climate models
☆245Updated 2 months ago
Alternatives and similar repositories for modulus-makani:
Users that are interested in modulus-makani are comparing it to the libraries listed below
- ☆62Updated last year
- NeuralGCM's differentiable dycore☆256Updated last week
- An open large-scale dataset for training high-resolution physics emulators in hybrid multi-scale climate simulators.☆148Updated last month
- Research Software for Neural Weather Prediction for Limited Area Modeling☆145Updated 3 weeks ago
- Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.☆142Updated last week
- Earth-2 Model Intercomparison Project (MIP) is a python framework that enables climate researchers and scientists to inter-compare AI mod…☆218Updated 3 months ago
- Implementation of MetNet-3, SOTA neural weather model out of Google Deepmind, in Pytorch☆211Updated last year
- Run AI NWP forecasts hassle-free, serverless in the cloud!☆112Updated 11 months ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆150Updated 3 months ago
- PyTorch implementation of Ryan Keisler's 2022 "Forecasting Global Weather with Graph Neural Networks" paper (https://arxiv.org/abs/2202.0…☆226Updated this week
- Implementation of the Aurora model for Earth system forecasting☆313Updated this week
- Implementation of the Prithvi WxC Foundation Model and Downstream Tasks☆132Updated 2 weeks ago
- [NeurIPS 24] Probablistic Emulation of a Global Climate Model with Spherical DYffusion☆26Updated this week
- ☆54Updated 4 months ago
- Initial public release of code, data, and model weights for FourCastNet☆566Updated last year
- Source code for ClimateLearn☆328Updated 11 months ago
- A benchmark for the next generation of data-driven global weather models.☆506Updated this week
- ☆452Updated 2 weeks ago
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆47Updated last month
- Ai2 Climate Emulator☆87Updated last month
- Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems☆58Updated last year
- Official implementation of Score-based Data Assimilation☆51Updated last year
- An ai-models plugin to run Microsoft's Aurora model☆12Updated 5 months ago
- Code for the paper Universal Physics Transformers☆97Updated last month
- A large compression model for weather and climate data, which compresses a 400+ TB ERA5 dataset into a new 0.8 TB CRA5 dataset.☆50Updated 4 months ago
- PyTorch Implementation of Google Research's MetNet and MetNet-2☆258Updated this week
- ☆53Updated this week
- A package to train machine learning models on geospatial data, mainly for weather and climate. Used to run ArchesWeather and ArchesWeathe…☆30Updated this week
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆135Updated 3 months ago