most citedJAX-FEM: A differentiable GPU-accelerated 3D finite element solver for automatic inverse design and mechanistic data science

120 citations · 120 across the 1 of their papers we have counts for

collaborators

1 paper

cs.MS2022★ 120 cited

JAX-FEM: A differentiable GPU-accelerated 3D finite element solver for automatic inverse design and mechanistic data science

Tianju Xue, Shuheng Liao, Zhengtao Gan +4

This paper introduces JAX-FEM, an open-source differentiable finite element method (FEM) library. Constructed on top of Google JAX, a rising machine learning library focusing on hi…