GitHub repository: Ceyron/exponax
Primary language: Jupyter Notebook
Description: Efficient Differentiable n-d PDE solvers in JAX.