The example implementation of GIN in DGL uses a class called “MLP”. The code of that class matches with authors implementation line by line.
Here’s the example implementation using DGL
Here’s the authors implementation.
In the DGL example the author’s implementation is referenced but it’s not explicitly stated that a part of it is used.