Towards Learning Self-Organized Criticality of Rydberg Atoms using Graph Neural Networks