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 2025Show all

2 papers · 1 filter

physics.comp-ph2025

Efficient GPU-computing simulation platform JAX-PF for differentiable phase field model

Fanglei Hu, Jiachen Guo, Stephen Niezgoda +2

We present JAX-PF, an open-source, GPU-accelerated, and differentiable Phase Field (PF) software package, supporting both explicit and implicit time stepping schemes. Leveraging th…

math.NA2025

Time-marching multi-level variational multiscale tensor decomposition algorithm for heat conduction with moving heat source

Xinyi Guan, Jiayi Hu, Lei Zhang +2

In this paper, we propose a time-marching multi-level Variational Multiscale-Tensor Decomposition (VMS-TD) algorithm to solve the heat equation with a moving heat source model that…