Wahlström pelasi myös alle 18-vuotiaiden MM-kilpailuissa vuosina 2017 ja 2018, joista ensimmäinen turnaus päättyi maailmanmestaruuteen ja jälkimmäinen hopeaan. Vuoden 2017 turnauksessa hän oli neljällä osumallaan Yhdysvaltain paras maalintekijä Maxwell Gildonin kanssa. Wahlstrom valittiin Valko-Venäjää vastaan pelatun avausottelun päätteeksi joukkueensa ottelun parhaaksi pelaajaksi. Vuoden 2018 turnauksessa hän teki välieräottelussa Tšekkiä vastaan Yhdysvaltain voittomaalin joukkueen edettyä finaaliin lukemin 4–1. Wahlstrom valittiin Yhdysvaltain ottelun parhaaksi pelaajaksi, kuten myös Kanadaa vastaan pelatun avausottelun päätteeksi. Hän sijoittui lopulta seitsemällä osumallaan koko turnauksen maalipörssissä toiseksi edellään vain Suomen Niklas Nordgren. Wahlstrom valittiin turnauksen tähdistökentälliseen ja valmennus valitsi hänet Yhdysvaltain yhdeksi kolmesta parhaasta pelaajasta.