Machine Learning, Lattice Structures, Graph Neural Networks, Architecture, Geometry.
within the emerging field of Geometric Deep Learning (aka deep learning on graphs and manifolds). We settled on using a graph neural network, as implemented in PyTorch Geometric. This allowed us to pass the lattice information to the network as an adjacency matrix, together with features on the