├── .github ├── CODEOWNERS ├── PULL_REQUEST_TEMPLATE.md ├── dependabot.yml └── workflows │ ├── README.md │ ├── check-pr-title.yml │ ├── checkpoint_converter.yml │ ├── cpu_unit_tests.yml │ ├── disabled │ └── e2e_prime.yml │ ├── doc.yml │ ├── e2e_ascend.yml │ ├── e2e_dapo.yml │ ├── e2e_eval_aime24.yml │ ├── e2e_genrm_remote.yml │ ├── e2e_one_step_off_policy.yml │ ├── e2e_ppo_trainer.yml │ ├── e2e_ppo_trainer_megatron_sglang.yml │ ├── e2e_ppo_trainer_megatron_vllm.yml │ ├── e2e_sft.yml │ ├── e2e_spin.yml │ ├── e2e_sppo.yml │ ├── gpu_unit_tests.yml │ ├── model.yml │ ├── pre-commit-full.yml │ ├── pre-commit.yml │ ├── sanity.yml │ ├── scorecard.yml │ ├── secrets_scan.yml │ ├── sgl.yml │ ├── type-coverage-check.yml │ └── vllm.yml ├── .gitignore ├── .readthedocs.yaml ├── LICENSE ├── Notice.txt ├── README.md ├── docker ├── Apptainerfile.rocm ├── Dockerfile.extention.awsefa ├── Dockerfile.ngc.vllm ├── Dockerfile.ngc.vllm0.8 ├── Dockerfile.ngc.vllm0.8.sagemaker ├── Dockerfile.rocm ├── Dockerfile.rocm_verl-0.3.0.post1 ├── Dockerfile.rocm_verl-0.4.1 ├── Dockerfile.sglang ├── Dockerfile.vemlp.vllm.te ├── Dockerfile.vllm.sglang.megatron.deepseek ├── README.md ├── verl0.4-cu124-torch2.6-fa2.7.4 │ ├── Dockerfile.app.sglang.vllm.mcore0.12 │ ├── Dockerfile.app.sglang.vllm.mcore0.12.deepep │ ├── Dockerfile.app.sglang.vllm.mcore0.13.preview │ ├── Dockerfile.app.vllm.mcore0.12 │ ├── Dockerfile.app.vllm.mcore0.12.deepep │ ├── Dockerfile.app.vllm.mcore0.13.preview │ ├── Dockerfile.base │ └── README.md ├── verl0.5-cu126-torch2.7.1-fa2.8.0 │ ├── Dockerfile.app.sglang.mcore0.12 │ ├── Dockerfile.app.sglang.mcore0.13.preview │ ├── Dockerfile.base │ └── README.md └── verl0.5-preview-cu128-torch2.7.1-fa2.8.0 │ ├── Dockerfile.app.sglang.megatron │ ├── Dockerfile.base │ └── README.md ├── docs ├── BandAI.png ├── Makefile ├── README.md ├── README_vllm0.7.md ├── README_vllm0.8.md ├── _static │ ├── js │ │ └── runllm-widget.js │ └── logo.png ├── advance │ ├── checkpoint.rst │ ├── dpo_extension.rst │ ├── fsdp_extension.rst │ ├── megatron_extension.rst │ ├── one_step_off.md │ ├── placement.rst │ ├── ppo_lora.rst │ ├── rollout_trace.rst │ └── rope.rst ├── algo │ ├── baseline.md │ ├── dapo.md │ ├── entropy.md │ ├── gpg.md │ ├── grpo.md │ ├── opo.md │ ├── ppo.md │ ├── spin.md │ └── sppo.md ├── amd_tutorial │ ├── amd_build_dockerfile_page.rst │ └── amd_vllm_page.rst ├── api │ ├── data.rst │ ├── single_controller.rst │ ├── trainer.rst │ └── utils.rst ├── ascend_tutorial │ ├── ascend_profiling.rst │ ├── ascend_profiling_en.rst │ └── ascend_quick_start.rst ├── conf.py ├── examples │ ├── config.rst │ ├── gsm8k_example.rst │ ├── multi_modal_example.rst │ ├── ppo_code_architecture.rst │ └── sandbox_fusion_example.rst ├── faq │ └── faq.rst ├── hybrid_flow.rst ├── index.rst ├── llm-example1.png ├── overall.png ├── perf │ ├── device_tuning.rst │ ├── dpsk.md │ ├── nsight_profiling.md │ └── perf_tuning.rst ├── preparation │ ├── prepare_data.rst │ └── reward_function.rst ├── requirements-docs.txt ├── sglang_multiturn │ ├── interaction_system.rst │ ├── multiturn.rst │ ├── sandbox_fusion.rst │ └── search_tool_example.rst ├── single_controller.rst ├── start │ ├── agentic_rl.rst │ ├── install.rst │ ├── more_resources.rst │ ├── multinode.rst │ ├── quickstart.rst │ └── ray_debug_tutorial.rst └── workers │ ├── fsdp_workers.rst │ ├── megatron_workers.rst │ ├── ray_trainer.rst │ └── sglang_worker.rst ├── evaluation ├── eval_mm.py ├── eval_text_only.py └── inference.py ├── examples ├── data_preprocess │ ├── aime2024_multiturn_w_tool.py │ ├── dapo_multiturn_w_tool.py │ ├── full_hh_rlhf.py │ ├── geo3k.py │ ├── geo3k_multiturn_w_tool.py │ ├── gsm8k.py │ ├── gsm8k_multiturn_w_interaction.py │ ├── gsm8k_multiturn_w_tool.py │ ├── gsm8k_tool_agent_loop.py │ ├── hellaswag.py │ ├── math_dataset.py │ ├── multiturn.py │ └── preprocess_search_r1_dataset.py ├── generation │ ├── run_deepseek7b_mutli_node.sh │ └── run_deepseek_v2_lite_math.sh ├── gpg_trainer │ ├── gpg.md │ ├── run_qwen2-7b_math.sh │ └── run_qwen2-7b_math_megatron.sh ├── grpo_trainer │ ├── README.md │ ├── run_deepseek671b_math_megatron.sh │ ├── run_deepseek7b_llm.sh │ ├── run_deepseek7b_llm_math.sh │ ├── run_deepseek7b_llm_math_megatron.sh │ ├── run_deepseek7b_llm_seq_balance.sh │ ├── run_minicpmo2_6.sh │ ├── run_moonlight16b_math_megatron.sh │ ├── run_qwen2-7b.sh │ ├── run_qwen2-7b_math.sh │ ├── run_qwen2-7b_math_megatron.sh │ ├── run_qwen2-7b_seq_balance.sh │ ├── run_qwen2-7b_seq_balance_math_megatron.sh │ ├── run_qwen2-7b_sgl_megatron.sh │ ├── run_qwen2_5-3b_gsm8k_grpo_lora.sh │ ├── run_qwen2_5-7b_math_megatron_diff_tp.sh │ ├── run_qwen2_5_32b_grpo_npu.sh │ ├── run_qwen2_5_7b_grpo_discrete_prof_npu.sh │ ├── run_qwen2_5_7b_grpo_e2e_prof_npu.sh │ ├── run_qwen2_5_7b_grpo_npu.sh │ ├── run_qwen2_5_vl-7b-megatron.sh │ ├── run_qwen2_5_vl-7b.sh │ ├── run_qwen2_5_vl-7b_lora.sh │ ├── run_qwen2_5_vl-7b_seq_balance.sh │ ├── run_qwen2_5_vl_32b_npu.sh │ ├── run_qwen2_5_vl_3b_npu.sh │ ├── run_qwen2_5_vl_7b_npu.sh │ ├── run_qwen3-236b_megatron.sh │ ├── run_qwen3-8b.sh │ └── run_qwen3moe-30b_megatron.sh ├── ppo_trainer │ ├── README.md │ ├── run_deepseek7b_llm.sh │ ├── run_deepseek7b_llm_modelscope.sh │ ├── run_deepseek7b_llm_pfppo.sh │ ├── run_deepseek7b_llm_sandbox_fusion.sh │ ├── run_deepseek7b_llm_sp2.sh │ ├── run_deepseek_full_hh_rlhf.sh │ ├── run_deepseek_math_gsm8k_megatron.sh │ ├── run_deepseek_math_gsm8k_megatron_nsys.sh │ ├── run_gemma.sh │ ├── run_moonlight16b_a3b_gsm8k_megatron.sh │ ├── run_qwen1.5_moe_a2.7b-gsm8k_megatron.sh │ ├── run_qwen2-7b_math_gsm8k_megatron.sh │ ├── run_qwen2-7b_rm.sh │ ├── run_qwen2-7b_rm_seq_balance.sh │ ├── run_qwen2-7b_rm_seq_balance_fused_kernels.sh │ ├── run_qwen2-7b_rm_seq_balance_nsys.sh │ ├── run_qwen2-7b_seq_balance.sh │ ├── run_qwen2-7b_sglang_seq_balance.sh │ └── run_qwen2.5-32b.sh ├── qwen_image_deployment │ ├── README.md │ ├── client.py │ ├── launcher.py │ └── server.py ├── ray │ └── tutorial.ipynb ├── reinforce_plus_plus_trainer │ ├── run_qwen2-7b_math_rf.sh │ └── run_qwen2-7b_math_rf_baseline.sh ├── remax_trainer │ ├── run_qwen2.5-3b_seq_balance.sh │ └── run_qwen2.5-7b_seq_balance.sh ├── rloo_trainer │ └── run_qwen2-7b.sh ├── sft │ ├── gsm8k │ │ ├── run_deepseek_6b7.sh │ │ ├── run_gemma_2b.sh │ │ ├── run_gemma_7b.sh │ │ ├── run_qwen2_5_05b_sft_peft_sp2_npu.sh │ │ ├── run_qwen_05_peft.sh │ │ ├── run_qwen_05_sp2.sh │ │ └── run_qwen_05_sp2_liger.sh │ └── multiturn │ │ └── run_qwen_05_sp2.sh ├── sglang_multiturn │ ├── README.md │ ├── config │ │ ├── geo3k_multiturn_grpo.yaml │ │ ├── geo3k_multiturn_megatron_grpo.yaml │ │ ├── gsm8k_multiturn_grpo.yaml │ │ ├── gsm8k_multiturn_grpo_w_interaction.yaml │ │ ├── gsm8k_multiturn_megatron_grpo.yaml │ │ ├── interaction_config │ │ │ └── gsm8k_interaction_config.yaml │ │ ├── retool_multiturn_grpo.yaml │ │ ├── search_multiturn_grpo.yaml │ │ └── tool_config │ │ │ ├── geo3k_tool_config.yaml │ │ │ ├── gsm8k_tool_config.yaml │ │ │ ├── mcp_server.json │ │ │ ├── mcp_tool_config.yaml │ │ │ ├── sandbox_fusion_tool_config.yaml │ │ │ └── search_tool_config.yaml │ ├── geo3k │ │ ├── run_qwen2.5-3b_geo3k_multiturn.sh │ │ ├── run_qwen2.5-3b_geo3k_multiturn_4xgpu.sh │ │ └── run_qwen2.5-3b_megatron_geo3k_multiturn.sh │ ├── run_qwen0.5b_gsm8k_multiturn_curriculum.sh │ ├── run_qwen2.5-0.5b_gsm8k_multiturn_w_interaction.sh │ ├── run_qwen2.5-3b_gsm8k_multiturn.sh │ ├── run_qwen2.5-3b_gsm8k_multiturn_4xgpu.sh │ ├── run_qwen2.5-3b_gsm8k_tool_agent_mlflow.sh │ ├── run_qwen2.5-3b_megatron_gsm8k_multiturn.sh │ ├── run_qwen3-4b_gsm8k_multiturn.sh │ └── search_r1_like │ │ ├── local_dense_retriever │ │ ├── download.py │ │ └── retrieval_server.py │ │ └── run_qwen2.5-3b_instruct_search_multiturn.sh ├── slurm │ └── ray_on_slurm.slurm ├── split_placement │ ├── README.md │ ├── config │ │ └── ppo_trainer_split.yaml │ ├── main_ppo_split.py │ ├── run_deepseek7b_llm.sh │ └── split_monkey_patch.py └── tuning │ ├── 0.5b │ └── qwen2-0.5b_grpo-lora_1_h100_fsdp_vllm.sh │ ├── 1.5b │ └── qwen2-1.5b_grpo-lora_1_h100_fsdp_vllm.sh │ ├── 14b │ ├── qwen2-14b_grpo-lora_2_h100_fsdp_vllm.sh │ └── qwen2_14b_grpo_4_h800_fsdp_vllm.sh │ ├── 32b │ ├── qwen2-32b_grpo-lora_4_h100_fsdp_vllm.sh │ └── qwen2_32B_grpo_8_h20_megatron_vllm.sh │ ├── 3b │ └── qwen2-3b_grpo-lora_1_h100_fsdp_vllm.sh │ ├── 70b │ ├── qwen2-70b_grpo_32_h20_fsdp_vllm.sh │ ├── qwen2-70b_grpo_32_h800_fsdp_vllm.sh │ └── qwen2-72b_grpo-lora_8_h100_fsdp_vllm.sh │ └── 7b │ ├── qwen2-7b_grpo-lora_1_h100_fsdp_vllm.sh │ └── qwen2-7b_grpo_2_h800_fsdp_vllm.sh ├── imageutils ├── __init__.py ├── code_exec.py ├── edit_qwen.py ├── edit_seed.py ├── generation_qwen.py ├── generation_seed.py ├── interleaved_gen.py └── search_client.py ├── pyproject.toml ├── recipe └── llmi │ ├── llmi_grpo.sh │ ├── llmi_gspo.sh │ └── mllmi_grpo.sh ├── requirements-npu.txt ├── requirements.txt ├── requirements_llmi.txt ├── requirements_sglang.txt ├── scripts ├── __init__.py ├── converter_hf_to_mcore.py ├── diagnose.py ├── generate_trainer_config.sh ├── init_random_model.py ├── install_vllm_sglang_mcore.sh ├── legacy_model_merger.py └── print_cfg.py ├── setup.py ├── system_prompts └── training │ ├── llmi.txt │ ├── llmjudge.txt │ └── mllmjudge.txt ├── tests ├── README.md ├── __init__.py ├── experimental │ └── agent_loop │ │ ├── agent_utils.py │ │ └── test_basic_agent_loop.py ├── interactions │ ├── __init__.py │ ├── test_gsm8k_interaction.py │ └── test_interaction_registry.py ├── kill_github_tests.sh ├── models │ ├── test_transformer.py │ └── test_transformers_ulysses.py ├── single_controller │ ├── __init__.py │ ├── base │ │ └── test_decorator.py │ ├── check_worker_alive │ │ └── main.py │ ├── detached_worker │ │ ├── README.md │ │ ├── client.py │ │ ├── run.sh │ │ └── server.py │ ├── test_auto_padding_on_cpu.py │ ├── test_colocated_workers.py │ ├── test_colocated_workers_fused.py │ ├── test_data_transfer.py │ ├── test_decorator_on_cpu.py │ ├── test_driverfunc_to_worker.py │ ├── test_fused_workers_on_cpu.py │ ├── test_high_level_scheduling_api.py │ ├── test_ray_collectives.py │ ├── test_ray_local_envs_on_cpu.py │ ├── test_ray_utils_on_cpu.py │ ├── test_rvdz.py │ ├── test_worker_group_basics.py │ └── test_worker_group_torch.py ├── special_distributed │ ├── README.md │ ├── run_all.sh │ ├── test_fsdp_ckpt.py │ └── test_tensor_dict.py ├── special_e2e │ ├── README.md │ ├── __init__.py │ ├── check_custom_rwd_fn.py │ ├── check_results.py │ ├── envs │ │ ├── __init__.py │ │ └── digit_completion │ │ │ ├── __init__.py │ │ │ ├── task.py │ │ │ └── tokenizer.py │ ├── generation │ │ └── run_gen_qwen05.sh │ ├── ppo_trainer │ │ ├── expert_parallel │ │ │ └── qwen2moe_minimal.json │ │ ├── run_function_reward.sh │ │ ├── run_model_reward.sh │ │ └── run_single_gpu.sh │ ├── run_dapo.sh │ ├── run_genrm_remote.sh │ ├── run_geo3k_fsdp_sgl_multiturn_w_tool.sh │ ├── run_grpo_lora_with_merge.sh │ ├── run_gsm8k_fsdp_sgl_multiturn_sf_tool.sh │ ├── run_gsm8k_fsdp_sgl_multiturn_w_tool.sh │ ├── run_one_step_off_policy.sh │ ├── run_ppo_trainer_megatron.sh │ ├── run_prime.sh │ ├── run_r1_distill_qwen_aime24_eval.sh │ ├── run_spin.sh │ ├── run_sppo.sh │ ├── run_test.sh │ └── sft │ │ ├── run_sft.sh │ │ └── test_sp_loss_match.py ├── special_npu │ ├── run_qwen2_5_05b_dapo.sh │ ├── run_qwen2_5_05b_grpo.sh │ ├── run_qwen2_5_05b_sft_peft_sp2.sh │ └── run_qwen2_5_vl_3b_npu.sh ├── special_sanity │ ├── check_api_docs.py │ ├── check_device_api_usage.py │ ├── check_docs_time_info.py │ ├── check_docstrings.py │ ├── check_license.py │ ├── check_pr_description.py │ ├── check_pr_title.py │ ├── test_config_docs.py │ ├── test_import.py │ ├── type_coverage_check.py │ ├── validate_imported_docs.py │ └── validate_structure.py ├── special_standalone │ ├── README.md │ └── test_memory_buffers.py ├── test_base_config_on_cpu.py ├── test_protocol_on_cpu.py ├── tools │ └── test_base_tool_on_cpu.py ├── trainer │ ├── __init__.py │ ├── config │ │ ├── __init__.py │ │ ├── legacy_ppo_megatron_trainer.yaml │ │ ├── legacy_ppo_trainer.yaml │ │ ├── test_algo_config_on_cpu.py │ │ ├── test_critic_config_on_cpu.py │ │ └── test_legacy_config_on_cpu.py │ └── ppo │ │ ├── __init__.py │ │ ├── test_core_algos_on_cpu.py │ │ └── test_metric_utils_on_cpu.py ├── utils │ ├── _test_module.py │ ├── ckpt │ │ └── test_esi_save_ckpt_on_cpu.py │ ├── dataset │ │ ├── test_create_rl_sampler_on_cpu.py │ │ ├── test_multiturn_sft_dataset_on_cpu.py │ │ ├── test_rl_dataset_on_cpu.py │ │ └── test_sft_dataset_on_cpu.py │ ├── megatron │ │ └── test_pipeline_parallel.py │ ├── reward_score │ │ ├── reward_score │ │ │ └── test_sandbox_fusion_on_cpu.py │ │ └── test_sandbox_on_cpu.py │ ├── test_activation_offload.py │ ├── test_config_on_cpu.py │ ├── test_flops_counter.py │ ├── test_fs_on_cpu.py │ ├── test_import_utils_on_cpu.py │ ├── test_linear_cross_entropy.py │ ├── test_linear_cross_entropy_tp.py │ ├── test_model_on_cpu.py │ ├── test_nvtx_profile.py │ ├── test_rollout_trace_on_cpu.py │ ├── test_seqlen_balancing.py │ ├── test_temp_env_on_cpu.py │ ├── test_timeout_decorator_cpu.py │ └── test_torch_functional.py └── workers │ ├── reward_manager │ └── test_registry_on_cpu.py │ └── rollout │ ├── async_rollout_utils.py │ ├── perf │ └── vllm_async_rollout.py │ ├── resource │ └── tool_configs │ │ ├── mcp_server.json │ │ ├── mcp_tool_config │ │ ├── sandbox_fusion_tool_config │ │ └── search_tool_config │ ├── rollout_vllm │ ├── run_fsdp_vllm.py │ ├── test_vllm_chat_scheduler.py │ ├── test_vllm_model_rope_scaling.py │ └── test_vllm_spmd.py │ ├── test_async_sglang_server.py │ ├── test_custom_completion_callback.py │ ├── test_hf_rollout.py │ ├── test_sglang_async_rollout_mcp_tools.py │ ├── test_sglang_async_rollout_multimodal_delta.py │ ├── test_sglang_async_rollout_search_tools.py │ ├── test_sglang_async_rollout_sf_tools.py │ ├── test_sglang_async_rollout_w_interaction.py │ ├── test_sglang_async_rollout_w_tools.py │ ├── test_sglang_multi_interaction.py │ ├── test_sglang_rollout_sharding_manager.py │ ├── test_sglang_spmd.py │ └── utils_sglang.py └── verl ├── __init__.py ├── base_config.py ├── experimental ├── __init__.py ├── agent_loop │ ├── __init__.py │ ├── agent_loop.py │ ├── single_turn_agent_loop.py │ ├── tool_agent_loop.py │ └── tool_parser.py ├── dataset │ ├── __init__.py │ └── sampler.py └── dynamic_dataset │ ├── __init__.py │ └── dynamicgen_dataset.py ├── interactions ├── __init__.py ├── base.py ├── gsm8k_interaction.py └── utils │ ├── __init__.py │ └── interaction_registry.py ├── model_merger ├── __init__.py ├── __main__.py ├── base_model_merger.py ├── fsdp_model_merger.py └── megatron_model_merger.py ├── models ├── README.md ├── __init__.py ├── llama │ ├── __init__.py │ └── megatron │ │ ├── __init__.py │ │ ├── checkpoint_utils │ │ ├── __init__.py │ │ ├── llama_loader.py │ │ ├── llama_loader_depracated.py │ │ └── llama_saver.py │ │ ├── layers │ │ ├── __init__.py │ │ ├── parallel_attention.py │ │ ├── parallel_decoder.py │ │ ├── parallel_linear.py │ │ ├── parallel_mlp.py │ │ └── parallel_rmsnorm.py │ │ └── modeling_llama_megatron.py ├── mcore │ ├── __init__.py │ ├── config_converter.py │ ├── loader.py │ ├── mbridge.py │ ├── model_forward.py │ ├── model_forward_fused.py │ ├── model_initializer.py │ ├── patch_v012.py │ ├── qwen2_5_vl │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── model.py │ │ ├── rope_utils.py │ │ ├── vision_config.py │ │ ├── vision_model.py │ │ └── vision_transformer_block.py │ ├── readme.md │ ├── registry.py │ ├── saver.py │ ├── util.py │ └── weight_converter.py ├── qwen2 │ ├── __init__.py │ └── megatron │ │ ├── __init__.py │ │ ├── checkpoint_utils │ │ ├── __init__.py │ │ ├── qwen2_loader.py │ │ ├── qwen2_loader_depracated.py │ │ └── qwen2_saver.py │ │ ├── layers │ │ ├── __init__.py │ │ ├── parallel_attention.py │ │ ├── parallel_decoder.py │ │ ├── parallel_linear.py │ │ ├── parallel_mlp.py │ │ └── parallel_rmsnorm.py │ │ └── modeling_qwen2_megatron.py ├── registry.py ├── transformers │ ├── __init__.py │ ├── dense_common.py │ ├── kimi_vl.py │ ├── llama.py │ ├── monkey_patch.py │ ├── npu_patch.py │ ├── qwen2.py │ ├── qwen2_5_vl.py │ └── qwen2_vl.py └── weight_loader_registry.py ├── protocol.py ├── py.typed ├── single_controller ├── __init__.py ├── base │ ├── __init__.py │ ├── decorator.py │ ├── megatron │ │ ├── __init__.py │ │ ├── worker.py │ │ └── worker_group.py │ ├── register_center │ │ ├── __init__.py │ │ └── ray.py │ ├── worker.py │ └── worker_group.py └── ray │ ├── __init__.py │ ├── base.py │ └── megatron.py ├── third_party ├── __init__.py ├── sglang │ ├── __init__.py │ └── parallel_state.py └── vllm │ └── __init__.py ├── tools ├── __init__.py ├── base_tool.py ├── geo3k_tool.py ├── gsm8k_tool.py ├── mcp_base_tool.py ├── mcp_search_tool.py ├── sandbox_fusion_tools.py ├── schemas.py ├── search_tool.py └── utils │ ├── __init__.py │ ├── mcp_clients │ ├── McpClientManager.py │ └── utils.py │ ├── search_r1_like_utils.py │ └── tool_registry.py ├── trainer ├── __init__.py ├── config │ ├── __init__.py │ ├── _generated_ppo_trainer.yaml │ ├── actor │ │ ├── actor.yaml │ │ ├── dp_actor.yaml │ │ └── megatron_actor.yaml │ ├── algorithm.py │ ├── config.py │ ├── critic │ │ ├── critic.yaml │ │ ├── dp_critic.yaml │ │ └── megatron_critic.yaml │ ├── data │ │ └── legacy_data.yaml │ ├── evaluation.yaml │ ├── generation.yaml │ ├── npu_profile │ │ └── npu_profile.yaml │ ├── ppo_megatron_trainer.yaml │ ├── ppo_trainer.yaml │ ├── ref │ │ ├── dp_ref.yaml │ │ ├── megatron_ref.yaml │ │ └── ref.yaml │ ├── reward_model │ │ ├── dp_reward_model.yaml │ │ ├── megatron_reward_model.yaml │ │ └── reward_model.yaml │ ├── rollout │ │ └── rollout.yaml │ └── sft_trainer.yaml ├── constants_ppo.py ├── fsdp_sft_trainer.py ├── main_eval.py ├── main_generation.py ├── main_ppo.py ├── ppo │ ├── __init__.py │ ├── core_algos.py │ ├── metric_utils.py │ ├── ray_trainer.py │ └── reward.py └── runtime_env.yaml ├── utils ├── __init__.py ├── activation_offload.py ├── checkpoint │ ├── __init__.py │ ├── checkpoint_manager.py │ ├── fsdp_checkpoint_manager.py │ └── megatron_checkpoint_manager.py ├── config.py ├── dataset │ ├── README.md │ ├── __init__.py │ ├── multiturn_sft_dataset.py │ ├── rl_dataset.py │ ├── rm_dataset.py │ ├── sft_dataset.py │ └── vision_utils.py ├── debug │ ├── __init__.py │ ├── performance.py │ └── trajectory_tracker.py ├── device.py ├── distributed.py ├── experimental │ ├── __init__.py │ └── torch_functional.py ├── flops_counter.py ├── fs.py ├── fsdp_utils.py ├── hdfs_io.py ├── import_utils.py ├── kernel │ ├── __init__.py │ ├── kernels.py │ └── linear_cross_entropy.py ├── logger │ ├── __init__.py │ └── aggregate_logger.py ├── logging_utils.py ├── megatron │ ├── __init__.py │ ├── dist_checkpointing.py │ ├── memory.py │ ├── optimizer.py │ ├── pipeline_parallel.py │ ├── sequence_parallel.py │ └── tensor_parallel.py ├── megatron_utils.py ├── memory_buffer.py ├── metric │ ├── __init__.py │ └── utils.py ├── model.py ├── net_utils.py ├── profiler │ ├── __init__.py │ ├── config.py │ ├── empty_annotations.py │ ├── mstx_profile.py │ ├── nvtx_profile.py │ ├── performance.py │ └── profile.py ├── py_functional.py ├── ray_utils.py ├── rendezvous │ ├── __init__.py │ └── ray_backend.py ├── reward_score │ ├── __init__.py │ ├── geo3k.py │ ├── gsm8k.py │ ├── judge_client.py │ ├── llmi_rm.py │ ├── math.py │ ├── math_batch.py │ ├── math_dapo.py │ ├── math_verify.py │ ├── prime_code │ │ ├── README.md │ │ ├── __init__.py │ │ ├── testing_util.py │ │ └── utils.py │ ├── prime_math │ │ ├── __init__.py │ │ ├── grader.py │ │ └── math_normalize.py │ ├── sandbox_fusion │ │ ├── __init__.py │ │ └── utils.py │ └── search_r1_like_qa_em.py ├── rollout_trace.py ├── seqlen_balancing.py ├── tokenizer.py ├── torch_dtypes.py ├── torch_functional.py ├── tracking.py ├── ulysses.py └── vllm_utils.py ├── version └── version └── workers ├── __init__.py ├── actor ├── __init__.py ├── base.py ├── dp_actor.py └── megatron_actor.py ├── critic ├── __init__.py ├── base.py ├── dp_critic.py └── megatron_critic.py ├── fsdp_workers.py ├── megatron_workers.py ├── reward_manager ├── __init__.py ├── batch.py ├── dapo.py ├── llmi.py ├── naive.py ├── prime.py └── registry.py ├── reward_model ├── __init__.py ├── base.py └── megatron │ ├── __init__.py │ └── reward_model.py ├── rollout ├── __init__.py ├── async_server.py ├── base.py ├── chat_scheduler.py ├── hf_rollout.py ├── naive │ ├── __init__.py │ └── naive_rollout.py ├── schemas.py ├── sglang_rollout │ ├── __init__.py │ ├── async_sglang_server.py │ ├── sglang_rollout.py │ └── utils.py ├── tokenizer.py └── vllm_rollout │ ├── __init__.py │ ├── vllm_async_server.py │ └── vllm_rollout_spmd.py └── sharding_manager ├── __init__.py ├── base.py ├── fsdp_sglang.py ├── fsdp_ulysses.py ├── fsdp_vllm.py ├── megatron_sglang.py └── megatron_vllm.py /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/README.md -------------------------------------------------------------------------------- /.github/workflows/check-pr-title.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/check-pr-title.yml -------------------------------------------------------------------------------- /.github/workflows/checkpoint_converter.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/checkpoint_converter.yml -------------------------------------------------------------------------------- /.github/workflows/cpu_unit_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/cpu_unit_tests.yml -------------------------------------------------------------------------------- /.github/workflows/disabled/e2e_prime.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/disabled/e2e_prime.yml -------------------------------------------------------------------------------- /.github/workflows/doc.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/doc.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_ascend.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_ascend.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_dapo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_dapo.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_eval_aime24.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_eval_aime24.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_genrm_remote.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_genrm_remote.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_one_step_off_policy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_one_step_off_policy.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_ppo_trainer.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_ppo_trainer.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_ppo_trainer_megatron_sglang.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_ppo_trainer_megatron_sglang.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_ppo_trainer_megatron_vllm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_ppo_trainer_megatron_vllm.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_sft.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_sft.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_spin.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_spin.yml -------------------------------------------------------------------------------- /.github/workflows/e2e_sppo.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/e2e_sppo.yml -------------------------------------------------------------------------------- /.github/workflows/gpu_unit_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/gpu_unit_tests.yml -------------------------------------------------------------------------------- /.github/workflows/model.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/model.yml -------------------------------------------------------------------------------- /.github/workflows/pre-commit-full.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/pre-commit-full.yml -------------------------------------------------------------------------------- /.github/workflows/pre-commit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/pre-commit.yml -------------------------------------------------------------------------------- /.github/workflows/sanity.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/sanity.yml -------------------------------------------------------------------------------- /.github/workflows/scorecard.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/scorecard.yml -------------------------------------------------------------------------------- /.github/workflows/secrets_scan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/secrets_scan.yml -------------------------------------------------------------------------------- /.github/workflows/sgl.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/sgl.yml -------------------------------------------------------------------------------- /.github/workflows/type-coverage-check.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/type-coverage-check.yml -------------------------------------------------------------------------------- /.github/workflows/vllm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.github/workflows/vllm.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.gitignore -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/LICENSE -------------------------------------------------------------------------------- /Notice.txt: -------------------------------------------------------------------------------- 1 | Copyright 2024-2025 Bytedance Ltd. and/or its affiliates -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/README.md -------------------------------------------------------------------------------- /docker/Apptainerfile.rocm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Apptainerfile.rocm -------------------------------------------------------------------------------- /docker/Dockerfile.extention.awsefa: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Dockerfile.extention.awsefa -------------------------------------------------------------------------------- /docker/Dockerfile.ngc.vllm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Dockerfile.ngc.vllm -------------------------------------------------------------------------------- /docker/Dockerfile.ngc.vllm0.8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Dockerfile.ngc.vllm0.8 -------------------------------------------------------------------------------- /docker/Dockerfile.ngc.vllm0.8.sagemaker: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Dockerfile.ngc.vllm0.8.sagemaker -------------------------------------------------------------------------------- /docker/Dockerfile.rocm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Dockerfile.rocm -------------------------------------------------------------------------------- /docker/Dockerfile.rocm_verl-0.3.0.post1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Dockerfile.rocm_verl-0.3.0.post1 -------------------------------------------------------------------------------- /docker/Dockerfile.rocm_verl-0.4.1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Dockerfile.rocm_verl-0.4.1 -------------------------------------------------------------------------------- /docker/Dockerfile.sglang: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Dockerfile.sglang -------------------------------------------------------------------------------- /docker/Dockerfile.vemlp.vllm.te: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Dockerfile.vemlp.vllm.te -------------------------------------------------------------------------------- /docker/Dockerfile.vllm.sglang.megatron.deepseek: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/Dockerfile.vllm.sglang.megatron.deepseek -------------------------------------------------------------------------------- /docker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/README.md -------------------------------------------------------------------------------- /docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.sglang.vllm.mcore0.12: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.sglang.vllm.mcore0.12 -------------------------------------------------------------------------------- /docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.sglang.vllm.mcore0.12.deepep: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.sglang.vllm.mcore0.12.deepep -------------------------------------------------------------------------------- /docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.sglang.vllm.mcore0.13.preview: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.sglang.vllm.mcore0.13.preview -------------------------------------------------------------------------------- /docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.vllm.mcore0.12: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.vllm.mcore0.12 -------------------------------------------------------------------------------- /docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.vllm.mcore0.12.deepep: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.vllm.mcore0.12.deepep -------------------------------------------------------------------------------- /docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.vllm.mcore0.13.preview: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.app.vllm.mcore0.13.preview -------------------------------------------------------------------------------- /docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.base: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.4-cu124-torch2.6-fa2.7.4/Dockerfile.base -------------------------------------------------------------------------------- /docker/verl0.4-cu124-torch2.6-fa2.7.4/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.4-cu124-torch2.6-fa2.7.4/README.md -------------------------------------------------------------------------------- /docker/verl0.5-cu126-torch2.7.1-fa2.8.0/Dockerfile.app.sglang.mcore0.12: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.5-cu126-torch2.7.1-fa2.8.0/Dockerfile.app.sglang.mcore0.12 -------------------------------------------------------------------------------- /docker/verl0.5-cu126-torch2.7.1-fa2.8.0/Dockerfile.app.sglang.mcore0.13.preview: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.5-cu126-torch2.7.1-fa2.8.0/Dockerfile.app.sglang.mcore0.13.preview -------------------------------------------------------------------------------- /docker/verl0.5-cu126-torch2.7.1-fa2.8.0/Dockerfile.base: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.5-cu126-torch2.7.1-fa2.8.0/Dockerfile.base -------------------------------------------------------------------------------- /docker/verl0.5-cu126-torch2.7.1-fa2.8.0/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.5-cu126-torch2.7.1-fa2.8.0/README.md -------------------------------------------------------------------------------- /docker/verl0.5-preview-cu128-torch2.7.1-fa2.8.0/Dockerfile.app.sglang.megatron: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.5-preview-cu128-torch2.7.1-fa2.8.0/Dockerfile.app.sglang.megatron -------------------------------------------------------------------------------- /docker/verl0.5-preview-cu128-torch2.7.1-fa2.8.0/Dockerfile.base: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.5-preview-cu128-torch2.7.1-fa2.8.0/Dockerfile.base -------------------------------------------------------------------------------- /docker/verl0.5-preview-cu128-torch2.7.1-fa2.8.0/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docker/verl0.5-preview-cu128-torch2.7.1-fa2.8.0/README.md -------------------------------------------------------------------------------- /docs/BandAI.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/BandAI.png -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/README_vllm0.7.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/README_vllm0.7.md -------------------------------------------------------------------------------- /docs/README_vllm0.8.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/README_vllm0.8.md -------------------------------------------------------------------------------- /docs/_static/js/runllm-widget.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/_static/js/runllm-widget.js -------------------------------------------------------------------------------- /docs/_static/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/_static/logo.png -------------------------------------------------------------------------------- /docs/advance/checkpoint.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/advance/checkpoint.rst -------------------------------------------------------------------------------- /docs/advance/dpo_extension.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/advance/dpo_extension.rst -------------------------------------------------------------------------------- /docs/advance/fsdp_extension.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/advance/fsdp_extension.rst -------------------------------------------------------------------------------- /docs/advance/megatron_extension.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/advance/megatron_extension.rst -------------------------------------------------------------------------------- /docs/advance/one_step_off.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/advance/one_step_off.md -------------------------------------------------------------------------------- /docs/advance/placement.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/advance/placement.rst -------------------------------------------------------------------------------- /docs/advance/ppo_lora.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/advance/ppo_lora.rst -------------------------------------------------------------------------------- /docs/advance/rollout_trace.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/advance/rollout_trace.rst -------------------------------------------------------------------------------- /docs/advance/rope.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/advance/rope.rst -------------------------------------------------------------------------------- /docs/algo/baseline.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/algo/baseline.md -------------------------------------------------------------------------------- /docs/algo/dapo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/algo/dapo.md -------------------------------------------------------------------------------- /docs/algo/entropy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/algo/entropy.md -------------------------------------------------------------------------------- /docs/algo/gpg.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/algo/gpg.md -------------------------------------------------------------------------------- /docs/algo/grpo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/algo/grpo.md -------------------------------------------------------------------------------- /docs/algo/opo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/algo/opo.md -------------------------------------------------------------------------------- /docs/algo/ppo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/algo/ppo.md -------------------------------------------------------------------------------- /docs/algo/spin.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/algo/spin.md -------------------------------------------------------------------------------- /docs/algo/sppo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/algo/sppo.md -------------------------------------------------------------------------------- /docs/amd_tutorial/amd_build_dockerfile_page.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/amd_tutorial/amd_build_dockerfile_page.rst -------------------------------------------------------------------------------- /docs/amd_tutorial/amd_vllm_page.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/amd_tutorial/amd_vllm_page.rst -------------------------------------------------------------------------------- /docs/api/data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/api/data.rst -------------------------------------------------------------------------------- /docs/api/single_controller.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/api/single_controller.rst -------------------------------------------------------------------------------- /docs/api/trainer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/api/trainer.rst -------------------------------------------------------------------------------- /docs/api/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/api/utils.rst -------------------------------------------------------------------------------- /docs/ascend_tutorial/ascend_profiling.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/ascend_tutorial/ascend_profiling.rst -------------------------------------------------------------------------------- /docs/ascend_tutorial/ascend_profiling_en.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/ascend_tutorial/ascend_profiling_en.rst -------------------------------------------------------------------------------- /docs/ascend_tutorial/ascend_quick_start.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/ascend_tutorial/ascend_quick_start.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/examples/config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/examples/config.rst -------------------------------------------------------------------------------- /docs/examples/gsm8k_example.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/examples/gsm8k_example.rst -------------------------------------------------------------------------------- /docs/examples/multi_modal_example.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/examples/multi_modal_example.rst -------------------------------------------------------------------------------- /docs/examples/ppo_code_architecture.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/examples/ppo_code_architecture.rst -------------------------------------------------------------------------------- /docs/examples/sandbox_fusion_example.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/examples/sandbox_fusion_example.rst -------------------------------------------------------------------------------- /docs/faq/faq.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/faq/faq.rst -------------------------------------------------------------------------------- /docs/hybrid_flow.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/hybrid_flow.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/llm-example1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/llm-example1.png -------------------------------------------------------------------------------- /docs/overall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/overall.png -------------------------------------------------------------------------------- /docs/perf/device_tuning.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/perf/device_tuning.rst -------------------------------------------------------------------------------- /docs/perf/dpsk.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/perf/dpsk.md -------------------------------------------------------------------------------- /docs/perf/nsight_profiling.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/perf/nsight_profiling.md -------------------------------------------------------------------------------- /docs/perf/perf_tuning.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/perf/perf_tuning.rst -------------------------------------------------------------------------------- /docs/preparation/prepare_data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/preparation/prepare_data.rst -------------------------------------------------------------------------------- /docs/preparation/reward_function.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/preparation/reward_function.rst -------------------------------------------------------------------------------- /docs/requirements-docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/requirements-docs.txt -------------------------------------------------------------------------------- /docs/sglang_multiturn/interaction_system.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/sglang_multiturn/interaction_system.rst -------------------------------------------------------------------------------- /docs/sglang_multiturn/multiturn.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/sglang_multiturn/multiturn.rst -------------------------------------------------------------------------------- /docs/sglang_multiturn/sandbox_fusion.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/sglang_multiturn/sandbox_fusion.rst -------------------------------------------------------------------------------- /docs/sglang_multiturn/search_tool_example.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/sglang_multiturn/search_tool_example.rst -------------------------------------------------------------------------------- /docs/single_controller.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/single_controller.rst -------------------------------------------------------------------------------- /docs/start/agentic_rl.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/start/agentic_rl.rst -------------------------------------------------------------------------------- /docs/start/install.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/start/install.rst -------------------------------------------------------------------------------- /docs/start/more_resources.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/start/more_resources.rst -------------------------------------------------------------------------------- /docs/start/multinode.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/start/multinode.rst -------------------------------------------------------------------------------- /docs/start/quickstart.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/start/quickstart.rst -------------------------------------------------------------------------------- /docs/start/ray_debug_tutorial.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/start/ray_debug_tutorial.rst -------------------------------------------------------------------------------- /docs/workers/fsdp_workers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/workers/fsdp_workers.rst -------------------------------------------------------------------------------- /docs/workers/megatron_workers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/workers/megatron_workers.rst -------------------------------------------------------------------------------- /docs/workers/ray_trainer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/workers/ray_trainer.rst -------------------------------------------------------------------------------- /docs/workers/sglang_worker.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/docs/workers/sglang_worker.rst -------------------------------------------------------------------------------- /evaluation/eval_mm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/evaluation/eval_mm.py -------------------------------------------------------------------------------- /evaluation/eval_text_only.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/evaluation/eval_text_only.py -------------------------------------------------------------------------------- /evaluation/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/evaluation/inference.py -------------------------------------------------------------------------------- /examples/data_preprocess/aime2024_multiturn_w_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/aime2024_multiturn_w_tool.py -------------------------------------------------------------------------------- /examples/data_preprocess/dapo_multiturn_w_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/dapo_multiturn_w_tool.py -------------------------------------------------------------------------------- /examples/data_preprocess/full_hh_rlhf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/full_hh_rlhf.py -------------------------------------------------------------------------------- /examples/data_preprocess/geo3k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/geo3k.py -------------------------------------------------------------------------------- /examples/data_preprocess/geo3k_multiturn_w_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/geo3k_multiturn_w_tool.py -------------------------------------------------------------------------------- /examples/data_preprocess/gsm8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/gsm8k.py -------------------------------------------------------------------------------- /examples/data_preprocess/gsm8k_multiturn_w_interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/gsm8k_multiturn_w_interaction.py -------------------------------------------------------------------------------- /examples/data_preprocess/gsm8k_multiturn_w_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/gsm8k_multiturn_w_tool.py -------------------------------------------------------------------------------- /examples/data_preprocess/gsm8k_tool_agent_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/gsm8k_tool_agent_loop.py -------------------------------------------------------------------------------- /examples/data_preprocess/hellaswag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/hellaswag.py -------------------------------------------------------------------------------- /examples/data_preprocess/math_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/math_dataset.py -------------------------------------------------------------------------------- /examples/data_preprocess/multiturn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/multiturn.py -------------------------------------------------------------------------------- /examples/data_preprocess/preprocess_search_r1_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/data_preprocess/preprocess_search_r1_dataset.py -------------------------------------------------------------------------------- /examples/generation/run_deepseek7b_mutli_node.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/generation/run_deepseek7b_mutli_node.sh -------------------------------------------------------------------------------- /examples/generation/run_deepseek_v2_lite_math.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/generation/run_deepseek_v2_lite_math.sh -------------------------------------------------------------------------------- /examples/gpg_trainer/gpg.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/gpg_trainer/gpg.md -------------------------------------------------------------------------------- /examples/gpg_trainer/run_qwen2-7b_math.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/gpg_trainer/run_qwen2-7b_math.sh -------------------------------------------------------------------------------- /examples/gpg_trainer/run_qwen2-7b_math_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/gpg_trainer/run_qwen2-7b_math_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/README.md -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek671b_math_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_deepseek671b_math_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek7b_llm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_deepseek7b_llm.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek7b_llm_math.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_deepseek7b_llm_math.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek7b_llm_math_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_deepseek7b_llm_math_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek7b_llm_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_deepseek7b_llm_seq_balance.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_minicpmo2_6.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_minicpmo2_6.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_moonlight16b_math_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_moonlight16b_math_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2-7b.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b_math.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2-7b_math.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b_math_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2-7b_math_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2-7b_seq_balance.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b_seq_balance_math_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2-7b_seq_balance_math_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b_sgl_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2-7b_sgl_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5-3b_gsm8k_grpo_lora.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5-3b_gsm8k_grpo_lora.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5-7b_math_megatron_diff_tp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5-7b_math_megatron_diff_tp.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_32b_grpo_npu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_32b_grpo_npu.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_7b_grpo_discrete_prof_npu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_7b_grpo_discrete_prof_npu.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_7b_grpo_e2e_prof_npu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_7b_grpo_e2e_prof_npu.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_7b_grpo_npu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_7b_grpo_npu.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_vl-7b-megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_vl-7b-megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_vl-7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_vl-7b.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_vl-7b_lora.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_vl-7b_lora.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_vl-7b_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_vl-7b_seq_balance.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_vl_32b_npu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_vl_32b_npu.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_vl_3b_npu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_vl_3b_npu.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_vl_7b_npu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen2_5_vl_7b_npu.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen3-236b_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen3-236b_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen3-8b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen3-8b.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen3moe-30b_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/grpo_trainer/run_qwen3moe-30b_megatron.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/README.md -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_deepseek7b_llm.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm_modelscope.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_deepseek7b_llm_modelscope.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm_pfppo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_deepseek7b_llm_pfppo.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm_sandbox_fusion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_deepseek7b_llm_sandbox_fusion.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm_sp2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_deepseek7b_llm_sp2.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek_full_hh_rlhf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_deepseek_full_hh_rlhf.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek_math_gsm8k_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_deepseek_math_gsm8k_megatron.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek_math_gsm8k_megatron_nsys.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_deepseek_math_gsm8k_megatron_nsys.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_gemma.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_gemma.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_moonlight16b_a3b_gsm8k_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_moonlight16b_a3b_gsm8k_megatron.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen1.5_moe_a2.7b-gsm8k_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_qwen1.5_moe_a2.7b-gsm8k_megatron.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_math_gsm8k_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_qwen2-7b_math_gsm8k_megatron.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_rm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_qwen2-7b_rm.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_rm_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_qwen2-7b_rm_seq_balance.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_rm_seq_balance_fused_kernels.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_qwen2-7b_rm_seq_balance_fused_kernels.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_rm_seq_balance_nsys.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_qwen2-7b_rm_seq_balance_nsys.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_qwen2-7b_seq_balance.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_sglang_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_qwen2-7b_sglang_seq_balance.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2.5-32b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ppo_trainer/run_qwen2.5-32b.sh -------------------------------------------------------------------------------- /examples/qwen_image_deployment/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/qwen_image_deployment/README.md -------------------------------------------------------------------------------- /examples/qwen_image_deployment/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/qwen_image_deployment/client.py -------------------------------------------------------------------------------- /examples/qwen_image_deployment/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/qwen_image_deployment/launcher.py -------------------------------------------------------------------------------- /examples/qwen_image_deployment/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/qwen_image_deployment/server.py -------------------------------------------------------------------------------- /examples/ray/tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/ray/tutorial.ipynb -------------------------------------------------------------------------------- /examples/reinforce_plus_plus_trainer/run_qwen2-7b_math_rf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/reinforce_plus_plus_trainer/run_qwen2-7b_math_rf.sh -------------------------------------------------------------------------------- /examples/reinforce_plus_plus_trainer/run_qwen2-7b_math_rf_baseline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/reinforce_plus_plus_trainer/run_qwen2-7b_math_rf_baseline.sh -------------------------------------------------------------------------------- /examples/remax_trainer/run_qwen2.5-3b_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/remax_trainer/run_qwen2.5-3b_seq_balance.sh -------------------------------------------------------------------------------- /examples/remax_trainer/run_qwen2.5-7b_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/remax_trainer/run_qwen2.5-7b_seq_balance.sh -------------------------------------------------------------------------------- /examples/rloo_trainer/run_qwen2-7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/rloo_trainer/run_qwen2-7b.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_deepseek_6b7.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sft/gsm8k/run_deepseek_6b7.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_gemma_2b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sft/gsm8k/run_gemma_2b.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_gemma_7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sft/gsm8k/run_gemma_7b.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_qwen2_5_05b_sft_peft_sp2_npu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sft/gsm8k/run_qwen2_5_05b_sft_peft_sp2_npu.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_qwen_05_peft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sft/gsm8k/run_qwen_05_peft.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_qwen_05_sp2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sft/gsm8k/run_qwen_05_sp2.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_qwen_05_sp2_liger.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sft/gsm8k/run_qwen_05_sp2_liger.sh -------------------------------------------------------------------------------- /examples/sft/multiturn/run_qwen_05_sp2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sft/multiturn/run_qwen_05_sp2.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/README.md -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/geo3k_multiturn_grpo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/geo3k_multiturn_grpo.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/geo3k_multiturn_megatron_grpo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/geo3k_multiturn_megatron_grpo.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/gsm8k_multiturn_grpo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/gsm8k_multiturn_grpo.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/gsm8k_multiturn_grpo_w_interaction.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/gsm8k_multiturn_grpo_w_interaction.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/gsm8k_multiturn_megatron_grpo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/gsm8k_multiturn_megatron_grpo.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/interaction_config/gsm8k_interaction_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/interaction_config/gsm8k_interaction_config.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/retool_multiturn_grpo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/retool_multiturn_grpo.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/search_multiturn_grpo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/search_multiturn_grpo.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/tool_config/geo3k_tool_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/tool_config/geo3k_tool_config.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/tool_config/gsm8k_tool_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/tool_config/gsm8k_tool_config.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/tool_config/mcp_server.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/tool_config/mcp_server.json -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/tool_config/mcp_tool_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/tool_config/mcp_tool_config.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/tool_config/sandbox_fusion_tool_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/tool_config/sandbox_fusion_tool_config.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/tool_config/search_tool_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/config/tool_config/search_tool_config.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/geo3k/run_qwen2.5-3b_geo3k_multiturn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/geo3k/run_qwen2.5-3b_geo3k_multiturn.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/geo3k/run_qwen2.5-3b_geo3k_multiturn_4xgpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/geo3k/run_qwen2.5-3b_geo3k_multiturn_4xgpu.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/geo3k/run_qwen2.5-3b_megatron_geo3k_multiturn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/geo3k/run_qwen2.5-3b_megatron_geo3k_multiturn.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/run_qwen0.5b_gsm8k_multiturn_curriculum.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/run_qwen0.5b_gsm8k_multiturn_curriculum.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/run_qwen2.5-0.5b_gsm8k_multiturn_w_interaction.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/run_qwen2.5-0.5b_gsm8k_multiturn_w_interaction.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/run_qwen2.5-3b_gsm8k_multiturn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/run_qwen2.5-3b_gsm8k_multiturn.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/run_qwen2.5-3b_gsm8k_multiturn_4xgpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/run_qwen2.5-3b_gsm8k_multiturn_4xgpu.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/run_qwen2.5-3b_gsm8k_tool_agent_mlflow.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/run_qwen2.5-3b_gsm8k_tool_agent_mlflow.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/run_qwen2.5-3b_megatron_gsm8k_multiturn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/run_qwen2.5-3b_megatron_gsm8k_multiturn.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/run_qwen3-4b_gsm8k_multiturn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/run_qwen3-4b_gsm8k_multiturn.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/search_r1_like/local_dense_retriever/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/search_r1_like/local_dense_retriever/download.py -------------------------------------------------------------------------------- /examples/sglang_multiturn/search_r1_like/local_dense_retriever/retrieval_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/search_r1_like/local_dense_retriever/retrieval_server.py -------------------------------------------------------------------------------- /examples/sglang_multiturn/search_r1_like/run_qwen2.5-3b_instruct_search_multiturn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/sglang_multiturn/search_r1_like/run_qwen2.5-3b_instruct_search_multiturn.sh -------------------------------------------------------------------------------- /examples/slurm/ray_on_slurm.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/slurm/ray_on_slurm.slurm -------------------------------------------------------------------------------- /examples/split_placement/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/split_placement/README.md -------------------------------------------------------------------------------- /examples/split_placement/config/ppo_trainer_split.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/split_placement/config/ppo_trainer_split.yaml -------------------------------------------------------------------------------- /examples/split_placement/main_ppo_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/split_placement/main_ppo_split.py -------------------------------------------------------------------------------- /examples/split_placement/run_deepseek7b_llm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/split_placement/run_deepseek7b_llm.sh -------------------------------------------------------------------------------- /examples/split_placement/split_monkey_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/split_placement/split_monkey_patch.py -------------------------------------------------------------------------------- /examples/tuning/0.5b/qwen2-0.5b_grpo-lora_1_h100_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/0.5b/qwen2-0.5b_grpo-lora_1_h100_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/1.5b/qwen2-1.5b_grpo-lora_1_h100_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/1.5b/qwen2-1.5b_grpo-lora_1_h100_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/14b/qwen2-14b_grpo-lora_2_h100_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/14b/qwen2-14b_grpo-lora_2_h100_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/14b/qwen2_14b_grpo_4_h800_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/14b/qwen2_14b_grpo_4_h800_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/32b/qwen2-32b_grpo-lora_4_h100_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/32b/qwen2-32b_grpo-lora_4_h100_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/32b/qwen2_32B_grpo_8_h20_megatron_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/32b/qwen2_32B_grpo_8_h20_megatron_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/3b/qwen2-3b_grpo-lora_1_h100_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/3b/qwen2-3b_grpo-lora_1_h100_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/70b/qwen2-70b_grpo_32_h20_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/70b/qwen2-70b_grpo_32_h20_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/70b/qwen2-70b_grpo_32_h800_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/70b/qwen2-70b_grpo_32_h800_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/70b/qwen2-72b_grpo-lora_8_h100_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/70b/qwen2-72b_grpo-lora_8_h100_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/7b/qwen2-7b_grpo-lora_1_h100_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/7b/qwen2-7b_grpo-lora_1_h100_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/7b/qwen2-7b_grpo_2_h800_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/examples/tuning/7b/qwen2-7b_grpo_2_h800_fsdp_vllm.sh -------------------------------------------------------------------------------- /imageutils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imageutils/code_exec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/imageutils/code_exec.py -------------------------------------------------------------------------------- /imageutils/edit_qwen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/imageutils/edit_qwen.py -------------------------------------------------------------------------------- /imageutils/edit_seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/imageutils/edit_seed.py -------------------------------------------------------------------------------- /imageutils/generation_qwen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/imageutils/generation_qwen.py -------------------------------------------------------------------------------- /imageutils/generation_seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/imageutils/generation_seed.py -------------------------------------------------------------------------------- /imageutils/interleaved_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/imageutils/interleaved_gen.py -------------------------------------------------------------------------------- /imageutils/search_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/imageutils/search_client.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/pyproject.toml -------------------------------------------------------------------------------- /recipe/llmi/llmi_grpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/recipe/llmi/llmi_grpo.sh -------------------------------------------------------------------------------- /recipe/llmi/llmi_gspo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/recipe/llmi/llmi_gspo.sh -------------------------------------------------------------------------------- /recipe/llmi/mllmi_grpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/recipe/llmi/mllmi_grpo.sh -------------------------------------------------------------------------------- /requirements-npu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/requirements-npu.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_llmi.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/requirements_llmi.txt -------------------------------------------------------------------------------- /requirements_sglang.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/requirements_sglang.txt -------------------------------------------------------------------------------- /scripts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/scripts/__init__.py -------------------------------------------------------------------------------- /scripts/converter_hf_to_mcore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/scripts/converter_hf_to_mcore.py -------------------------------------------------------------------------------- /scripts/diagnose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/scripts/diagnose.py -------------------------------------------------------------------------------- /scripts/generate_trainer_config.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/scripts/generate_trainer_config.sh -------------------------------------------------------------------------------- /scripts/init_random_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/scripts/init_random_model.py -------------------------------------------------------------------------------- /scripts/install_vllm_sglang_mcore.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/scripts/install_vllm_sglang_mcore.sh -------------------------------------------------------------------------------- /scripts/legacy_model_merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/scripts/legacy_model_merger.py -------------------------------------------------------------------------------- /scripts/print_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/scripts/print_cfg.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/setup.py -------------------------------------------------------------------------------- /system_prompts/training/llmi.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/system_prompts/training/llmi.txt -------------------------------------------------------------------------------- /system_prompts/training/llmjudge.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/system_prompts/training/llmjudge.txt -------------------------------------------------------------------------------- /system_prompts/training/mllmjudge.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/system_prompts/training/mllmjudge.txt -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/README.md -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/experimental/agent_loop/agent_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/experimental/agent_loop/agent_utils.py -------------------------------------------------------------------------------- /tests/experimental/agent_loop/test_basic_agent_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/experimental/agent_loop/test_basic_agent_loop.py -------------------------------------------------------------------------------- /tests/interactions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/interactions/__init__.py -------------------------------------------------------------------------------- /tests/interactions/test_gsm8k_interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/interactions/test_gsm8k_interaction.py -------------------------------------------------------------------------------- /tests/interactions/test_interaction_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/interactions/test_interaction_registry.py -------------------------------------------------------------------------------- /tests/kill_github_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/kill_github_tests.sh -------------------------------------------------------------------------------- /tests/models/test_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/models/test_transformer.py -------------------------------------------------------------------------------- /tests/models/test_transformers_ulysses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/models/test_transformers_ulysses.py -------------------------------------------------------------------------------- /tests/single_controller/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/__init__.py -------------------------------------------------------------------------------- /tests/single_controller/base/test_decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/base/test_decorator.py -------------------------------------------------------------------------------- /tests/single_controller/check_worker_alive/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/check_worker_alive/main.py -------------------------------------------------------------------------------- /tests/single_controller/detached_worker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/detached_worker/README.md -------------------------------------------------------------------------------- /tests/single_controller/detached_worker/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/detached_worker/client.py -------------------------------------------------------------------------------- /tests/single_controller/detached_worker/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/detached_worker/run.sh -------------------------------------------------------------------------------- /tests/single_controller/detached_worker/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/detached_worker/server.py -------------------------------------------------------------------------------- /tests/single_controller/test_auto_padding_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_auto_padding_on_cpu.py -------------------------------------------------------------------------------- /tests/single_controller/test_colocated_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_colocated_workers.py -------------------------------------------------------------------------------- /tests/single_controller/test_colocated_workers_fused.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_colocated_workers_fused.py -------------------------------------------------------------------------------- /tests/single_controller/test_data_transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_data_transfer.py -------------------------------------------------------------------------------- /tests/single_controller/test_decorator_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_decorator_on_cpu.py -------------------------------------------------------------------------------- /tests/single_controller/test_driverfunc_to_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_driverfunc_to_worker.py -------------------------------------------------------------------------------- /tests/single_controller/test_fused_workers_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_fused_workers_on_cpu.py -------------------------------------------------------------------------------- /tests/single_controller/test_high_level_scheduling_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_high_level_scheduling_api.py -------------------------------------------------------------------------------- /tests/single_controller/test_ray_collectives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_ray_collectives.py -------------------------------------------------------------------------------- /tests/single_controller/test_ray_local_envs_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_ray_local_envs_on_cpu.py -------------------------------------------------------------------------------- /tests/single_controller/test_ray_utils_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_ray_utils_on_cpu.py -------------------------------------------------------------------------------- /tests/single_controller/test_rvdz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_rvdz.py -------------------------------------------------------------------------------- /tests/single_controller/test_worker_group_basics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_worker_group_basics.py -------------------------------------------------------------------------------- /tests/single_controller/test_worker_group_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/single_controller/test_worker_group_torch.py -------------------------------------------------------------------------------- /tests/special_distributed/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_distributed/README.md -------------------------------------------------------------------------------- /tests/special_distributed/run_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_distributed/run_all.sh -------------------------------------------------------------------------------- /tests/special_distributed/test_fsdp_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_distributed/test_fsdp_ckpt.py -------------------------------------------------------------------------------- /tests/special_distributed/test_tensor_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_distributed/test_tensor_dict.py -------------------------------------------------------------------------------- /tests/special_e2e/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/README.md -------------------------------------------------------------------------------- /tests/special_e2e/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/__init__.py -------------------------------------------------------------------------------- /tests/special_e2e/check_custom_rwd_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/check_custom_rwd_fn.py -------------------------------------------------------------------------------- /tests/special_e2e/check_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/check_results.py -------------------------------------------------------------------------------- /tests/special_e2e/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/envs/__init__.py -------------------------------------------------------------------------------- /tests/special_e2e/envs/digit_completion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/envs/digit_completion/__init__.py -------------------------------------------------------------------------------- /tests/special_e2e/envs/digit_completion/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/envs/digit_completion/task.py -------------------------------------------------------------------------------- /tests/special_e2e/envs/digit_completion/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/envs/digit_completion/tokenizer.py -------------------------------------------------------------------------------- /tests/special_e2e/generation/run_gen_qwen05.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/generation/run_gen_qwen05.sh -------------------------------------------------------------------------------- /tests/special_e2e/ppo_trainer/expert_parallel/qwen2moe_minimal.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/ppo_trainer/expert_parallel/qwen2moe_minimal.json -------------------------------------------------------------------------------- /tests/special_e2e/ppo_trainer/run_function_reward.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/ppo_trainer/run_function_reward.sh -------------------------------------------------------------------------------- /tests/special_e2e/ppo_trainer/run_model_reward.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/ppo_trainer/run_model_reward.sh -------------------------------------------------------------------------------- /tests/special_e2e/ppo_trainer/run_single_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/ppo_trainer/run_single_gpu.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_dapo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_dapo.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_genrm_remote.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_genrm_remote.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_geo3k_fsdp_sgl_multiturn_w_tool.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_geo3k_fsdp_sgl_multiturn_w_tool.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_grpo_lora_with_merge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_grpo_lora_with_merge.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_gsm8k_fsdp_sgl_multiturn_sf_tool.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_gsm8k_fsdp_sgl_multiturn_sf_tool.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_gsm8k_fsdp_sgl_multiturn_w_tool.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_gsm8k_fsdp_sgl_multiturn_w_tool.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_one_step_off_policy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_one_step_off_policy.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_ppo_trainer_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_ppo_trainer_megatron.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_prime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_prime.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_r1_distill_qwen_aime24_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_r1_distill_qwen_aime24_eval.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_spin.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_spin.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_sppo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_sppo.sh -------------------------------------------------------------------------------- /tests/special_e2e/run_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/run_test.sh -------------------------------------------------------------------------------- /tests/special_e2e/sft/run_sft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/sft/run_sft.sh -------------------------------------------------------------------------------- /tests/special_e2e/sft/test_sp_loss_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_e2e/sft/test_sp_loss_match.py -------------------------------------------------------------------------------- /tests/special_npu/run_qwen2_5_05b_dapo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_npu/run_qwen2_5_05b_dapo.sh -------------------------------------------------------------------------------- /tests/special_npu/run_qwen2_5_05b_grpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_npu/run_qwen2_5_05b_grpo.sh -------------------------------------------------------------------------------- /tests/special_npu/run_qwen2_5_05b_sft_peft_sp2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_npu/run_qwen2_5_05b_sft_peft_sp2.sh -------------------------------------------------------------------------------- /tests/special_npu/run_qwen2_5_vl_3b_npu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_npu/run_qwen2_5_vl_3b_npu.sh -------------------------------------------------------------------------------- /tests/special_sanity/check_api_docs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/check_api_docs.py -------------------------------------------------------------------------------- /tests/special_sanity/check_device_api_usage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/check_device_api_usage.py -------------------------------------------------------------------------------- /tests/special_sanity/check_docs_time_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/check_docs_time_info.py -------------------------------------------------------------------------------- /tests/special_sanity/check_docstrings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/check_docstrings.py -------------------------------------------------------------------------------- /tests/special_sanity/check_license.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/check_license.py -------------------------------------------------------------------------------- /tests/special_sanity/check_pr_description.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/check_pr_description.py -------------------------------------------------------------------------------- /tests/special_sanity/check_pr_title.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/check_pr_title.py -------------------------------------------------------------------------------- /tests/special_sanity/test_config_docs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/test_config_docs.py -------------------------------------------------------------------------------- /tests/special_sanity/test_import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/test_import.py -------------------------------------------------------------------------------- /tests/special_sanity/type_coverage_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/type_coverage_check.py -------------------------------------------------------------------------------- /tests/special_sanity/validate_imported_docs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/validate_imported_docs.py -------------------------------------------------------------------------------- /tests/special_sanity/validate_structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_sanity/validate_structure.py -------------------------------------------------------------------------------- /tests/special_standalone/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_standalone/README.md -------------------------------------------------------------------------------- /tests/special_standalone/test_memory_buffers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/special_standalone/test_memory_buffers.py -------------------------------------------------------------------------------- /tests/test_base_config_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/test_base_config_on_cpu.py -------------------------------------------------------------------------------- /tests/test_protocol_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/test_protocol_on_cpu.py -------------------------------------------------------------------------------- /tests/tools/test_base_tool_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/tools/test_base_tool_on_cpu.py -------------------------------------------------------------------------------- /tests/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/trainer/__init__.py -------------------------------------------------------------------------------- /tests/trainer/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/trainer/config/__init__.py -------------------------------------------------------------------------------- /tests/trainer/config/legacy_ppo_megatron_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/trainer/config/legacy_ppo_megatron_trainer.yaml -------------------------------------------------------------------------------- /tests/trainer/config/legacy_ppo_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/trainer/config/legacy_ppo_trainer.yaml -------------------------------------------------------------------------------- /tests/trainer/config/test_algo_config_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/trainer/config/test_algo_config_on_cpu.py -------------------------------------------------------------------------------- /tests/trainer/config/test_critic_config_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/trainer/config/test_critic_config_on_cpu.py -------------------------------------------------------------------------------- /tests/trainer/config/test_legacy_config_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/trainer/config/test_legacy_config_on_cpu.py -------------------------------------------------------------------------------- /tests/trainer/ppo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/trainer/ppo/__init__.py -------------------------------------------------------------------------------- /tests/trainer/ppo/test_core_algos_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/trainer/ppo/test_core_algos_on_cpu.py -------------------------------------------------------------------------------- /tests/trainer/ppo/test_metric_utils_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/trainer/ppo/test_metric_utils_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/_test_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/_test_module.py -------------------------------------------------------------------------------- /tests/utils/ckpt/test_esi_save_ckpt_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/ckpt/test_esi_save_ckpt_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/dataset/test_create_rl_sampler_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/dataset/test_create_rl_sampler_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/dataset/test_multiturn_sft_dataset_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/dataset/test_multiturn_sft_dataset_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/dataset/test_rl_dataset_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/dataset/test_rl_dataset_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/dataset/test_sft_dataset_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/dataset/test_sft_dataset_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/megatron/test_pipeline_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/megatron/test_pipeline_parallel.py -------------------------------------------------------------------------------- /tests/utils/reward_score/reward_score/test_sandbox_fusion_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/reward_score/reward_score/test_sandbox_fusion_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/reward_score/test_sandbox_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/reward_score/test_sandbox_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/test_activation_offload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_activation_offload.py -------------------------------------------------------------------------------- /tests/utils/test_config_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_config_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/test_flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_flops_counter.py -------------------------------------------------------------------------------- /tests/utils/test_fs_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_fs_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/test_import_utils_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_import_utils_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/test_linear_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_linear_cross_entropy.py -------------------------------------------------------------------------------- /tests/utils/test_linear_cross_entropy_tp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_linear_cross_entropy_tp.py -------------------------------------------------------------------------------- /tests/utils/test_model_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_model_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/test_nvtx_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_nvtx_profile.py -------------------------------------------------------------------------------- /tests/utils/test_rollout_trace_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_rollout_trace_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/test_seqlen_balancing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_seqlen_balancing.py -------------------------------------------------------------------------------- /tests/utils/test_temp_env_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_temp_env_on_cpu.py -------------------------------------------------------------------------------- /tests/utils/test_timeout_decorator_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_timeout_decorator_cpu.py -------------------------------------------------------------------------------- /tests/utils/test_torch_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/utils/test_torch_functional.py -------------------------------------------------------------------------------- /tests/workers/reward_manager/test_registry_on_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/reward_manager/test_registry_on_cpu.py -------------------------------------------------------------------------------- /tests/workers/rollout/async_rollout_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/async_rollout_utils.py -------------------------------------------------------------------------------- /tests/workers/rollout/perf/vllm_async_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/perf/vllm_async_rollout.py -------------------------------------------------------------------------------- /tests/workers/rollout/resource/tool_configs/mcp_server.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/resource/tool_configs/mcp_server.json -------------------------------------------------------------------------------- /tests/workers/rollout/resource/tool_configs/mcp_tool_config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/resource/tool_configs/mcp_tool_config -------------------------------------------------------------------------------- /tests/workers/rollout/resource/tool_configs/sandbox_fusion_tool_config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/resource/tool_configs/sandbox_fusion_tool_config -------------------------------------------------------------------------------- /tests/workers/rollout/resource/tool_configs/search_tool_config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/resource/tool_configs/search_tool_config -------------------------------------------------------------------------------- /tests/workers/rollout/rollout_vllm/run_fsdp_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/rollout_vllm/run_fsdp_vllm.py -------------------------------------------------------------------------------- /tests/workers/rollout/rollout_vllm/test_vllm_chat_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/rollout_vllm/test_vllm_chat_scheduler.py -------------------------------------------------------------------------------- /tests/workers/rollout/rollout_vllm/test_vllm_model_rope_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/rollout_vllm/test_vllm_model_rope_scaling.py -------------------------------------------------------------------------------- /tests/workers/rollout/rollout_vllm/test_vllm_spmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/rollout_vllm/test_vllm_spmd.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_async_sglang_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_async_sglang_server.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_custom_completion_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_custom_completion_callback.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_hf_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_hf_rollout.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_async_rollout_mcp_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_sglang_async_rollout_mcp_tools.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_async_rollout_multimodal_delta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_sglang_async_rollout_multimodal_delta.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_async_rollout_search_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_sglang_async_rollout_search_tools.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_async_rollout_sf_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_sglang_async_rollout_sf_tools.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_async_rollout_w_interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_sglang_async_rollout_w_interaction.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_async_rollout_w_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_sglang_async_rollout_w_tools.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_multi_interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_sglang_multi_interaction.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_rollout_sharding_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_sglang_rollout_sharding_manager.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_spmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/test_sglang_spmd.py -------------------------------------------------------------------------------- /tests/workers/rollout/utils_sglang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/tests/workers/rollout/utils_sglang.py -------------------------------------------------------------------------------- /verl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/__init__.py -------------------------------------------------------------------------------- /verl/base_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/base_config.py -------------------------------------------------------------------------------- /verl/experimental/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/experimental/__init__.py -------------------------------------------------------------------------------- /verl/experimental/agent_loop/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/experimental/agent_loop/__init__.py -------------------------------------------------------------------------------- /verl/experimental/agent_loop/agent_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/experimental/agent_loop/agent_loop.py -------------------------------------------------------------------------------- /verl/experimental/agent_loop/single_turn_agent_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/experimental/agent_loop/single_turn_agent_loop.py -------------------------------------------------------------------------------- /verl/experimental/agent_loop/tool_agent_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/experimental/agent_loop/tool_agent_loop.py -------------------------------------------------------------------------------- /verl/experimental/agent_loop/tool_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/experimental/agent_loop/tool_parser.py -------------------------------------------------------------------------------- /verl/experimental/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/experimental/dataset/__init__.py -------------------------------------------------------------------------------- /verl/experimental/dataset/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/experimental/dataset/sampler.py -------------------------------------------------------------------------------- /verl/experimental/dynamic_dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/experimental/dynamic_dataset/__init__.py -------------------------------------------------------------------------------- /verl/experimental/dynamic_dataset/dynamicgen_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/experimental/dynamic_dataset/dynamicgen_dataset.py -------------------------------------------------------------------------------- /verl/interactions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/interactions/__init__.py -------------------------------------------------------------------------------- /verl/interactions/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/interactions/base.py -------------------------------------------------------------------------------- /verl/interactions/gsm8k_interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/interactions/gsm8k_interaction.py -------------------------------------------------------------------------------- /verl/interactions/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/interactions/utils/__init__.py -------------------------------------------------------------------------------- /verl/interactions/utils/interaction_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/interactions/utils/interaction_registry.py -------------------------------------------------------------------------------- /verl/model_merger/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/model_merger/__init__.py -------------------------------------------------------------------------------- /verl/model_merger/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/model_merger/__main__.py -------------------------------------------------------------------------------- /verl/model_merger/base_model_merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/model_merger/base_model_merger.py -------------------------------------------------------------------------------- /verl/model_merger/fsdp_model_merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/model_merger/fsdp_model_merger.py -------------------------------------------------------------------------------- /verl/model_merger/megatron_model_merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/model_merger/megatron_model_merger.py -------------------------------------------------------------------------------- /verl/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/README.md -------------------------------------------------------------------------------- /verl/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/checkpoint_utils/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/llama_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/checkpoint_utils/llama_loader.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/llama_loader_depracated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/checkpoint_utils/llama_loader_depracated.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/llama_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/checkpoint_utils/llama_saver.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/layers/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/layers/parallel_attention.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/layers/parallel_decoder.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/layers/parallel_linear.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/layers/parallel_mlp.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/layers/parallel_rmsnorm.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/modeling_llama_megatron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/llama/megatron/modeling_llama_megatron.py -------------------------------------------------------------------------------- /verl/models/mcore/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/__init__.py -------------------------------------------------------------------------------- /verl/models/mcore/config_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/config_converter.py -------------------------------------------------------------------------------- /verl/models/mcore/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/loader.py -------------------------------------------------------------------------------- /verl/models/mcore/mbridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/mbridge.py -------------------------------------------------------------------------------- /verl/models/mcore/model_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/model_forward.py -------------------------------------------------------------------------------- /verl/models/mcore/model_forward_fused.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/model_forward_fused.py -------------------------------------------------------------------------------- /verl/models/mcore/model_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/model_initializer.py -------------------------------------------------------------------------------- /verl/models/mcore/patch_v012.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/patch_v012.py -------------------------------------------------------------------------------- /verl/models/mcore/qwen2_5_vl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/qwen2_5_vl/__init__.py -------------------------------------------------------------------------------- /verl/models/mcore/qwen2_5_vl/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/qwen2_5_vl/attention.py -------------------------------------------------------------------------------- /verl/models/mcore/qwen2_5_vl/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/qwen2_5_vl/model.py -------------------------------------------------------------------------------- /verl/models/mcore/qwen2_5_vl/rope_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/qwen2_5_vl/rope_utils.py -------------------------------------------------------------------------------- /verl/models/mcore/qwen2_5_vl/vision_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/qwen2_5_vl/vision_config.py -------------------------------------------------------------------------------- /verl/models/mcore/qwen2_5_vl/vision_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/qwen2_5_vl/vision_model.py -------------------------------------------------------------------------------- /verl/models/mcore/qwen2_5_vl/vision_transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/qwen2_5_vl/vision_transformer_block.py -------------------------------------------------------------------------------- /verl/models/mcore/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/readme.md -------------------------------------------------------------------------------- /verl/models/mcore/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/registry.py -------------------------------------------------------------------------------- /verl/models/mcore/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/saver.py -------------------------------------------------------------------------------- /verl/models/mcore/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/util.py -------------------------------------------------------------------------------- /verl/models/mcore/weight_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/mcore/weight_converter.py -------------------------------------------------------------------------------- /verl/models/qwen2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/__init__.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/__init__.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/checkpoint_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/checkpoint_utils/__init__.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader_depracated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader_depracated.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/checkpoint_utils/qwen2_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/checkpoint_utils/qwen2_saver.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/layers/__init__.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/parallel_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/layers/parallel_attention.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/parallel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/layers/parallel_decoder.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/parallel_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/layers/parallel_linear.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/parallel_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/layers/parallel_mlp.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/parallel_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/layers/parallel_rmsnorm.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/modeling_qwen2_megatron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/qwen2/megatron/modeling_qwen2_megatron.py -------------------------------------------------------------------------------- /verl/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/registry.py -------------------------------------------------------------------------------- /verl/models/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/transformers/__init__.py -------------------------------------------------------------------------------- /verl/models/transformers/dense_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/transformers/dense_common.py -------------------------------------------------------------------------------- /verl/models/transformers/kimi_vl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/transformers/kimi_vl.py -------------------------------------------------------------------------------- /verl/models/transformers/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/transformers/llama.py -------------------------------------------------------------------------------- /verl/models/transformers/monkey_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/transformers/monkey_patch.py -------------------------------------------------------------------------------- /verl/models/transformers/npu_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/transformers/npu_patch.py -------------------------------------------------------------------------------- /verl/models/transformers/qwen2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/transformers/qwen2.py -------------------------------------------------------------------------------- /verl/models/transformers/qwen2_5_vl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/transformers/qwen2_5_vl.py -------------------------------------------------------------------------------- /verl/models/transformers/qwen2_vl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/transformers/qwen2_vl.py -------------------------------------------------------------------------------- /verl/models/weight_loader_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/models/weight_loader_registry.py -------------------------------------------------------------------------------- /verl/protocol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/protocol.py -------------------------------------------------------------------------------- /verl/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /verl/single_controller/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/base/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/base/decorator.py -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/base/megatron/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/base/megatron/worker.py -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/worker_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/base/megatron/worker_group.py -------------------------------------------------------------------------------- /verl/single_controller/base/register_center/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/base/register_center/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/register_center/ray.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/base/register_center/ray.py -------------------------------------------------------------------------------- /verl/single_controller/base/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/base/worker.py -------------------------------------------------------------------------------- /verl/single_controller/base/worker_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/base/worker_group.py -------------------------------------------------------------------------------- /verl/single_controller/ray/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/ray/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/ray/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/ray/base.py -------------------------------------------------------------------------------- /verl/single_controller/ray/megatron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/single_controller/ray/megatron.py -------------------------------------------------------------------------------- /verl/third_party/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/third_party/__init__.py -------------------------------------------------------------------------------- /verl/third_party/sglang/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/third_party/sglang/__init__.py -------------------------------------------------------------------------------- /verl/third_party/sglang/parallel_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/third_party/sglang/parallel_state.py -------------------------------------------------------------------------------- /verl/third_party/vllm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/third_party/vllm/__init__.py -------------------------------------------------------------------------------- /verl/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/__init__.py -------------------------------------------------------------------------------- /verl/tools/base_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/base_tool.py -------------------------------------------------------------------------------- /verl/tools/geo3k_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/geo3k_tool.py -------------------------------------------------------------------------------- /verl/tools/gsm8k_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/gsm8k_tool.py -------------------------------------------------------------------------------- /verl/tools/mcp_base_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/mcp_base_tool.py -------------------------------------------------------------------------------- /verl/tools/mcp_search_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/mcp_search_tool.py -------------------------------------------------------------------------------- /verl/tools/sandbox_fusion_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/sandbox_fusion_tools.py -------------------------------------------------------------------------------- /verl/tools/schemas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/schemas.py -------------------------------------------------------------------------------- /verl/tools/search_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/search_tool.py -------------------------------------------------------------------------------- /verl/tools/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/utils/__init__.py -------------------------------------------------------------------------------- /verl/tools/utils/mcp_clients/McpClientManager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/utils/mcp_clients/McpClientManager.py -------------------------------------------------------------------------------- /verl/tools/utils/mcp_clients/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/utils/mcp_clients/utils.py -------------------------------------------------------------------------------- /verl/tools/utils/search_r1_like_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/utils/search_r1_like_utils.py -------------------------------------------------------------------------------- /verl/tools/utils/tool_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/tools/utils/tool_registry.py -------------------------------------------------------------------------------- /verl/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/__init__.py -------------------------------------------------------------------------------- /verl/trainer/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/__init__.py -------------------------------------------------------------------------------- /verl/trainer/config/_generated_ppo_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/_generated_ppo_trainer.yaml -------------------------------------------------------------------------------- /verl/trainer/config/actor/actor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/actor/actor.yaml -------------------------------------------------------------------------------- /verl/trainer/config/actor/dp_actor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/actor/dp_actor.yaml -------------------------------------------------------------------------------- /verl/trainer/config/actor/megatron_actor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/actor/megatron_actor.yaml -------------------------------------------------------------------------------- /verl/trainer/config/algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/algorithm.py -------------------------------------------------------------------------------- /verl/trainer/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/config.py -------------------------------------------------------------------------------- /verl/trainer/config/critic/critic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/critic/critic.yaml -------------------------------------------------------------------------------- /verl/trainer/config/critic/dp_critic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/critic/dp_critic.yaml -------------------------------------------------------------------------------- /verl/trainer/config/critic/megatron_critic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/critic/megatron_critic.yaml -------------------------------------------------------------------------------- /verl/trainer/config/data/legacy_data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/data/legacy_data.yaml -------------------------------------------------------------------------------- /verl/trainer/config/evaluation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/evaluation.yaml -------------------------------------------------------------------------------- /verl/trainer/config/generation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/generation.yaml -------------------------------------------------------------------------------- /verl/trainer/config/npu_profile/npu_profile.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/npu_profile/npu_profile.yaml -------------------------------------------------------------------------------- /verl/trainer/config/ppo_megatron_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/ppo_megatron_trainer.yaml -------------------------------------------------------------------------------- /verl/trainer/config/ppo_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/ppo_trainer.yaml -------------------------------------------------------------------------------- /verl/trainer/config/ref/dp_ref.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/ref/dp_ref.yaml -------------------------------------------------------------------------------- /verl/trainer/config/ref/megatron_ref.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/ref/megatron_ref.yaml -------------------------------------------------------------------------------- /verl/trainer/config/ref/ref.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/ref/ref.yaml -------------------------------------------------------------------------------- /verl/trainer/config/reward_model/dp_reward_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/reward_model/dp_reward_model.yaml -------------------------------------------------------------------------------- /verl/trainer/config/reward_model/megatron_reward_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/reward_model/megatron_reward_model.yaml -------------------------------------------------------------------------------- /verl/trainer/config/reward_model/reward_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/reward_model/reward_model.yaml -------------------------------------------------------------------------------- /verl/trainer/config/rollout/rollout.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/rollout/rollout.yaml -------------------------------------------------------------------------------- /verl/trainer/config/sft_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/config/sft_trainer.yaml -------------------------------------------------------------------------------- /verl/trainer/constants_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/constants_ppo.py -------------------------------------------------------------------------------- /verl/trainer/fsdp_sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/fsdp_sft_trainer.py -------------------------------------------------------------------------------- /verl/trainer/main_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/main_eval.py -------------------------------------------------------------------------------- /verl/trainer/main_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/main_generation.py -------------------------------------------------------------------------------- /verl/trainer/main_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/main_ppo.py -------------------------------------------------------------------------------- /verl/trainer/ppo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/ppo/__init__.py -------------------------------------------------------------------------------- /verl/trainer/ppo/core_algos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/ppo/core_algos.py -------------------------------------------------------------------------------- /verl/trainer/ppo/metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/ppo/metric_utils.py -------------------------------------------------------------------------------- /verl/trainer/ppo/ray_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/ppo/ray_trainer.py -------------------------------------------------------------------------------- /verl/trainer/ppo/reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/trainer/ppo/reward.py -------------------------------------------------------------------------------- /verl/trainer/runtime_env.yaml: -------------------------------------------------------------------------------- 1 | working_dir: ./ 2 | excludes: ["/.git/"] 3 | env_vars: 4 | TORCH_NCCL_AVOID_RECORD_STREAMS: "1" 5 | -------------------------------------------------------------------------------- /verl/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/__init__.py -------------------------------------------------------------------------------- /verl/utils/activation_offload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/activation_offload.py -------------------------------------------------------------------------------- /verl/utils/checkpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/checkpoint/__init__.py -------------------------------------------------------------------------------- /verl/utils/checkpoint/checkpoint_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/checkpoint/checkpoint_manager.py -------------------------------------------------------------------------------- /verl/utils/checkpoint/fsdp_checkpoint_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/checkpoint/fsdp_checkpoint_manager.py -------------------------------------------------------------------------------- /verl/utils/checkpoint/megatron_checkpoint_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/checkpoint/megatron_checkpoint_manager.py -------------------------------------------------------------------------------- /verl/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/config.py -------------------------------------------------------------------------------- /verl/utils/dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/dataset/README.md -------------------------------------------------------------------------------- /verl/utils/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/dataset/__init__.py -------------------------------------------------------------------------------- /verl/utils/dataset/multiturn_sft_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/dataset/multiturn_sft_dataset.py -------------------------------------------------------------------------------- /verl/utils/dataset/rl_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/dataset/rl_dataset.py -------------------------------------------------------------------------------- /verl/utils/dataset/rm_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/dataset/rm_dataset.py -------------------------------------------------------------------------------- /verl/utils/dataset/sft_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/dataset/sft_dataset.py -------------------------------------------------------------------------------- /verl/utils/dataset/vision_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/dataset/vision_utils.py -------------------------------------------------------------------------------- /verl/utils/debug/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/debug/__init__.py -------------------------------------------------------------------------------- /verl/utils/debug/performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/debug/performance.py -------------------------------------------------------------------------------- /verl/utils/debug/trajectory_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/debug/trajectory_tracker.py -------------------------------------------------------------------------------- /verl/utils/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/device.py -------------------------------------------------------------------------------- /verl/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/distributed.py -------------------------------------------------------------------------------- /verl/utils/experimental/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/experimental/__init__.py -------------------------------------------------------------------------------- /verl/utils/experimental/torch_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/experimental/torch_functional.py -------------------------------------------------------------------------------- /verl/utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/flops_counter.py -------------------------------------------------------------------------------- /verl/utils/fs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/fs.py -------------------------------------------------------------------------------- /verl/utils/fsdp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/fsdp_utils.py -------------------------------------------------------------------------------- /verl/utils/hdfs_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/hdfs_io.py -------------------------------------------------------------------------------- /verl/utils/import_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/import_utils.py -------------------------------------------------------------------------------- /verl/utils/kernel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/kernel/__init__.py -------------------------------------------------------------------------------- /verl/utils/kernel/kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/kernel/kernels.py -------------------------------------------------------------------------------- /verl/utils/kernel/linear_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/kernel/linear_cross_entropy.py -------------------------------------------------------------------------------- /verl/utils/logger/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/logger/__init__.py -------------------------------------------------------------------------------- /verl/utils/logger/aggregate_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/logger/aggregate_logger.py -------------------------------------------------------------------------------- /verl/utils/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/logging_utils.py -------------------------------------------------------------------------------- /verl/utils/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/megatron/__init__.py -------------------------------------------------------------------------------- /verl/utils/megatron/dist_checkpointing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/megatron/dist_checkpointing.py -------------------------------------------------------------------------------- /verl/utils/megatron/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/megatron/memory.py -------------------------------------------------------------------------------- /verl/utils/megatron/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/megatron/optimizer.py -------------------------------------------------------------------------------- /verl/utils/megatron/pipeline_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/megatron/pipeline_parallel.py -------------------------------------------------------------------------------- /verl/utils/megatron/sequence_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/megatron/sequence_parallel.py -------------------------------------------------------------------------------- /verl/utils/megatron/tensor_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/megatron/tensor_parallel.py -------------------------------------------------------------------------------- /verl/utils/megatron_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/megatron_utils.py -------------------------------------------------------------------------------- /verl/utils/memory_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/memory_buffer.py -------------------------------------------------------------------------------- /verl/utils/metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/metric/__init__.py -------------------------------------------------------------------------------- /verl/utils/metric/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/metric/utils.py -------------------------------------------------------------------------------- /verl/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/model.py -------------------------------------------------------------------------------- /verl/utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/net_utils.py -------------------------------------------------------------------------------- /verl/utils/profiler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/profiler/__init__.py -------------------------------------------------------------------------------- /verl/utils/profiler/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/profiler/config.py -------------------------------------------------------------------------------- /verl/utils/profiler/empty_annotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/profiler/empty_annotations.py -------------------------------------------------------------------------------- /verl/utils/profiler/mstx_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/profiler/mstx_profile.py -------------------------------------------------------------------------------- /verl/utils/profiler/nvtx_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/profiler/nvtx_profile.py -------------------------------------------------------------------------------- /verl/utils/profiler/performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/profiler/performance.py -------------------------------------------------------------------------------- /verl/utils/profiler/profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/profiler/profile.py -------------------------------------------------------------------------------- /verl/utils/py_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/py_functional.py -------------------------------------------------------------------------------- /verl/utils/ray_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/ray_utils.py -------------------------------------------------------------------------------- /verl/utils/rendezvous/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/rendezvous/__init__.py -------------------------------------------------------------------------------- /verl/utils/rendezvous/ray_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/rendezvous/ray_backend.py -------------------------------------------------------------------------------- /verl/utils/reward_score/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/__init__.py -------------------------------------------------------------------------------- /verl/utils/reward_score/geo3k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/geo3k.py -------------------------------------------------------------------------------- /verl/utils/reward_score/gsm8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/gsm8k.py -------------------------------------------------------------------------------- /verl/utils/reward_score/judge_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/judge_client.py -------------------------------------------------------------------------------- /verl/utils/reward_score/llmi_rm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/llmi_rm.py -------------------------------------------------------------------------------- /verl/utils/reward_score/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/math.py -------------------------------------------------------------------------------- /verl/utils/reward_score/math_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/math_batch.py -------------------------------------------------------------------------------- /verl/utils/reward_score/math_dapo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/math_dapo.py -------------------------------------------------------------------------------- /verl/utils/reward_score/math_verify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/math_verify.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_code/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/prime_code/README.md -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_code/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/prime_code/__init__.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_code/testing_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/prime_code/testing_util.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_code/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/prime_code/utils.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_math/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/prime_math/__init__.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_math/grader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/prime_math/grader.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_math/math_normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/prime_math/math_normalize.py -------------------------------------------------------------------------------- /verl/utils/reward_score/sandbox_fusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/sandbox_fusion/__init__.py -------------------------------------------------------------------------------- /verl/utils/reward_score/sandbox_fusion/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/sandbox_fusion/utils.py -------------------------------------------------------------------------------- /verl/utils/reward_score/search_r1_like_qa_em.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/reward_score/search_r1_like_qa_em.py -------------------------------------------------------------------------------- /verl/utils/rollout_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/rollout_trace.py -------------------------------------------------------------------------------- /verl/utils/seqlen_balancing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/seqlen_balancing.py -------------------------------------------------------------------------------- /verl/utils/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/tokenizer.py -------------------------------------------------------------------------------- /verl/utils/torch_dtypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/torch_dtypes.py -------------------------------------------------------------------------------- /verl/utils/torch_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/torch_functional.py -------------------------------------------------------------------------------- /verl/utils/tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/tracking.py -------------------------------------------------------------------------------- /verl/utils/ulysses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/ulysses.py -------------------------------------------------------------------------------- /verl/utils/vllm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/utils/vllm_utils.py -------------------------------------------------------------------------------- /verl/version/version: -------------------------------------------------------------------------------- 1 | 0.4.1.dev 2 | -------------------------------------------------------------------------------- /verl/workers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/__init__.py -------------------------------------------------------------------------------- /verl/workers/actor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/actor/__init__.py -------------------------------------------------------------------------------- /verl/workers/actor/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/actor/base.py -------------------------------------------------------------------------------- /verl/workers/actor/dp_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/actor/dp_actor.py -------------------------------------------------------------------------------- /verl/workers/actor/megatron_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/actor/megatron_actor.py -------------------------------------------------------------------------------- /verl/workers/critic/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/critic/__init__.py -------------------------------------------------------------------------------- /verl/workers/critic/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/critic/base.py -------------------------------------------------------------------------------- /verl/workers/critic/dp_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/critic/dp_critic.py -------------------------------------------------------------------------------- /verl/workers/critic/megatron_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/critic/megatron_critic.py -------------------------------------------------------------------------------- /verl/workers/fsdp_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/fsdp_workers.py -------------------------------------------------------------------------------- /verl/workers/megatron_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/megatron_workers.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_manager/__init__.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_manager/batch.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/dapo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_manager/dapo.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/llmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_manager/llmi.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/naive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_manager/naive.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/prime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_manager/prime.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_manager/registry.py -------------------------------------------------------------------------------- /verl/workers/reward_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_model/__init__.py -------------------------------------------------------------------------------- /verl/workers/reward_model/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_model/base.py -------------------------------------------------------------------------------- /verl/workers/reward_model/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_model/megatron/__init__.py -------------------------------------------------------------------------------- /verl/workers/reward_model/megatron/reward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/reward_model/megatron/reward_model.py -------------------------------------------------------------------------------- /verl/workers/rollout/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/async_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/async_server.py -------------------------------------------------------------------------------- /verl/workers/rollout/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/base.py -------------------------------------------------------------------------------- /verl/workers/rollout/chat_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/chat_scheduler.py -------------------------------------------------------------------------------- /verl/workers/rollout/hf_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/hf_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/naive/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/naive/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/naive/naive_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/naive/naive_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/schemas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/schemas.py -------------------------------------------------------------------------------- /verl/workers/rollout/sglang_rollout/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/sglang_rollout/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/sglang_rollout/async_sglang_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/sglang_rollout/async_sglang_server.py -------------------------------------------------------------------------------- /verl/workers/rollout/sglang_rollout/sglang_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/sglang_rollout/sglang_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/sglang_rollout/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/sglang_rollout/utils.py -------------------------------------------------------------------------------- /verl/workers/rollout/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/tokenizer.py -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/vllm_rollout/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/vllm_async_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/vllm_rollout/vllm_async_server.py -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/vllm_rollout_spmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/rollout/vllm_rollout/vllm_rollout_spmd.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/sharding_manager/__init__.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/sharding_manager/base.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/fsdp_sglang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/sharding_manager/fsdp_sglang.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/fsdp_ulysses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/sharding_manager/fsdp_ulysses.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/fsdp_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/sharding_manager/fsdp_vllm.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/megatron_sglang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/sharding_manager/megatron_sglang.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/megatron_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ByteDance-BandAI/LLM-I/HEAD/verl/workers/sharding_manager/megatron_vllm.py --------------------------------------------------------------------------------