How AI Training Can Lead to Productivity Gains