Elon Musk kommer att fortsätta som rådgivare till USA:s president Donald Trump, uppger vicepresident JD Vance.