Improving Equality and Productivity with AI-Driven Tax Policies