I haven’t done any tasks including edge features. You could check the following graph libraries https://neptune.ai/blog/graph-neural-networks-libraries-tools-learning-resources