DGL implementation of GTN

Hi,
I was wondering if there is an implementation of the Graph Transformer Network using DGL.
I am specifically refferring to this paper.


It is highly memory intensive, and I since DGLs are supposed to wok well with huge graphs, it would really help if a GTN module exists for GTN.
Thanks
Janani