AWS is investing billions to put AI into production for the public sector