ai_infra_bench_dataset
