An unofficial implementation of the Fourier Neural Operator in Flax
☆19May 25, 2024Updated last year
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
Sorting:
- ☆67Mar 21, 2025Updated 11 months ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- ☆15Nov 20, 2023Updated 2 years ago
- Code of ICML paper arxiv.org/abs/2302.08105☆14May 4, 2023Updated 2 years ago
- Official codebase for "Score-based Diffusion Models in Function Space"☆17Jan 23, 2025Updated last year
- Solving High Dimensional Partial Differential Equations with Deep Neural Networks☆34Dec 21, 2021Updated 4 years ago
- Code for "Robust flow field reconstruction from limited measurements vis sparse representation" (J. Callaham, K. Maeda, and S. Brunton 20…☆14Oct 23, 2018Updated 7 years ago
- Neural Emulator Architectures in JAX.☆23Feb 20, 2026Updated 2 weeks ago
- Gaussian Mixture Models in Jax☆24Jul 14, 2025Updated 7 months ago
- PDEBench: An Extensive Benchmark for Scientific Machine Learning☆1,078Feb 24, 2026Updated last week
- Implementation of physics informed neural networks with PyTorch☆20Jan 5, 2021Updated 5 years ago
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆109Dec 13, 2024Updated last year
- ☆25Dec 8, 2023Updated 2 years ago
- Datasets and code for results presented in the ProbConserv paper☆56Jun 14, 2024Updated last year
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆57May 23, 2023Updated 2 years ago
- [NeurIPS 2021] Galerkin Transformer: Neural Operator built on Attention for PDEs☆257Jun 14, 2024Updated last year
- 2D Burger's Equation (Convection + Diffusion)☆10May 13, 2021Updated 4 years ago
- [ICML 2023] Non-Uniform Neural Operator (NUNO)☆26Jul 3, 2023Updated 2 years ago
- Model hub for all your DiffeqML needs. Pretrained weights, modules, and basic inference infrastructure☆28Mar 9, 2023Updated 3 years ago
- The Senseiver: attention-based global field reconstruction from sparse observations☆30Nov 21, 2024Updated last year
- A short course in pseudospectral collocation methods for wave equations, with implementations in Python.☆64Sep 21, 2025Updated 5 months ago
- [ICLR 2023] Factorized Fourier Neural Operators☆189Oct 13, 2023Updated 2 years ago
- Python package 'dgpsi' for deep and linked Gaussian process emulations☆28Dec 2, 2025Updated 3 months ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month
- AGC 5차 대회 소스 코드 저장소입니다.☆10Jun 6, 2023Updated 2 years ago
- ☆29Jan 25, 2022Updated 4 years ago
- DM i AI 2022☆10Oct 4, 2022Updated 3 years ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆155Mar 2, 2026Updated last week
- Code for the paper Universal Physics Transformers☆151Jul 2, 2025Updated 8 months ago
- This repository contains code, which was used to generate large-scale results in the HINTS paper.☆36Oct 17, 2024Updated last year
- ☆41Jul 22, 2024Updated last year
- Applications of PINOs☆146Oct 10, 2022Updated 3 years ago
- ☆42Sep 2, 2025Updated 6 months ago
- This repository contains code for the paper "MAgNet: Mesh-Agnostic Neural PDE Solver" https://arxiv.org/abs/2210.05495☆38Jun 21, 2023Updated 2 years ago
- A comprehensive and fair comparison of two neural operators (with practical extensions) based on FAIR data☆361Jul 14, 2023Updated 2 years ago
- PDE-VAE: Variational Autoencoder for Extracting Interpretable Physical Parameters from Spatiotemporal Systems using Unsupervised Learning☆37Dec 18, 2021Updated 4 years ago
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆171Apr 10, 2025Updated 10 months ago
- Time2Vec neural network components. From paper: "Time2Vec: Learning a Vector Representation of Time" - https://arxiv.org/pdf/1907.05321.p…☆10Nov 25, 2024Updated last year
- ☆45Dec 7, 2022Updated 3 years ago