JAX-FEM: A differentiable GPU-accelerated 3D finite element solver for automatic inverse design and mechanistic data science - Xue, Tianju; Liao, Shuheng; Gan, Zhengtao; Park, Chanwook; Xie, Xiaoyu; Liu, Wing Kam; Cao, Jian (Computer Physics Communications, 2023)