A foundational tool that my research relies on is CARDIAX. This is a fully differentiable, GPU-enabled finite element library based in JAX with a focus on solving cardiac mechanics problems. It is maintained by the WCCMS group.

Here we demonstrate the ability to recover the true stiffness of a hyperelastic cube. The video shows steps taken from initial guess to true solution.

This video shows the ability to solve a reaction-diffusion equation over time. This is the simplified version of the Monodomain problem because of the lack of ion channels.

This is the ultimate example, solving a realistic biventricular ovine model. An inverse model is solved for muscle contraction to match the measured pressure-volume loop.