RydIQule: graph-based modeling of Rydberg quantum sensors in Python

We present the Rydberg Interactive Quantum Module (RydIQule), a Python package designed to flexibly and efficiently model quantum sensors based on thermal atoms. We then demonstrate the utility of RydIQule in a simulation of a thermal quantum sensor with a 5-tone input signal. We develop an architecture to represent multilevel atomic systems using directed network graphs. This approach is novel and enables Hamiltonian generation to be scripted, making it the natural way to represent such a system. Efficient use of tensor contractions allows associated differential equations to be solved feasibly over high-dimensional parameter spaces on a typical desktop computer, including over velocity classes in a doppler-broadened gas. Solutions can be generated in the steady state or time domain, even given arbitrary input signals. While RydIQule can be used to define systems in a barebones, atom-agnostic manner, it also contains built-in tools for calling the ARC library, fetching Rydberg atomic data and using it to populate parameters automatically. This array of simple, powerful features makes RydIQule a uniquely capable tool for modeling Rydberg sensing experiments in a timeframe shorter than that allowed by other available tools.