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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆67Mar 21, 2025Updated last year
- Training methodologies for autoregressive neural operators/emulators in JAX.☆11Nov 5, 2024Updated last year
- Code of ICML paper arxiv.org/abs/2302.08105☆14May 4, 2023Updated 2 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Official codebase for "Score-based Diffusion Models in Function Space"☆18Jan 23, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- ☆15Nov 20, 2023Updated 2 years ago
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆25Jan 9, 2026Updated 2 months ago
- ☆10Jun 13, 2022Updated 3 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
- PDEBench: An Extensive Benchmark for Scientific Machine Learning☆1,091Feb 24, 2026Updated last month
- ☆25Dec 8, 2023Updated 2 years ago
- Gaussian Mixture Models in Jax☆25Jul 14, 2025Updated 8 months ago
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆111Dec 13, 2024Updated last year
- PF-PINNs: physics-informed neural networks framework for solving coupled Allen-Cahn and Cahn-Hilliard phase field equations☆29Feb 18, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Neural Emulator Architectures in JAX.☆24Feb 20, 2026Updated last month
- CoSaMP algorithm in Python language for sparse recovery problems.☆10Jan 14, 2022Updated 4 years ago
- Kernels, the machine learning ones☆14Mar 6, 2023Updated 3 years ago
- Datasets and code for results presented in the ProbConserv paper☆56Jun 14, 2024Updated last year
- DM i AI 2022☆10Oct 4, 2022Updated 3 years ago
- Implementation of physics informed neural networks with PyTorch☆20Jan 5, 2021Updated 5 years ago
- PySpectral is a Python package for solving the partial differential equation (PDE) of Burgers' equation in its deterministic and stochast…☆15Jan 26, 2023Updated 3 years ago
- 2D Burger's Equation (Convection + Diffusion)☆10May 13, 2021Updated 4 years ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆50Sep 14, 2024Updated last year
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Python package 'dgpsi' for deep and linked Gaussian process emulations☆27Mar 17, 2026Updated last week
- ☆42Jul 22, 2024Updated last year
- FCOS: Fully Convolutional One-Stage Object Detection☆11Oct 13, 2019Updated 6 years ago
- [NeurIPS 2021] Galerkin Transformer: Neural Operator built on Attention for PDEs☆257Jun 14, 2024Updated last year
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month
- Applications of PINOs☆147Oct 10, 2022Updated 3 years ago
- Code for the paper Universal Physics Transformers☆154Jul 2, 2025Updated 8 months ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆12May 8, 2023Updated 2 years ago
- Flux reconstruction fluid flow solver for 1D PDEs written in Julia. Linear advection, Burgers, viscous Burgers, and Euler equations.☆13Apr 28, 2022Updated 3 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆58May 23, 2023Updated 2 years ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆171Mar 2, 2026Updated 3 weeks ago
- ☆15Nov 18, 2024Updated last year
- Multiwavelets-based operator model☆69Feb 16, 2022Updated 4 years ago
- Minimal PyTorch implementation of SOLOv2.☆14Jan 14, 2025Updated last year
- The Senseiver: attention-based global field reconstruction from sparse observations☆30Nov 21, 2024Updated last year
- Solving High Dimensional Partial Differential Equations with Deep Neural Networks☆34Dec 21, 2021Updated 4 years ago