Nach einem verhaltenen Beginn der &lt;a href="Freestyle-Skiing-Weltcup%202016/17"&gt;Saison 2016/17&lt;/a&gt; erreichte Wallberg Mitte Januar 2017 mit dem fünften Platz im Moguls in Lake Placid seine erste Top-10-Platzierung im Weltcup. Beim folgenden Weltcup in Val Saint-Côme kam er mit dem dritten Platz im Moguls-Wettbewerb erstmals aufs Podium. Im April 2017 wurde er schwedische Meister im Dual Moguls, bei den Juniorenweltmeisterschaften 2017 in Chiesa in Valmalenco wurde er Sechster im Moguls und Vierter im Dual Moguls. Die FIS kürte ihn zum "Rookie of the Year". Im folgenden Jahr errang er bei den Olympischen Winterspielen in Pyeongchang den 21. Platz im Moguls-Wettbewerb und gewann bei den Juniorenweltmeisterschaften in Duved die Goldmedaille im Dual Moguls. Zudem wurde er in Bygsiljum schwedischer Meister im Moguls und Dual Moguls. Nach Platz drei im Moguls in Ruka zu Beginn der Saison 2018/19, errang er in Calgary und Lake Placid jeweils den zweiten und in Shymbulak den dritten Platz und erreichte damit den achten Platz im Gesamtweltcup und den vierten Rang im Moguls-Weltcup. Beim Saisonhöhepunkt, den Freestyle-Skiing-Weltmeisterschaften 2019 in Park City, kam er auf den neunten Platz im Moguls und auf den vierten Rang im Dual Moguls. Ende März 2019 wurde er im Kungsberget erneut schwedischer Meister im Dual Moguls. In der Saison 2019/20 belegte er mit zwei dritten Plätzen und einen zweiten Platz, den siebten Platz im Moguls-Weltcup.