The son-in-law of Trump's attorney-general nominee is reportedly leaving the Justice Department to join president's White House legal team