Submissions for team JAX on problem shadows_main