Lee won gold alongside Tom Daley at the 2020 Tokyo Olympics