task_categories:text-generation language:en license:apache-2.0 size_categories:10B<n<100B format:parquet library:datasets library:dask library:polars library:mlcroissant arxiv:2505.20225 region:us mixture-of-experts routing-traces moe