In 2022, Klein enrolled at the University of Michigan and played in two games as a freshman before redshirting the season.[5] In 2023, he played in ten games, winning the national championship with Michigan that season. He also made his first collegiate reception, finishing the year with one catch for eight yards.[6]
In 2024, Klein moved into a larger role. In week three versus Arkansas State, he was Michigan’s leading receiver with three receptions for 43 yards.[7] Klein played in all 13 games for the Wolverines, starting six times. He totaled 13 receptions for 108 yards and was a key contributor in the running game as a blocker.[8]