├── .gitignore ├── LICENSE ├── README.md ├── pyproject.toml ├── r2r ├── __init__.py ├── data │ ├── README.md │ ├── data_process.py │ ├── generation_controller.py │ ├── live_labeler.py │ ├── utils │ │ └── convert_eos_tokens.py │ └── verify_model.py ├── evaluate │ ├── codegen_metrics.py │ ├── eval_utils.py │ └── extract_livecodebench_answer.py ├── models │ ├── __init__.py │ ├── batch_inference │ │ ├── llm_server.py │ │ ├── schedule_req.py │ │ ├── sl_disaggregation_system.py │ │ └── slm_server.py │ ├── dynamic_sglang_selector.py │ ├── dynamic_sglang_selector_opt.py │ ├── dynamic_sglang_selector_simple.py │ ├── recorder.py │ ├── router.py │ └── sglang_wrapper.py ├── train │ ├── logger.py │ ├── loss.py │ └── optimizer.py └── utils │ ├── __init__.py │ ├── config.py │ ├── dataclass.py │ ├── dataset_conversion.py │ ├── metrics.py │ ├── model_configs.json │ ├── sampling.py │ ├── switching.py │ └── token_manager.py ├── resource ├── default_training_config.json ├── default_training_config_qwen3_0_6b.json ├── default_training_config_qwen3_1_7b.json ├── default_training_config_qwen3_4b.json ├── default_training_config_qwen_r1.json └── logo.png └── script ├── data_labeling ├── auto_scheduler.py ├── data_generation.py ├── init_dataset_conversion.py ├── run_pipeline_scripts.md ├── step_0_llm_response.py ├── step_1_slm_prefill.py ├── step_2_llm_continuation.py ├── step_3_verify.py ├── step_4_construct_label_dataset.py └── support_dataset_config.json ├── data_labeling_api ├── run_pipeline_scripts.md ├── step_0_llm_response.py └── step_2_llm_continuation.py ├── evaluate ├── eval_configs │ ├── dataset_configs.json │ ├── model_configs.json │ └── r2r_configs.json └── hf_dataset_sglang.py ├── experimental └── live_labeler │ ├── demo_divergent_labeler.py │ ├── extract_live_continuation_results.py │ └── live_continuation.py ├── playground ├── demo │ └── gradio_demo.py ├── generate_job_scripts.py ├── interactive_chat.py ├── launch_llm_continuation_multi_node.py └── speed_benchmark.py ├── train ├── train_router.py └── train_router_param_search.py └── utils └── generate_pipeline_scripts.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/README.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/pyproject.toml -------------------------------------------------------------------------------- /r2r/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /r2r/data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/data/README.md -------------------------------------------------------------------------------- /r2r/data/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/data/data_process.py -------------------------------------------------------------------------------- /r2r/data/generation_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/data/generation_controller.py -------------------------------------------------------------------------------- /r2r/data/live_labeler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/data/live_labeler.py -------------------------------------------------------------------------------- /r2r/data/utils/convert_eos_tokens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/data/utils/convert_eos_tokens.py -------------------------------------------------------------------------------- /r2r/data/verify_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/data/verify_model.py -------------------------------------------------------------------------------- /r2r/evaluate/codegen_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/evaluate/codegen_metrics.py -------------------------------------------------------------------------------- /r2r/evaluate/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/evaluate/eval_utils.py -------------------------------------------------------------------------------- /r2r/evaluate/extract_livecodebench_answer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/evaluate/extract_livecodebench_answer.py -------------------------------------------------------------------------------- /r2r/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /r2r/models/batch_inference/llm_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/models/batch_inference/llm_server.py -------------------------------------------------------------------------------- /r2r/models/batch_inference/schedule_req.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/models/batch_inference/schedule_req.py -------------------------------------------------------------------------------- /r2r/models/batch_inference/sl_disaggregation_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/models/batch_inference/sl_disaggregation_system.py -------------------------------------------------------------------------------- /r2r/models/batch_inference/slm_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/models/batch_inference/slm_server.py -------------------------------------------------------------------------------- /r2r/models/dynamic_sglang_selector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/models/dynamic_sglang_selector.py -------------------------------------------------------------------------------- /r2r/models/dynamic_sglang_selector_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/models/dynamic_sglang_selector_opt.py -------------------------------------------------------------------------------- /r2r/models/dynamic_sglang_selector_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/models/dynamic_sglang_selector_simple.py -------------------------------------------------------------------------------- /r2r/models/recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/models/recorder.py -------------------------------------------------------------------------------- /r2r/models/router.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/models/router.py -------------------------------------------------------------------------------- /r2r/models/sglang_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/models/sglang_wrapper.py -------------------------------------------------------------------------------- /r2r/train/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/train/logger.py -------------------------------------------------------------------------------- /r2r/train/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/train/loss.py -------------------------------------------------------------------------------- /r2r/train/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/train/optimizer.py -------------------------------------------------------------------------------- /r2r/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /r2r/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/utils/config.py -------------------------------------------------------------------------------- /r2r/utils/dataclass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/utils/dataclass.py -------------------------------------------------------------------------------- /r2r/utils/dataset_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/utils/dataset_conversion.py -------------------------------------------------------------------------------- /r2r/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/utils/metrics.py -------------------------------------------------------------------------------- /r2r/utils/model_configs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/utils/model_configs.json -------------------------------------------------------------------------------- /r2r/utils/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/utils/sampling.py -------------------------------------------------------------------------------- /r2r/utils/switching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/utils/switching.py -------------------------------------------------------------------------------- /r2r/utils/token_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/r2r/utils/token_manager.py -------------------------------------------------------------------------------- /resource/default_training_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/resource/default_training_config.json -------------------------------------------------------------------------------- /resource/default_training_config_qwen3_0_6b.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/resource/default_training_config_qwen3_0_6b.json -------------------------------------------------------------------------------- /resource/default_training_config_qwen3_1_7b.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/resource/default_training_config_qwen3_1_7b.json -------------------------------------------------------------------------------- /resource/default_training_config_qwen3_4b.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/resource/default_training_config_qwen3_4b.json -------------------------------------------------------------------------------- /resource/default_training_config_qwen_r1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/resource/default_training_config_qwen_r1.json -------------------------------------------------------------------------------- /resource/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/resource/logo.png -------------------------------------------------------------------------------- /script/data_labeling/auto_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling/auto_scheduler.py -------------------------------------------------------------------------------- /script/data_labeling/data_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling/data_generation.py -------------------------------------------------------------------------------- /script/data_labeling/init_dataset_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling/init_dataset_conversion.py -------------------------------------------------------------------------------- /script/data_labeling/run_pipeline_scripts.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling/run_pipeline_scripts.md -------------------------------------------------------------------------------- /script/data_labeling/step_0_llm_response.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling/step_0_llm_response.py -------------------------------------------------------------------------------- /script/data_labeling/step_1_slm_prefill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling/step_1_slm_prefill.py -------------------------------------------------------------------------------- /script/data_labeling/step_2_llm_continuation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling/step_2_llm_continuation.py -------------------------------------------------------------------------------- /script/data_labeling/step_3_verify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling/step_3_verify.py -------------------------------------------------------------------------------- /script/data_labeling/step_4_construct_label_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling/step_4_construct_label_dataset.py -------------------------------------------------------------------------------- /script/data_labeling/support_dataset_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling/support_dataset_config.json -------------------------------------------------------------------------------- /script/data_labeling_api/run_pipeline_scripts.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling_api/run_pipeline_scripts.md -------------------------------------------------------------------------------- /script/data_labeling_api/step_0_llm_response.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling_api/step_0_llm_response.py -------------------------------------------------------------------------------- /script/data_labeling_api/step_2_llm_continuation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/data_labeling_api/step_2_llm_continuation.py -------------------------------------------------------------------------------- /script/evaluate/eval_configs/dataset_configs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/evaluate/eval_configs/dataset_configs.json -------------------------------------------------------------------------------- /script/evaluate/eval_configs/model_configs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/evaluate/eval_configs/model_configs.json -------------------------------------------------------------------------------- /script/evaluate/eval_configs/r2r_configs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/evaluate/eval_configs/r2r_configs.json -------------------------------------------------------------------------------- /script/evaluate/hf_dataset_sglang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/evaluate/hf_dataset_sglang.py -------------------------------------------------------------------------------- /script/experimental/live_labeler/demo_divergent_labeler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/experimental/live_labeler/demo_divergent_labeler.py -------------------------------------------------------------------------------- /script/experimental/live_labeler/extract_live_continuation_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/experimental/live_labeler/extract_live_continuation_results.py -------------------------------------------------------------------------------- /script/experimental/live_labeler/live_continuation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/experimental/live_labeler/live_continuation.py -------------------------------------------------------------------------------- /script/playground/demo/gradio_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/playground/demo/gradio_demo.py -------------------------------------------------------------------------------- /script/playground/generate_job_scripts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/playground/generate_job_scripts.py -------------------------------------------------------------------------------- /script/playground/interactive_chat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/playground/interactive_chat.py -------------------------------------------------------------------------------- /script/playground/launch_llm_continuation_multi_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/playground/launch_llm_continuation_multi_node.py -------------------------------------------------------------------------------- /script/playground/speed_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/playground/speed_benchmark.py -------------------------------------------------------------------------------- /script/train/train_router.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/train/train_router.py -------------------------------------------------------------------------------- /script/train/train_router_param_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/train/train_router_param_search.py -------------------------------------------------------------------------------- /script/utils/generate_pipeline_scripts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thu-nics/R2R/HEAD/script/utils/generate_pipeline_scripts.py --------------------------------------------------------------------------------