An unofficial implementation of the Fourier Neural Operator in Flax
☆19May 25, 2024Updated 2 years 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:
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆29Jul 21, 2026Updated last week
- ☆68Mar 21, 2025Updated last year
- Code of ICML paper arxiv.org/abs/2302.08105☆14May 4, 2023Updated 3 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Official codebase for "Score-based Diffusion Models in Function Space"☆20Jan 23, 2025Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆117Dec 13, 2024Updated last year
- ☆16Nov 20, 2023Updated 2 years ago
- ☆10Jun 13, 2022Updated 4 years ago
- Staggered HARd-constrained Physics-Informed Neural Networks for Phase-field modelling of corrosion☆16Feb 19, 2025Updated last year
- 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,177Mar 30, 2026Updated 4 months ago
- ☆28Dec 8, 2023Updated 2 years ago
- Gaussian Mixture Models in Jax☆26Jul 14, 2025Updated last year
- Powerful shallow-water implementations in pure Python☆18Aug 18, 2024Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- CoSaMP algorithm in Python language for sparse recovery problems.☆10Jan 14, 2022Updated 4 years ago
- ☆10Aug 5, 2020Updated 5 years ago
- Kernels, the machine learning ones☆14Mar 6, 2023Updated 3 years ago
- Implementation of physics informed neural networks with PyTorch☆22Jan 5, 2021Updated 5 years ago
- Datasets and code for results presented in the ProbConserv paper☆55Jun 14, 2024Updated 2 years ago
- Implementation of deconvolution algorithms using torch.☆15Oct 19, 2022Updated 3 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
- Python package 'dgpsi' for deep and linked Gaussian process emulations☆28Mar 17, 2026Updated 4 months ago
- Neural Emulator Architectures in JAX.☆25Jun 1, 2026Updated last month
- 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.
- JAX-DIPS is a differentiable interfacial PDE solver.☆53Sep 14, 2024Updated last year
- Stainless neural networks in JAX☆36Jun 13, 2026Updated last month
- Code for the paper "Refining Language Model with Compositional Explanation" (NeurIPS 2021)☆11Oct 25, 2021Updated 4 years ago
- [ICML 2023] Non-Uniform Neural Operator (NUNO)☆27Jul 3, 2023Updated 3 years ago
- Physics-Constrained Flow Matching: Sampling Generative Models with Hard Constraints (NeurIPS 2025)☆27Jul 10, 2026Updated 2 weeks ago
- [NeurIPS 2021] Galerkin Transformer: Neural Operator built on Attention for PDEs☆267Jun 14, 2024Updated 2 years ago
- Flux reconstruction fluid flow solver for 1D PDEs written in Julia. Linear advection, Burgers, viscous Burgers, and Euler equations.☆14Apr 28, 2022Updated 4 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆58May 23, 2023Updated 3 years ago
- Applications of PINOs☆149Oct 10, 2022Updated 3 years 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.
- ☆15Nov 18, 2024Updated last year
- Multiwavelets-based operator model☆70Feb 16, 2022Updated 4 years ago
- This repository is for the master thesis project: "DiffFluids: A Diffusion-based Generative Model for Fluid Simulations"☆13Oct 2, 2023Updated 2 years ago
- Solving High Dimensional Partial Differential Equations with Deep Neural Networks☆34Dec 21, 2021Updated 4 years ago
- The 2D discrete wavelet transform for JAX☆45Feb 28, 2023Updated 3 years ago
- JAX Multi-Agent RL, Neuro-Evolution, and A-Life Library☆15Oct 12, 2025Updated 9 months ago
- Minimal PyTorch implementation of SOLOv2.☆15Jan 14, 2025Updated last year