What is FluxFEM? A weak-form-centric differentiable finite element framework in JAX, where variational forms are treated as first-class, differentiable programs. https://github.com/kevin-tofu/fluxfem Coding nonlinear structural analysis with FluxFEM Neohookean with Weakform Newton method interface Result It seems completely fine.