search

deepmodeling / jax-fem

Differentiable Finite Element Method with JAX
GNU General Public License v3.0
248 stars 40 forks source link

Time-progressive FEM #33

Open MRiabov opened 2 weeks ago

MRiabov commented 2 weeks ago

Hello JAX-FEM, I'm a deep modelling engineer myself and I've ran into an issue that the current time-dependant (multibody, ones that progress over time) open-source solvers don't have GPU support, only CPU. (Project Chrono FEM does not support GPU)

I will have to write a multibody solver myself, but I'm an AI engineer, not a physics sim PhD.

Would JAX-FEM be suitable as a part of an existing code base? Would anybody help implement it?

Cheers, Maksym Riabov

tianjuxue commented 2 weeks ago

Please go through the demo examples and see if they fit your purpose. For multi-body solver, note that we don't handle contact at this point.