Extended Data Fig. 4: Detailed view of AlphaTensor’s architecture, included torso, policy and value head.
From: Discovering faster matrix multiplication algorithms with reinforcement learning
We refer to Algorithms A.1-A.11 in Supplementary Information for the details of each component.