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