Efficient Differentiable n-d PDE solvers in JAX.
deep-learning simulation fft pde equinox spectral-methods pde-solver jax differentiable-physics etdrk
-
Updated
Oct 28, 2025 - Jupyter Notebook