lineno">22</span><span class="kn">from</span> <span class="nn">labml_helpers.device</span> <span class="kn">import</span> <span class="n">DeviceConfigs</span>
<span class="lineno">23</span><span class="kn">from</span> <span class="nn">labml_helpers.module</span> <span class="kn">import</span> <span class="n">Module</span>
<span class="lineno">24</span><span class="kn">from</span> <span class="nn">labml_nn.graphs.gat</span> <span class="kn">import</span> <span class="n">GraphAttentionLayer</span>
<span class