Submissions for team AlphaStar Transformers on problem nah_main