An unofficial implementation of the Fourier Neural Operator in Flax
☆19Aug 12, 2026Updated 3 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Code of ICML paper arxiv.org/abs/2302.08105☆14May 4, 2023Updated 3 years ago
- ☆68Mar 21, 2025Updated last year
- Training methodologies for autoregressive neural operators/emulators in JAX.☆16Jun 1, 2026Updated 5 months ago
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆120Dec 13, 2024Updated last year
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Code for "Robust flow field reconstruction from limited measurements vis sparse representation" (J. Callaham, K. Maeda, and S. Brunton 20…☆14Oct 23, 2018Updated 8 years ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆58Sep 14, 2024Updated 2 years ago
- ☆10Jun 13, 2022Updated 4 years ago
- Official codebase for "Score-based Diffusion Models in Function Space"☆20Jan 23, 2025Updated last year
- Code for Mesh Transformer describes in the EAGLE dataset☆43Feb 21, 2025Updated last year
- Powerful shallow-water implementations in pure Python☆18Aug 18, 2024Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 years ago
- Practicum on Supervised Learning in Function Spaces☆36Feb 17, 2022Updated 4 years ago
- [ICLR 2025] Neural Operator-Assisted Computational Fluid Dynamics in PyTorch☆88Nov 21, 2025Updated 11 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆10Aug 5, 2020Updated 6 years ago
- ☆27Dec 8, 2023Updated 2 years ago
- Staggered HARd-constrained Physics-Informed Neural Networks for Phase-field modelling of corrosion☆19Feb 19, 2025Updated last year
- ☆16Nov 20, 2023Updated 2 years ago
- PDEBench: An Extensive Benchmark for Scientific Machine Learning☆1,231Mar 30, 2026Updated 7 months ago
- EUMETSAT Agent Skills for data access & discovery services.☆33Updated this week
- ☆12Feb 26, 2025Updated last year
- Datasets and code for results presented in the ProbConserv paper☆55Jun 14, 2024Updated 2 years ago
- ✨💯Gaussian Mixture Models in JAX 💯✨☆26Aug 7, 2026Updated 3 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Implementation of physics informed neural networks with PyTorch☆23Jan 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
- Official repo for separable operator networks -- extreme-scale operator learning for parametric PDEs.☆41Nov 2, 2024Updated 2 years ago
- ☆53Jun 8, 2026Updated 5 months ago
- Python interface to the RTE+RRTMGP Fortran software package☆18Updated this week
- Quantile regression neural networks☆20Dec 2, 2023Updated 2 years ago
- This repository is the official implementation of the paper Convolutional Neural Operators for robust and accurate learning of PDEs☆237Nov 24, 2025Updated 11 months ago
- ☆10Nov 2, 2017Updated 9 years ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆234Nov 1, 2026Updated last week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Linear Operators in JAX.☆12May 11, 2023Updated 3 years ago
- Implementation of deconvolution algorithms using torch.☆15Oct 19, 2022Updated 4 years ago
- Kernels, the machine learning ones☆15Mar 6, 2023Updated 3 years ago
- Code for the paper "Refining Language Model with Compositional Explanation" (NeurIPS 2021)☆11Oct 25, 2021Updated 5 years ago
- Tree traversals using JAX☆11Jun 2, 2026Updated 5 months ago
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 3 years ago
- Python package 'dgpsi' for deep and linked Gaussian process emulations☆29Updated this week
Are these results useful?