Submissions for team JAX on problem mycelium_main