JAX-FEM

A Differentiable Finite Element Method Library based on JAX (Collaborator)