Čolak bestritt bereits für die Nachwuchsmannschaften des kroatischen Verbandes (HNS) Länderspiele. Sein Debüt für die U-18-Auswahlmannschaft gab er am 2. Mai 2011 in Bánovce nad Bebravou bei der 0:2-Niederlage gegen die Auswahl Norwegens. Für die U-19-Nationalmannschaft debütierte er am 8. Februar 2011 in Umag beim 1:1 im Spiel gegen die Auswahl Irlands, wobei er mit dem Führungstreffer in der 3. Minute sein erstes Länderspieltor erzielte. Mit der U-19-Nationalmannschaft nahm Čolak im Juli 2012 in Estland an der Europameisterschaft teil und bestritt alle drei Gruppenspiele. Für die U-20-Nationalmannschaft absolvierte er drei Länderspiele, sein erstes am 8. Dezember 2011 in Dekani beim 1:0-Sieg gegen die Auswahl Sloweniens. Im November 2020 debütierte Čolak in einem Freundschaftsspiel gegen die Türkei in der kroatischen A-Nationalmannschaft.