JAX-FEM

A Differentiable Finite Element Method Library based on JAX.