Computational solid mechanics made easy with Jax - View it on GitHub