Ted Chiang is incredible, my favorite writer.
I also recommend "Exhalation", though that has nothing to do with AI.