Official Implementation of ICML 2025 "ROS: A GNN-based Relax-Optimize-and-Sample Framework for Max-k-Cut Problems"