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