max_num_batched_tokens