Learning rigid dynamics with face interaction graph networks