["size_categories:1M<n<10M", "format:parquet", "modality:text", "library:datasets", "library:dask", "library:mlcroissant", "library:polars", "region:us"]
[{"name": "query", "dtype": "string"}, {"name": "document", "dtype": "string"}, {"name": "query_ids", "sequence": "int32"}, {"name": "query_attention_mask", "sequence": "int8"}, {"name": "context_ids", "sequence": "int32"}, {"name": "context_attention_mask", "sequence": "int8"}]