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