AI4SIM / model-collection
This project contains a collection of deep learning models developed by the AI4Sim team with various partners. This is structured on the basis of use-cases providing canonical PyTorch Lightning pipelines allowing to train neural network models that are able to surrogate various physical processes.
☆16Updated this week
Alternatives and similar repositories for model-collection:
Users that are interested in model-collection are comparing it to the libraries listed below
- ☆34Updated 3 years ago
- Pytorch bindings for Fortran☆91Updated last year
- Python package for NN generation from physics☆14Updated 2 years ago
- ☆35Updated last year
- Python package 'dgpsi' for deep and linked Gaussian process emulations☆24Updated 2 months ago
- PyTorch implemetation of multi-layer quasi-geostrophic model on rectangular domain with solid boundaries.☆10Updated 2 years ago
- ☆57Updated last year
- A library for directly calling PyTorch ML models from Fortran.☆107Updated this week
- Python code for "Applications of Deep Learning to Ocean Data Inference and Sub-Grid Parameterisation" (https://doi.org/10.1029/2018MS0014…☆48Updated 6 years ago
- Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems☆58Updated last year
- Experiments demonstrating coupled online learning for machine learning parameterizations☆21Updated 2 years ago
- Code for neural network parameterization project☆74Updated 4 years ago
- Quasi-geostrophic spectral solver in PyTorch☆16Updated 2 years ago
- This repository has moved, please visit https://github.com/ai2cm/pace for the latest development of fv3core.☆13Updated 2 years ago
- Python library for generating high-performance implementations of stencil kernels for weather and climate modeling from a domain-specific…☆117Updated this week
- GNN-Surrogate: A Hierarchical and Adaptive Graph Neural Network for Parameter Space Exploration of Unstructured-Mesh Ocean Simulations - …☆19Updated 9 months ago
- A spherical CNN for weather forecasting☆71Updated 2 years ago
- Official implementation of Score-based Data Assimilation☆54Updated last year
- Easy emulating of geophysical models including (but not limited to!) Earth System Models.☆56Updated 2 weeks ago
- Lorenz 1996 two time-scale model for learning machine learning☆50Updated last month
- A Python library for training neural ODEs.☆20Updated last month
- Libraries and utilities to develop performance portable applications for weather and climate.☆64Updated last month
- Public-facing git repository for JAMES submission, "Benchmarking of machine learning ocean parameterizations in an idealized model"☆11Updated 2 years ago
- Code for neural network parameterization project☆18Updated last year
- Leibniz is a python package which provide facilities to express learnable partial differential equations with PyTorch☆15Updated 3 years ago
- Graph-based operator learning in arbitrary geometries☆20Updated this week
- A two-way deep learning bridge between Keras and Fortran☆179Updated last year
- explore the FV3 data for parameterization☆17Updated 2 months ago
- Lorenz 96 model with GAN parameterization of unresolved scale (Y).☆36Updated 5 years ago
- PyTorch training at CSCS☆15Updated last year