task_categories:text-generation language:en size_categories:1K<n<10K format:parquet modality:text library:datasets library:pandas library:mlcroissant library:polars arxiv:2510.00237 region:us sft reinforcement-learning sokoban general-points chain-of-thought reasoning decision-making prompt-diversity