NVIDIA / modulus-makani
Massively parallel training of machine-learning based weather and climate models
☆244Updated last month
Alternatives and similar repositories for modulus-makani:
Users that are interested in modulus-makani are comparing it to the libraries listed below
- An open large-scale dataset for training high-resolution physics emulators in hybrid multi-scale climate simulators.☆147Updated 2 weeks ago
- NeuralGCM's differentiable dycore☆251Updated last week
- ☆62Updated last year
- PyTorch implementation of Ryan Keisler's 2022 "Forecasting Global Weather with Graph Neural Networks" paper (https://arxiv.org/abs/2202.0…☆218Updated last week
- Earth-2 Model Intercomparison Project (MIP) is a python framework that enables climate researchers and scientists to inter-compare AI mod…☆214Updated 2 months ago
- Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.☆135Updated 3 weeks ago
- Research Software for Neural Weather Prediction for Limited Area Modeling☆139Updated this week
- Implementation of MetNet-3, SOTA neural weather model out of Google Deepmind, in Pytorch☆209Updated last year
- A benchmark for the next generation of data-driven global weather models.☆495Updated last week
- Implementation of the Aurora model for Earth system forecasting☆301Updated last week
- Run AI NWP forecasts hassle-free, serverless in the cloud!☆112Updated 10 months ago
- Implementation of the Prithvi WxC Foundation Model and Downstream Tasks☆128Updated last month
- Initial public release of code, data, and model weights for FourCastNet☆559Updated last year
- Differentiable signal processing on the sphere for PyTorch☆438Updated last week
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆144Updated 2 months ago
- ☆51Updated 3 months ago
- A package to train machine learning models on geospatial data, mainly for weather and climate. Used to run ArchesWeather and ArchesWeathe…☆27Updated this week
- Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems☆58Updated last year
- Source code for ClimateLearn☆325Updated 11 months ago
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆44Updated 2 weeks ago
- ☆242Updated 5 months ago
- Official implementation of Score-based Data Assimilation☆47Updated last year
- Code for the paper Universal Physics Transformers☆90Updated last week
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆130Updated 2 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆82Updated this week
- Ai2 Climate Emulator☆82Updated last week
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆82Updated 2 months ago
- PyTorch Implementation of Google Research's MetNet and MetNet-2☆253Updated last week
- [NeurIPS 24] Probablistic Emulation of a Global Climate Model with Spherical DYffusion☆24Updated 2 months ago
- ☆73Updated 4 months ago