A better preconditioner for jax solver

tianjuxue / jax-am

Additive manufacturing simulation with JAX.

https://jax-am.readthedocs.io/en/latest/

GNU General Public License v3.0

273 stars 56 forks source link

Open SNMS95 opened 1 year ago

SNMS95 commented 1 year ago

At the moment, for some BCs (especially in TopOpt), the petsc version converges but the jax_solve does not.
From our tests, we understood that this is due to the lack of a good preconditioner. PETSc solver has ilu as the preconditioner while the jax solver is using a simple jacobi

Can we implement an ILU preconditioner for jax solver ?