In April 2021, Wallace was named as a coach for the United States national baseball team, for the team's final efforts to qualify for baseball at the 2020 Summer Olympics in Tokyo in 2021. The team qualified, with Wallace serving as the team's pitching coach for the Olympics. The team went on to win silver, falling to Japan in the gold-medal game.
