/jax-fem_cfd

Differentiable Finite Element Method with JAX

Primary LanguagePython

JAX-FEM

A differentiable 2D/3D finite element solver for automatic inverse design and mechanistic data science. This is the source code directory. Please visit tutorial for more information.