The U.S. Department of the Treasury’s assistant secretary for tax policy will leave the role to return to New York University School of Law in the spring as a tax law professor, the school announced.