Efficient Differentiable n-d PDE Solvers in JAX.
-
Updated
Mar 2, 2026 - Jupyter Notebook
Efficient Differentiable n-d PDE Solvers in JAX.
Stochastic PDE solvers (SPDE) built on top of exponax: Exponential Euler-Maruyama stepper for the stochastic Allen-Cahn equation with additive/multiplicative Q-Wiener noise, tamed nonlinearities, ensemble utilities, Richardson extrapolation, and a Strang-split hybrid SSA scaffold.
Solver for the one dimensional Kuramoto-Sivashinsky using the ETDRK4 method.
Add a description, image, and links to the etdrk topic page so that developers can more easily learn about it.
To associate your repository with the etdrk topic, visit your repo's landing page and select "manage topics."