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

120 citations · 233 across the 8 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…