search

ucl-bug / jaxdf

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
GNU Lesser General Public License v3.0
117 stars 7 forks source link

MPI FiniteDifferences #94

Open astanziola opened 2 years ago

astanziola commented 2 years ago

It would be nice to have a new FinitedDifferences-like class based on mpi4jax, to enable HPC applications.

A good milestone could be to reproduce the shallow water example of the library (original code here)