AI in chip architecture