After running "pip install labml_nn" I was able to find all directories except for /gatv2:

This was confirmed by a visit to pypi.org where after downloading the source file (labml-nn-0.4.103.tar.gz) found the following within the labml__nn/graphs/ directory:

I'm guessing this is a mistake since making a GraphAttentionv2 layer requires this file. I apologize if this isn't the right place to raise this.