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