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

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

collaborators
Showing cs.MSShow all

1 paper · 1 filter

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…