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…