1 code implementation • 28 Apr 2023 • Guoliang He, Sean Parker, Eiko Yoneki
Tensor graph superoptimisation systems perform a sequence of subgraph substitution to neural networks, to find the optimal computation graph structure.
1 code implementation • 3 May 2022 • Sean Parker, Sami Alabed, Eiko Yoneki
Our proposed approach RLFlow can learn to perform neural network subgraph transformations, without the need for expertly designed heuristics to achieve a high level of performance.
Model-based Reinforcement Learning reinforcement-learning +1