Frequently asked questions (FAQ)

Frequently asked questions (FAQ)#

General#

  • I saw JAX-FEM supports third-party non-differentiable solvers such as PETSc. Does that mean it is not differentiable if I use PETSc?

It is still differentiable. We apply implicit differentiation based on the adjoint formulation to cover that. We never have to differentiate through the linear solver or Newton’s method involved in the forward problem.

Mesh#

  • How to check the supported mesh type?

You can refer to jax_fem.basis.get_elements

Solver#

  • How to utilize GPU to accelerate my computation?

Make sure that you have installed the GPU version of jax and the computation will be automatically performed on GPU.

Remarks#

If your question isn’t covered above, you can search existing issues or open a new issue on Github.

Contents