astanziola / fourier-neural-operator-flax
An unofficial implementation of the Fourier Neural Operator in Flax
☆17Updated 11 months ago
Alternatives and similar repositories for fourier-neural-operator-flax:
Users that are interested in fourier-neural-operator-flax are comparing it to the libraries listed below
- Code of ICML paper arxiv.org/abs/2302.08105☆14Updated last year
- ☆41Updated 2 years ago
- An extension of Fourier Neural Operator to finite-dimensional input and/or output spaces.☆19Updated 10 months ago
- Spectral Neural Operator☆78Updated last year
- Here you find the Spectral Operator Learning Under construcTION☆19Updated 11 months ago
- Official implementation of Scalable Transformer for PDE surrogate modelling☆41Updated last year
- Pseudospectral Kolmogorov Flow Solver☆38Updated last year
- Efficient Differentiable n-d PDE solvers in JAX.☆28Updated 5 months ago
- ☆16Updated 8 months ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆68Updated 5 months ago
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆40Updated last year
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- ☆32Updated 10 months ago
- Derivative-Informed Neural Operator: An Efficient Framework for High-Dimensional Parametric Derivative Learning☆17Updated last year
- Practicum on Supervised Learning in Function Spaces☆33Updated 3 years ago
- ☆26Updated 9 months ago
- ☆31Updated last year
- Distributed Fourier Neural Operators☆29Updated 2 years ago
- [ICLR 2024] Scaling physics-informed hard constraints with mixture-of-experts.☆32Updated 10 months ago
- Generative Learning for Forecasting the Dynamics of High Dimensional Complex Systems☆28Updated last month
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆53Updated last month
- Datasets and code for results presented in the ProbConserv paper☆53Updated 10 months ago
- ☆14Updated last year
- ☆29Updated last month
- The public repository about our joint FINN research project☆36Updated 2 years ago
- ☆13Updated last year
- ☆73Updated 7 months ago
- Differential equation neural operator☆20Updated last year
- Code repository for "Learned Turbulence Modelling with Differentiable Fluid Solvers"☆36Updated 2 years ago
- PROSE: Predicting Multiple Operators and Symbolic Expressions☆23Updated 3 weeks ago