Efficient Differentiable n-d PDE Solvers in JAX.
deep-learning simulation fft pde equinox spectral-methods pde-solver jax differentiable-physics etdrk
-
Updated
Feb 13, 2026 - Jupyter Notebook