AI, productivity, and diffusion