Submissions for team JAX on problem piezoelectric_main