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