https://medium.com/@monocosmo77/updates-on-neural-algorithmic-reasoning-part4-machine-learning-b3f7f5086cb4