Welcome to JAX-FEM’s documentation!

🚀 JAX-FEM is a differentiable finite element package based on JAX.

Linear static analysis of a bracket.

Topology optimization with differentiable simulation.

Stokes flow: velocity (left) and pressure(right).

Crystal plasticity: grain structure (left) and stress-xx (right).

Thermal profile in direct energy deposition.