Wahlstrom läuft auf internationaler Ebene für die Vereinigten Staaten auf und vertrat diese erstmals bei den Olympischen Jugend-Winterspielen 2016, wo er mit dem Team prompt die Goldmedaille gewann. Mit den U17- und U18-Auswahlen, die jeweils vom NTDP gestellt werden, nahm er anschließend an der World U-17 Hockey Challenge 2016 (5. Platz) sowie an den U18-Weltmeisterschaften 2017 und 2018 teil. Dabei errang man 2017 den Weltmeistertitel, während die Titelverteidigung 2018 an einer 2:3-Finalniederlage gegen Finnland scheiterte, wobei Wahlstrom selbst jedoch nach neun Punkten in sieben Partien ins All-Star-Team des Turniers gewählt wurde. In der Folge gehörte der Flügelstürmer auch zum Aufgebot der U20-Nationalmannschaft seines Heimatlandes bei den U20-Weltmeisterschaften 2019 und 2020. Dort wiederum gewann man 2019 nach einer erneuten 2:3-Endspielniederlage gegen die finnische Auswahl eine weitere Silbermedaille, bevor 2020 ein sechster Rang folgte.