Welcome to JAX-FEM’s documentation!

🚀 JAX-FEM is a GPU-accelerated differentiable finite element analysis package based on JAX. Used to be part of the suite of open-source python packages for Additive Manufacturing (AM) research, JAX-AM.

💻 This project was initially developed by Prof. Tianju Xue during his postdoctoral research at NU and is currently being actively maintained by the CMSL research group at the HKUST.

🔥 We warmly welcome contributions from the community (e.g., bug reports, feature requests, or code submissions).

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.