RLFlow: Optimising Neural Network Subgraph Transformation with World Models. - Sean Parker; Sami Alabed; Eiko Yoneki (CoRR, 2022)

X-RLflow: Graph Reinforcement Learning for Neural Network Subgraphs Transformation. - Guoliang He; Sean Parker; Eiko Yoneki (MLSys, 2023)