["size_categories:1M<n<10M", "format:parquet", "library:datasets", "library:dask", "library:mlcroissant", "library:polars", "region:us"]
[{"name": "language", "dtype": "int64"}, {"name": "input_ids", "sequence": "int32"}, {"name": "attention_mask", "sequence": "int8"}, {"name": "depths", "sequence": "int64"}, {"name": "sibling_indices", "sequence": "int64"}, {"name": "tree_attention_mask", "sequence": "int64"}]