deepmind/gnn_single_rigids not found