Martin Štajnoch reprezentoval Slovensko na majstrovstvách sveta v ľadovom hokeji hráčov do 18 rokov v roku 2008, kde bol vybraný aj do trojice najlepších hráčov Slovenska na šampionáte. Na juniorských majstrovstách sveta do 20 rokov reprezentoval Slovensko v roku 2009. Na nich bol opäť vyhlásený do trojice najlepších hráčov tímu a zároveň bol vyhlásený za najlepšieho hráča zápasu proti Švédsku. Vďaka svojim výkonom na juniorských majstrovstvách sveta ho tréner seniorskej reprezentácie Ján Filc nominoval na prípravné zápasy proti Bielorusku a Švajčiarsku.