NVlabs / AFNO-transformer
Adaptive FNO transformer - official Pytorch implementation
☆232Updated 2 years ago
Alternatives and similar repositories for AFNO-transformer:
Users that are interested in AFNO-transformer are comparing it to the libraries listed below
- ☆25Updated 2 years ago
- [NeurIPS 2023] A Dynamics-informed Diffusion Model for Spatiotemporal Forecasting☆178Updated 3 months ago
- ☆54Updated 8 months ago
- [ICLR 2023] Factorized Fourier Neural Operators☆149Updated last year
- [NeurIPS 2021] Galerkin Transformer: a linear attention without softmax for Partial Differential Equations☆227Updated 7 months ago
- ☆66Updated last year
- About Code Release for "Solving High-Dimensional PDEs with Latent Spectral Models" (ICML 2023), https://arxiv.org/abs/2301.12664☆68Updated 9 months ago
- ClimODE: Climate and Weather Forecasting With Physics-informed Neural ODEs☆82Updated 5 months ago
- Geometry-Aware Fourier Neural Operator (Geo-FNO)☆202Updated last year
- Time- and space-continuous neural PDE forecaster based on INRs and ODEs☆55Updated last year
- Multiwavelets-based operator model☆57Updated 2 years ago
- ☆45Updated 5 months ago
- PyTorch implementation of the diffusion-based method for CFD data super-resolution proposed in the paper "A Physics-informed Diffusion Mo…☆149Updated last year
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆140Updated last month
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆80Updated last month
- About code release of "Transolver: A Fast Transformer Solver for PDEs on General Geometries", ICML 2024 Spotlight. https://arxiv.org/abs/…☆110Updated 3 months ago
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆119Updated last month
- ☆37Updated 4 months ago
- A small research project about advanced machine learning with neural networks. The Fourier neural operator was implemented here to see if…☆17Updated last year
- ☆63Updated 4 months ago
- ☆394Updated 10 months ago
- Wavelet Neural Operator for solving parametric partialdifferential equations in computational mechanics problems☆17Updated 5 months ago
- U-FNO - an enhanced Fourier neural operator-based deep-learning model for multiphase flow☆115Updated 4 months ago
- Initial public release of code, data, and model weights for FourCastNet☆553Updated last year
- This repository is the official implementation of the paper Convolutional Neural Operators for robust and accurate learning of PDEs☆146Updated 3 months ago
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆96Updated 2 months ago
- [NeurIPS 2023] Official implementation of "PreDiff: Precipitation Nowcasting with Latent Diffusion Models"☆110Updated 3 months ago
- ☆239Updated 4 months ago
- Spectral Neural Operator☆75Updated last year