Change log#
JAX-FEM 0.0.10 (Aug 08, 2025)#
Major updates
Use
jax.tree_util.tree_map()
to replace the deprecatedjax.tree_map()
. See JAX updates jax 0.4.26 (April 3, 2024).Support AMGX linear solver for better performance on GPU.
JAX-FEM 0.0.9 (May 05, 2025)#
Major updates
Support customized linear solvers
JAX-FEM 0.0.8 (Oct 30, 2024)#
Major updates
Implement the arc-length solver
JAX-FEM 0.0.7 (Aug 18, 2024)#
Major updates
Update
jax_fem.solver
to includeUMFPACK
solver provided byscipy