Submitted paper to NeurIPS 2025.