There has been speculation that Mr Cohn, a supporter of free trade, was angered by Mr Trump's plans to impose tariffs on aluminium and steel imports.
In a statement released by the White House, Mr Cohn said it had been "an honour to serve my country".