tags: ["task_categories:graph-ml", "license:cc-by-sa-4.0", "size_categories:1K<n<10K", "format:parquet", "library:datasets", "library:dask", "library:mlcroissant", "library:polars", "arxiv:2305.12871", "region:us", "physics learning", "geometry learning"]