Work with Diego to start looking which nodes need to be ML, and implementation of them + PyTorch script