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 last week
Related projects: ⓘ
- Pytorch bindings for Fortran☆83Updated 8 months ago
- ☆32Updated 3 years ago
- ☆22Updated this week
- Lecture and hands-on material for Track 8- Machine Learning of Argonne Training Program on Extreme-Scale Computing☆32Updated last month
- A library for directly calling PyTorch ML models from Fortran.☆64Updated this week
- PyTorch implemetation of multi-layer quasi-geostrophic model on rectangular domain with solid boundaries.☆9Updated 2 years ago
- Python package for NN generation from physics☆14Updated last year
- Datasets and code for results presented in the BOON paper☆41Updated last year
- explore the FV3 data for parameterization☆16Updated 2 months ago
- Try to approximate the solution of Fokker-Planck equation☆9Updated 11 months ago
- Ai2 Climate Emulator☆33Updated 2 weeks ago
- Call python from fortran☆82Updated last year
- Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems☆54Updated 9 months ago
- Experiments demonstrating coupled online learning for machine learning parameterizations☆18Updated last year
- An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py☆57Updated 3 years ago
- This repository containts materials for End-to-End AI for Science☆98Updated 3 weeks ago
- Turbulent flow network source code☆55Updated 9 months ago
- Code for the FiniteNet ICML Paper☆15Updated 4 years ago
- Quasi-geostrophic spectral solver in PyTorch☆15Updated 2 years ago
- Distributed Fourier Neural Operators☆28Updated 2 years ago
- Python code for "Applications of Deep Learning to Ocean Data Inference and Sub-Grid Parameterisation" (https://doi.org/10.1029/2018MS0014…☆47Updated 5 years ago
- One-dimensional DNS and LES of Burgers turbulence written in Python☆26Updated 6 months ago
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆81Updated 3 months ago
- ☆33Updated last year
- Code for Characterizing Scaling and Transfer Learning Behavior of FNO in SciML☆33Updated last year
- Simulation-Enabled Prediction, Inference, and Analysis: physics-informed statistical learning.☆29Updated 11 months ago
- Quasigeostrophic model in JAX (port of PyQG)☆15Updated 2 weeks ago
- Datasets and code for results presented in the ProbConserv paper☆51Updated 3 months ago
- Scientific Machine Learning Tutorials☆36Updated 2 years ago
- Uncertainty Quantification in the POD-NN framework☆19Updated 4 years ago