https://github.com/dsgiitr/graph_nets?fbclid=IwAR39aQbvMVhVxNCbQOoxkxdkYgnfd3tMuqzvWBqz048i8iE8Sjavcqbh56M