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