AI for Rydberg-Atom Quantum Simulation