White House chief of staff Ron Klain, who's spent more than two years as President Joe Biden’s top aide, is preparing to leave his job in the coming weeks.