├── LICENSE ├── README.md ├── dataset-annotation-json ├── dev.jsonl ├── test.jsonl └── train.jsonl ├── docs ├── 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 │ ├── placement.rst │ └── rope.rst ├── amd_tutorial │ ├── amd_build_dockerfile_page.rst │ └── amd_vllm_page.rst ├── api │ ├── data.rst │ ├── single_controller.rst │ ├── trainer.rst │ └── utils.rst ├── ascend │ └── ascend_vllm073.rst ├── conf.py ├── examples │ ├── config.rst │ ├── gsm8k_example.rst │ ├── multi_modal_example.rst │ ├── ppo_code_architecture.rst │ └── sandbox_fusion_example.rst ├── experiment │ └── ppo.rst ├── faq │ └── faq.rst ├── hybrid_flow.rst ├── index.rst ├── perf │ ├── device_tuning.rst │ └── perf_tuning.rst ├── preparation │ ├── prepare_data.rst │ └── reward_function.rst ├── requirements-docs.txt ├── sglang_multiturn │ └── multiturn.rst ├── single_controller.rst ├── start │ ├── install.rst │ ├── multinode.rst │ ├── quickstart.rst │ └── ray_debug_tutorial.rst └── workers │ ├── fsdp_workers.rst │ ├── megatron_workers.rst │ ├── ray_trainer.rst │ └── sglang_worker.rst ├── download_qwen25omni.py ├── examples ├── data_preprocess │ ├── alr.py │ ├── alr_audio.py │ ├── alr_dataset.py │ ├── alr_text.py │ ├── full_hh_rlhf.py │ ├── geo3k.py │ ├── gsm8k.py │ ├── gsm8k_multiturn_w_tool.py │ ├── hellaswag.py │ ├── math_dataset.py │ └── multiturn.py ├── generation │ ├── run_deepseek7b_mutli_node.sh │ └── run_deepseek_v2_lite_math.sh ├── grpo_trainer │ ├── run_deepseek7b_llm.sh │ ├── run_deepseek7b_llm_math.sh │ ├── run_deepseek7b_llm_math_megatron.sh │ ├── run_deepseek7b_llm_seq_balance.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_sgl_megatron.sh │ ├── run_qwen2_5-7b_math_megatron_diff_tp.sh │ ├── run_qwen2_5_vl-7b.sh │ ├── run_qwen3-8b.sh │ └── run_qwen3moe-30b_megatron.sh ├── ppo_trainer │ ├── naive_chat_scheduler.py │ ├── run_deepseek7b_llm.sh │ ├── run_deepseek7b_llm_modelscope.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_gemma.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_seq_balance.sh │ ├── run_qwen2-7b_sglang_seq_balance.sh │ └── run_qwen2.5-32b.sh ├── 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_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 │ │ ├── gsm8k_multiturn_grpo.yaml │ │ ├── gsm8k_multiturn_megatron_grpo.yaml │ │ └── tool_config │ │ │ └── gsm8k_tool_config.yaml │ ├── run_qwen2.5-3b_gsm8k_multiturn.sh │ ├── run_qwen2.5-3b_gsm8k_multiturn_4xgpu.sh │ └── run_qwen2.5-3b_megatron_gsm8k_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 │ ├── 14b │ └── qwen2_14b_grpo_4_h800_fsdp_vllm.sh │ ├── 32b │ └── qwen2_32B_grpo_8_h20_megatron_vllm.sh │ ├── 70b │ ├── qwen2-70b_grpo_32_h20_fsdp_vllm.sh │ └── qwen2-70b_grpo_32_h800_fsdp_vllm.sh │ └── 7b │ └── qwen2-7b_grpo_2_h800_fsdp_vllm.sh ├── figs ├── f1.png └── f2.png ├── main_grpo.sh ├── pyproject.toml ├── qwen_transformers_example.py ├── recipe ├── README.md ├── dapo │ ├── README.md │ ├── config │ │ └── dapo_trainer.yaml │ ├── dapo_ray_trainer.py │ ├── main_dapo.py │ ├── prepare_dapo_data.sh │ ├── run_dapo_early_qwen2.5_32b.sh │ ├── run_dapo_qwen2.5_32b.sh │ ├── run_dapo_wo_ds_qwen2.5_32b.sh │ ├── test_dapo_7b.sh │ ├── test_dapo_7b_math.sh │ └── test_dapo_7b_math_megatron.sh ├── drgrpo │ └── README.md ├── prime │ ├── __init__.py │ ├── config │ │ └── prime_trainer.yaml │ ├── main_prime.py │ ├── prime_core_algos.py │ ├── prime_dp_rm.py │ ├── prime_fsdp_workers.py │ ├── prime_ray_trainer.py │ └── run_prime_qwen.sh ├── r1 │ ├── README.md │ ├── __init__.py │ ├── config │ │ └── evaluation.yaml │ ├── data_process.py │ ├── main_eval.py │ ├── reward_score.py │ ├── run_r1_distill_qwen.sh │ └── tasks │ │ ├── __init__.py │ │ ├── gpqa.py │ │ ├── livecodebench.py │ │ └── math.py ├── spin │ ├── README.md │ ├── config │ │ └── spin_trainer.yaml │ ├── core_algos.py │ ├── dp_actor.py │ ├── fsdp_workers.py │ ├── main_spin.py │ ├── run_spin.sh │ └── spin_trainer.py └── sppo │ ├── README.md │ ├── __init__.py │ ├── config │ └── sppo_trainer.yaml │ ├── dp_actor.py │ ├── main_sppo.py │ ├── run_qwen2.5-7b_rm.sh │ ├── sppo_ray_trainer.py │ └── sppo_worker.py ├── requirements-npu.txt ├── requirements.txt ├── requirements_sglang.txt ├── rl-scripts ├── curriculum.sh ├── format.sh ├── train_grpo_4gpu_7Binstruct.sh ├── train_ppo_3B_4gpu.sh ├── train_ppo_7B_4gpu.sh └── train_reinforce_plus_4gpu_7Binstruct.sh ├── scripts ├── converter_hf_to_mcore.py ├── diagnose.py ├── install_vllm_sglang_mcore.sh └── model_merger.py ├── setup.py ├── tests ├── __init__.py ├── distributed │ ├── run_all.sh │ └── test_tensor_dict.py ├── e2e │ ├── __init__.py │ ├── arithmetic_sequence │ │ ├── data │ │ │ ├── create_dataset.py │ │ │ ├── test.parquet │ │ │ └── train.parquet │ │ ├── model │ │ │ ├── config.json │ │ │ ├── create_model_tokenizer.py │ │ │ ├── generation_config.json │ │ │ ├── model.safetensors │ │ │ └── tokenizer_config.json │ │ └── rl │ │ │ ├── README.md │ │ │ └── main_trainer.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 │ │ ├── run_function_reward.sh │ │ └── run_model_reward.sh │ ├── run_dapo.sh │ ├── run_gsm8k_fsdp_sgl_multiturn_w_tool.sh │ ├── run_ppo_trainer_megatron.sh │ ├── run_prime.sh │ ├── run_r1_distill_qwen_aime24_eval.sh │ ├── run_ray_trainer.sh │ ├── run_ray_trainer_fire_sampling.sh │ ├── run_ray_trainer_rmpad.sh │ ├── run_spin.sh │ ├── run_sppo.sh │ ├── run_test.sh │ └── sft │ │ ├── run_sft.sh │ │ └── test_sp_loss_match.py ├── gpu_utility │ ├── test_memory_buffers.py │ ├── test_ops.py │ └── test_torch_functional.py ├── kernels │ └── test_linear_cross_entropy.py ├── kill_github_tests.sh ├── models │ ├── test_transformer.py │ └── test_transformers_ulysses.py ├── npu │ ├── run_qwen2_5_05b_grpo.sh │ ├── run_qwen2_5_32b_grpo.sh │ └── run_qwen2_5_7b_grpo.sh ├── ray_cpu │ ├── check_worker_alive │ │ └── main.py │ ├── test_auto_padding.py │ ├── test_check_worker_alive.py │ ├── test_decorator.py │ ├── test_fused_workers.py │ └── test_ray_local_envs.py ├── ray_gpu │ ├── detached_worker │ │ ├── README.md │ │ ├── client.py │ │ ├── run.sh │ │ └── server.py │ ├── test_colocated_workers.py │ ├── test_colocated_workers_fused.py │ ├── test_data_transfer.py │ ├── test_driverfunc_to_worker.py │ ├── test_high_level_scheduling_api.py │ ├── test_rvdz.py │ ├── test_worker_group_basics.py │ └── test_worker_group_torch.py ├── reward_score │ └── test_sandbox_fusion.py ├── sandbox │ └── test_sandbox.py ├── sanity │ ├── check_license.py │ └── test_import.py ├── single_controller │ └── base │ │ └── test_decorator.py ├── test_protocol.py ├── trainer │ ├── __init__.py │ └── ppo │ │ ├── __init__.py │ │ └── test_metric_utils.py ├── utils │ ├── cpu_tests │ │ ├── test_fs.py │ │ ├── test_import_utils.py │ │ ├── test_model.py │ │ ├── test_module.py │ │ └── test_timeout_decorator.py │ └── gpu_tests │ │ ├── checkpoint │ │ └── test_fsdp_ckpt.py │ │ ├── dataset │ │ ├── test_multiturn_sft_dataset.py │ │ ├── test_rl_dataset.py │ │ ├── test_rm_dataset.py │ │ └── test_sft_dataset.py │ │ ├── test_activation_offload.py │ │ ├── test_flops_counter.py │ │ ├── test_seqlen_balancing.py │ │ └── test_torch_functional.py └── workers │ └── rollout │ ├── async_rollout_utils.py │ ├── run_fsdp_vllm.py │ ├── test_hf_rollout.py │ ├── test_sglang_async_rollout_w_tools.py │ ├── test_sglang_async_spmd.py │ ├── test_sglang_spmd.py │ ├── test_vllm_hf_loader.py │ ├── test_vllm_multi_turn.py │ ├── test_vllm_spmd.py │ ├── test_vllm_tool_calling.py │ └── utils_sglang.py └── verl ├── __init__.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 │ ├── model_forward.py │ ├── model_initializer.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 │ ├── llama.py │ ├── monkey_patch.py │ ├── qwen2.py │ ├── qwen2_5_vl.py │ └── qwen2_vl.py └── weight_loader_registry.py ├── protocol.py ├── 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 │ ├── vllm_v_0_5_4 │ ├── __init__.py │ ├── arg_utils.py │ ├── config.py │ ├── dtensor_weight_loaders.py │ ├── hf_weight_loader.py │ ├── llm.py │ ├── llm_engine_sp.py │ ├── megatron_weight_loaders.py │ ├── model_loader.py │ ├── model_runner.py │ ├── parallel_state.py │ ├── spmd_gpu_executor.py │ ├── tokenizer.py │ └── worker.py │ └── vllm_v_0_6_3 │ ├── __init__.py │ ├── arg_utils.py │ ├── config.py │ ├── dtensor_weight_loaders.py │ ├── hf_weight_loader.py │ ├── llm.py │ ├── llm_engine_sp.py │ ├── megatron_weight_loaders.py │ ├── model_loader.py │ ├── model_runner.py │ ├── parallel_state.py │ ├── spmd_gpu_executor.py │ ├── tokenizer.py │ └── worker.py ├── tools ├── __init__.py ├── base_tool.py ├── gsm8k_tool.py └── schemas.py ├── trainer ├── __init__.py ├── config │ ├── evaluation.yaml │ ├── generation.yaml │ ├── ppo_megatron_trainer.yaml │ ├── ppo_trainer.yaml │ └── sft_trainer.yaml ├── 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 │ ├── audio_utils.py │ ├── multiturn_sft_dataset.py │ ├── rl_dataset.py │ ├── rm_dataset.py │ ├── sft_dataset.py │ └── vision_utils.py ├── debug │ ├── __init__.py │ ├── performance.py │ ├── profile.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 ├── logger │ ├── __init__.py │ └── aggregate_logger.py ├── logging_utils.py ├── megatron │ ├── __init__.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 ├── py_functional.py ├── ray_utils.py ├── rendezvous │ ├── __init__.py │ └── ray_backend.py ├── reward_score │ ├── __init__.py │ ├── alr.py │ ├── geo3k.py │ ├── gsm8k.py │ ├── math.py │ ├── math_batch.py │ ├── math_dapo.py │ ├── math_verify.py │ ├── prime_code │ │ ├── __init__.py │ │ ├── testing_util.py │ │ └── utils.py │ ├── prime_math │ │ ├── __init__.py │ │ ├── grader.py │ │ └── math_normalize.py │ └── sandbox_fusion │ │ ├── __init__.py │ │ └── utils.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 ├── naive.py └── prime.py ├── reward_model ├── __init__.py ├── base.py └── megatron │ ├── __init__.py │ └── reward_model.py ├── rollout ├── __init__.py ├── async_server.py ├── base.py ├── hf_rollout.py ├── naive │ ├── __init__.py │ └── naive_rollout.py ├── schemas.py ├── sglang_rollout │ ├── __init__.py │ ├── async_sglang_rollout.py │ ├── sglang_rollout.py │ └── utils.py ├── tokenizer.py └── vllm_rollout │ ├── __init__.py │ ├── fire_vllm_rollout.py │ ├── vllm_async_server.py │ ├── vllm_rollout.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 /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/README.md -------------------------------------------------------------------------------- /dataset-annotation-json/dev.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/dataset-annotation-json/dev.jsonl -------------------------------------------------------------------------------- /dataset-annotation-json/test.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/dataset-annotation-json/test.jsonl -------------------------------------------------------------------------------- /dataset-annotation-json/train.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/dataset-annotation-json/train.jsonl -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/README_vllm0.7.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/README_vllm0.7.md -------------------------------------------------------------------------------- /docs/README_vllm0.8.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/README_vllm0.8.md -------------------------------------------------------------------------------- /docs/_static/js/runllm-widget.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/_static/js/runllm-widget.js -------------------------------------------------------------------------------- /docs/_static/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/_static/logo.png -------------------------------------------------------------------------------- /docs/advance/checkpoint.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/advance/checkpoint.rst -------------------------------------------------------------------------------- /docs/advance/dpo_extension.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/advance/dpo_extension.rst -------------------------------------------------------------------------------- /docs/advance/fsdp_extension.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/advance/fsdp_extension.rst -------------------------------------------------------------------------------- /docs/advance/megatron_extension.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/advance/megatron_extension.rst -------------------------------------------------------------------------------- /docs/advance/placement.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/advance/placement.rst -------------------------------------------------------------------------------- /docs/advance/rope.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/advance/rope.rst -------------------------------------------------------------------------------- /docs/amd_tutorial/amd_build_dockerfile_page.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/amd_tutorial/amd_build_dockerfile_page.rst -------------------------------------------------------------------------------- /docs/amd_tutorial/amd_vllm_page.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/amd_tutorial/amd_vllm_page.rst -------------------------------------------------------------------------------- /docs/api/data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/api/data.rst -------------------------------------------------------------------------------- /docs/api/single_controller.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/api/single_controller.rst -------------------------------------------------------------------------------- /docs/api/trainer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/api/trainer.rst -------------------------------------------------------------------------------- /docs/api/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/api/utils.rst -------------------------------------------------------------------------------- /docs/ascend/ascend_vllm073.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/ascend/ascend_vllm073.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/examples/config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/examples/config.rst -------------------------------------------------------------------------------- /docs/examples/gsm8k_example.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/examples/gsm8k_example.rst -------------------------------------------------------------------------------- /docs/examples/multi_modal_example.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/examples/multi_modal_example.rst -------------------------------------------------------------------------------- /docs/examples/ppo_code_architecture.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/examples/ppo_code_architecture.rst -------------------------------------------------------------------------------- /docs/examples/sandbox_fusion_example.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/examples/sandbox_fusion_example.rst -------------------------------------------------------------------------------- /docs/experiment/ppo.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/experiment/ppo.rst -------------------------------------------------------------------------------- /docs/faq/faq.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/faq/faq.rst -------------------------------------------------------------------------------- /docs/hybrid_flow.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/hybrid_flow.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/perf/device_tuning.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/perf/device_tuning.rst -------------------------------------------------------------------------------- /docs/perf/perf_tuning.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/perf/perf_tuning.rst -------------------------------------------------------------------------------- /docs/preparation/prepare_data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/preparation/prepare_data.rst -------------------------------------------------------------------------------- /docs/preparation/reward_function.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/preparation/reward_function.rst -------------------------------------------------------------------------------- /docs/requirements-docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/requirements-docs.txt -------------------------------------------------------------------------------- /docs/sglang_multiturn/multiturn.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/sglang_multiturn/multiturn.rst -------------------------------------------------------------------------------- /docs/single_controller.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/single_controller.rst -------------------------------------------------------------------------------- /docs/start/install.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/start/install.rst -------------------------------------------------------------------------------- /docs/start/multinode.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/start/multinode.rst -------------------------------------------------------------------------------- /docs/start/quickstart.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/start/quickstart.rst -------------------------------------------------------------------------------- /docs/start/ray_debug_tutorial.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/start/ray_debug_tutorial.rst -------------------------------------------------------------------------------- /docs/workers/fsdp_workers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/workers/fsdp_workers.rst -------------------------------------------------------------------------------- /docs/workers/megatron_workers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/workers/megatron_workers.rst -------------------------------------------------------------------------------- /docs/workers/ray_trainer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/workers/ray_trainer.rst -------------------------------------------------------------------------------- /docs/workers/sglang_worker.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/docs/workers/sglang_worker.rst -------------------------------------------------------------------------------- /download_qwen25omni.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/download_qwen25omni.py -------------------------------------------------------------------------------- /examples/data_preprocess/alr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/alr.py -------------------------------------------------------------------------------- /examples/data_preprocess/alr_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/alr_audio.py -------------------------------------------------------------------------------- /examples/data_preprocess/alr_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/alr_dataset.py -------------------------------------------------------------------------------- /examples/data_preprocess/alr_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/alr_text.py -------------------------------------------------------------------------------- /examples/data_preprocess/full_hh_rlhf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/full_hh_rlhf.py -------------------------------------------------------------------------------- /examples/data_preprocess/geo3k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/geo3k.py -------------------------------------------------------------------------------- /examples/data_preprocess/gsm8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/gsm8k.py -------------------------------------------------------------------------------- /examples/data_preprocess/gsm8k_multiturn_w_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/gsm8k_multiturn_w_tool.py -------------------------------------------------------------------------------- /examples/data_preprocess/hellaswag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/hellaswag.py -------------------------------------------------------------------------------- /examples/data_preprocess/math_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/math_dataset.py -------------------------------------------------------------------------------- /examples/data_preprocess/multiturn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/data_preprocess/multiturn.py -------------------------------------------------------------------------------- /examples/generation/run_deepseek7b_mutli_node.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/generation/run_deepseek7b_mutli_node.sh -------------------------------------------------------------------------------- /examples/generation/run_deepseek_v2_lite_math.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/generation/run_deepseek_v2_lite_math.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek7b_llm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_deepseek7b_llm.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek7b_llm_math.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_deepseek7b_llm_math.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek7b_llm_math_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_deepseek7b_llm_math_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek7b_llm_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_deepseek7b_llm_seq_balance.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_qwen2-7b.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b_math.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_qwen2-7b_math.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b_math_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_qwen2-7b_math_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_qwen2-7b_seq_balance.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b_sgl_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_qwen2-7b_sgl_megatron.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5-7b_math_megatron_diff_tp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_qwen2_5-7b_math_megatron_diff_tp.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2_5_vl-7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_qwen2_5_vl-7b.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen3-8b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_qwen3-8b.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen3moe-30b_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/grpo_trainer/run_qwen3moe-30b_megatron.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/naive_chat_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/naive_chat_scheduler.py -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_deepseek7b_llm.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm_modelscope.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_deepseek7b_llm_modelscope.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm_sandbox_fusion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_deepseek7b_llm_sandbox_fusion.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm_sp2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_deepseek7b_llm_sp2.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek_full_hh_rlhf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_deepseek_full_hh_rlhf.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek_math_gsm8k_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_deepseek_math_gsm8k_megatron.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_gemma.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_gemma.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen1.5_moe_a2.7b-gsm8k_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/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/xid32/SoundMind/HEAD/examples/ppo_trainer/run_qwen2-7b_math_gsm8k_megatron.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_rm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_qwen2-7b_rm.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_rm_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_qwen2-7b_rm_seq_balance.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_qwen2-7b_seq_balance.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_sglang_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_qwen2-7b_sglang_seq_balance.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2.5-32b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ppo_trainer/run_qwen2.5-32b.sh -------------------------------------------------------------------------------- /examples/ray/tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/ray/tutorial.ipynb -------------------------------------------------------------------------------- /examples/reinforce_plus_plus_trainer/run_qwen2-7b_math_rf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/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/xid32/SoundMind/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/xid32/SoundMind/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/xid32/SoundMind/HEAD/examples/remax_trainer/run_qwen2.5-7b_seq_balance.sh -------------------------------------------------------------------------------- /examples/rloo_trainer/run_qwen2-7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/rloo_trainer/run_qwen2-7b.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_deepseek_6b7.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sft/gsm8k/run_deepseek_6b7.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_gemma_2b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sft/gsm8k/run_gemma_2b.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_gemma_7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sft/gsm8k/run_gemma_7b.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_qwen_05_peft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sft/gsm8k/run_qwen_05_peft.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_qwen_05_sp2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sft/gsm8k/run_qwen_05_sp2.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_qwen_05_sp2_liger.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sft/gsm8k/run_qwen_05_sp2_liger.sh -------------------------------------------------------------------------------- /examples/sft/multiturn/run_qwen_05_sp2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sft/multiturn/run_qwen_05_sp2.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sglang_multiturn/README.md -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/gsm8k_multiturn_grpo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sglang_multiturn/config/gsm8k_multiturn_grpo.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/gsm8k_multiturn_megatron_grpo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sglang_multiturn/config/gsm8k_multiturn_megatron_grpo.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/config/tool_config/gsm8k_tool_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sglang_multiturn/config/tool_config/gsm8k_tool_config.yaml -------------------------------------------------------------------------------- /examples/sglang_multiturn/run_qwen2.5-3b_gsm8k_multiturn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/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/xid32/SoundMind/HEAD/examples/sglang_multiturn/run_qwen2.5-3b_gsm8k_multiturn_4xgpu.sh -------------------------------------------------------------------------------- /examples/sglang_multiturn/run_qwen2.5-3b_megatron_gsm8k_multiturn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/sglang_multiturn/run_qwen2.5-3b_megatron_gsm8k_multiturn.sh -------------------------------------------------------------------------------- /examples/slurm/ray_on_slurm.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/slurm/ray_on_slurm.slurm -------------------------------------------------------------------------------- /examples/split_placement/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/split_placement/README.md -------------------------------------------------------------------------------- /examples/split_placement/config/ppo_trainer_split.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/split_placement/config/ppo_trainer_split.yaml -------------------------------------------------------------------------------- /examples/split_placement/main_ppo_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/split_placement/main_ppo_split.py -------------------------------------------------------------------------------- /examples/split_placement/run_deepseek7b_llm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/split_placement/run_deepseek7b_llm.sh -------------------------------------------------------------------------------- /examples/split_placement/split_monkey_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/split_placement/split_monkey_patch.py -------------------------------------------------------------------------------- /examples/tuning/14b/qwen2_14b_grpo_4_h800_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/tuning/14b/qwen2_14b_grpo_4_h800_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/32b/qwen2_32B_grpo_8_h20_megatron_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/tuning/32b/qwen2_32B_grpo_8_h20_megatron_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/70b/qwen2-70b_grpo_32_h20_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/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/xid32/SoundMind/HEAD/examples/tuning/70b/qwen2-70b_grpo_32_h800_fsdp_vllm.sh -------------------------------------------------------------------------------- /examples/tuning/7b/qwen2-7b_grpo_2_h800_fsdp_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/examples/tuning/7b/qwen2-7b_grpo_2_h800_fsdp_vllm.sh -------------------------------------------------------------------------------- /figs/f1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/figs/f1.png -------------------------------------------------------------------------------- /figs/f2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/figs/f2.png -------------------------------------------------------------------------------- /main_grpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/main_grpo.sh -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/pyproject.toml -------------------------------------------------------------------------------- /qwen_transformers_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/qwen_transformers_example.py -------------------------------------------------------------------------------- /recipe/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/README.md -------------------------------------------------------------------------------- /recipe/dapo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/README.md -------------------------------------------------------------------------------- /recipe/dapo/config/dapo_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/config/dapo_trainer.yaml -------------------------------------------------------------------------------- /recipe/dapo/dapo_ray_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/dapo_ray_trainer.py -------------------------------------------------------------------------------- /recipe/dapo/main_dapo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/main_dapo.py -------------------------------------------------------------------------------- /recipe/dapo/prepare_dapo_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/prepare_dapo_data.sh -------------------------------------------------------------------------------- /recipe/dapo/run_dapo_early_qwen2.5_32b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/run_dapo_early_qwen2.5_32b.sh -------------------------------------------------------------------------------- /recipe/dapo/run_dapo_qwen2.5_32b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/run_dapo_qwen2.5_32b.sh -------------------------------------------------------------------------------- /recipe/dapo/run_dapo_wo_ds_qwen2.5_32b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/run_dapo_wo_ds_qwen2.5_32b.sh -------------------------------------------------------------------------------- /recipe/dapo/test_dapo_7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/test_dapo_7b.sh -------------------------------------------------------------------------------- /recipe/dapo/test_dapo_7b_math.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/test_dapo_7b_math.sh -------------------------------------------------------------------------------- /recipe/dapo/test_dapo_7b_math_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/dapo/test_dapo_7b_math_megatron.sh -------------------------------------------------------------------------------- /recipe/drgrpo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/drgrpo/README.md -------------------------------------------------------------------------------- /recipe/prime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/prime/__init__.py -------------------------------------------------------------------------------- /recipe/prime/config/prime_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/prime/config/prime_trainer.yaml -------------------------------------------------------------------------------- /recipe/prime/main_prime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/prime/main_prime.py -------------------------------------------------------------------------------- /recipe/prime/prime_core_algos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/prime/prime_core_algos.py -------------------------------------------------------------------------------- /recipe/prime/prime_dp_rm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/prime/prime_dp_rm.py -------------------------------------------------------------------------------- /recipe/prime/prime_fsdp_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/prime/prime_fsdp_workers.py -------------------------------------------------------------------------------- /recipe/prime/prime_ray_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/prime/prime_ray_trainer.py -------------------------------------------------------------------------------- /recipe/prime/run_prime_qwen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/prime/run_prime_qwen.sh -------------------------------------------------------------------------------- /recipe/r1/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/README.md -------------------------------------------------------------------------------- /recipe/r1/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/__init__.py -------------------------------------------------------------------------------- /recipe/r1/config/evaluation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/config/evaluation.yaml -------------------------------------------------------------------------------- /recipe/r1/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/data_process.py -------------------------------------------------------------------------------- /recipe/r1/main_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/main_eval.py -------------------------------------------------------------------------------- /recipe/r1/reward_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/reward_score.py -------------------------------------------------------------------------------- /recipe/r1/run_r1_distill_qwen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/run_r1_distill_qwen.sh -------------------------------------------------------------------------------- /recipe/r1/tasks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/tasks/__init__.py -------------------------------------------------------------------------------- /recipe/r1/tasks/gpqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/tasks/gpqa.py -------------------------------------------------------------------------------- /recipe/r1/tasks/livecodebench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/tasks/livecodebench.py -------------------------------------------------------------------------------- /recipe/r1/tasks/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/r1/tasks/math.py -------------------------------------------------------------------------------- /recipe/spin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/spin/README.md -------------------------------------------------------------------------------- /recipe/spin/config/spin_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/spin/config/spin_trainer.yaml -------------------------------------------------------------------------------- /recipe/spin/core_algos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/spin/core_algos.py -------------------------------------------------------------------------------- /recipe/spin/dp_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/spin/dp_actor.py -------------------------------------------------------------------------------- /recipe/spin/fsdp_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/spin/fsdp_workers.py -------------------------------------------------------------------------------- /recipe/spin/main_spin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/spin/main_spin.py -------------------------------------------------------------------------------- /recipe/spin/run_spin.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/spin/run_spin.sh -------------------------------------------------------------------------------- /recipe/spin/spin_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/spin/spin_trainer.py -------------------------------------------------------------------------------- /recipe/sppo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/sppo/README.md -------------------------------------------------------------------------------- /recipe/sppo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/sppo/__init__.py -------------------------------------------------------------------------------- /recipe/sppo/config/sppo_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/sppo/config/sppo_trainer.yaml -------------------------------------------------------------------------------- /recipe/sppo/dp_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/sppo/dp_actor.py -------------------------------------------------------------------------------- /recipe/sppo/main_sppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/sppo/main_sppo.py -------------------------------------------------------------------------------- /recipe/sppo/run_qwen2.5-7b_rm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/sppo/run_qwen2.5-7b_rm.sh -------------------------------------------------------------------------------- /recipe/sppo/sppo_ray_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/sppo/sppo_ray_trainer.py -------------------------------------------------------------------------------- /recipe/sppo/sppo_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/recipe/sppo/sppo_worker.py -------------------------------------------------------------------------------- /requirements-npu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/requirements-npu.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_sglang.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/requirements_sglang.txt -------------------------------------------------------------------------------- /rl-scripts/curriculum.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/rl-scripts/curriculum.sh -------------------------------------------------------------------------------- /rl-scripts/format.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/rl-scripts/format.sh -------------------------------------------------------------------------------- /rl-scripts/train_grpo_4gpu_7Binstruct.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/rl-scripts/train_grpo_4gpu_7Binstruct.sh -------------------------------------------------------------------------------- /rl-scripts/train_ppo_3B_4gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/rl-scripts/train_ppo_3B_4gpu.sh -------------------------------------------------------------------------------- /rl-scripts/train_ppo_7B_4gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/rl-scripts/train_ppo_7B_4gpu.sh -------------------------------------------------------------------------------- /rl-scripts/train_reinforce_plus_4gpu_7Binstruct.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/rl-scripts/train_reinforce_plus_4gpu_7Binstruct.sh -------------------------------------------------------------------------------- /scripts/converter_hf_to_mcore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/scripts/converter_hf_to_mcore.py -------------------------------------------------------------------------------- /scripts/diagnose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/scripts/diagnose.py -------------------------------------------------------------------------------- /scripts/install_vllm_sglang_mcore.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/scripts/install_vllm_sglang_mcore.sh -------------------------------------------------------------------------------- /scripts/model_merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/scripts/model_merger.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/distributed/run_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/distributed/run_all.sh -------------------------------------------------------------------------------- /tests/distributed/test_tensor_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/distributed/test_tensor_dict.py -------------------------------------------------------------------------------- /tests/e2e/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/__init__.py -------------------------------------------------------------------------------- /tests/e2e/arithmetic_sequence/data/create_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/arithmetic_sequence/data/create_dataset.py -------------------------------------------------------------------------------- /tests/e2e/arithmetic_sequence/data/test.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/arithmetic_sequence/data/test.parquet -------------------------------------------------------------------------------- /tests/e2e/arithmetic_sequence/data/train.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/arithmetic_sequence/data/train.parquet -------------------------------------------------------------------------------- /tests/e2e/arithmetic_sequence/model/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/arithmetic_sequence/model/config.json -------------------------------------------------------------------------------- /tests/e2e/arithmetic_sequence/model/create_model_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/arithmetic_sequence/model/create_model_tokenizer.py -------------------------------------------------------------------------------- /tests/e2e/arithmetic_sequence/model/generation_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/arithmetic_sequence/model/generation_config.json -------------------------------------------------------------------------------- /tests/e2e/arithmetic_sequence/model/model.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/arithmetic_sequence/model/model.safetensors -------------------------------------------------------------------------------- /tests/e2e/arithmetic_sequence/model/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/arithmetic_sequence/model/tokenizer_config.json -------------------------------------------------------------------------------- /tests/e2e/arithmetic_sequence/rl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/arithmetic_sequence/rl/README.md -------------------------------------------------------------------------------- /tests/e2e/arithmetic_sequence/rl/main_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/arithmetic_sequence/rl/main_trainer.py -------------------------------------------------------------------------------- /tests/e2e/check_custom_rwd_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/check_custom_rwd_fn.py -------------------------------------------------------------------------------- /tests/e2e/check_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/check_results.py -------------------------------------------------------------------------------- /tests/e2e/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/envs/__init__.py -------------------------------------------------------------------------------- /tests/e2e/envs/digit_completion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/envs/digit_completion/__init__.py -------------------------------------------------------------------------------- /tests/e2e/envs/digit_completion/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/envs/digit_completion/task.py -------------------------------------------------------------------------------- /tests/e2e/envs/digit_completion/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/envs/digit_completion/tokenizer.py -------------------------------------------------------------------------------- /tests/e2e/generation/run_gen_qwen05.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/generation/run_gen_qwen05.sh -------------------------------------------------------------------------------- /tests/e2e/ppo_trainer/run_function_reward.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/ppo_trainer/run_function_reward.sh -------------------------------------------------------------------------------- /tests/e2e/ppo_trainer/run_model_reward.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/ppo_trainer/run_model_reward.sh -------------------------------------------------------------------------------- /tests/e2e/run_dapo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_dapo.sh -------------------------------------------------------------------------------- /tests/e2e/run_gsm8k_fsdp_sgl_multiturn_w_tool.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_gsm8k_fsdp_sgl_multiturn_w_tool.sh -------------------------------------------------------------------------------- /tests/e2e/run_ppo_trainer_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_ppo_trainer_megatron.sh -------------------------------------------------------------------------------- /tests/e2e/run_prime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_prime.sh -------------------------------------------------------------------------------- /tests/e2e/run_r1_distill_qwen_aime24_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_r1_distill_qwen_aime24_eval.sh -------------------------------------------------------------------------------- /tests/e2e/run_ray_trainer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_ray_trainer.sh -------------------------------------------------------------------------------- /tests/e2e/run_ray_trainer_fire_sampling.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_ray_trainer_fire_sampling.sh -------------------------------------------------------------------------------- /tests/e2e/run_ray_trainer_rmpad.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_ray_trainer_rmpad.sh -------------------------------------------------------------------------------- /tests/e2e/run_spin.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_spin.sh -------------------------------------------------------------------------------- /tests/e2e/run_sppo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_sppo.sh -------------------------------------------------------------------------------- /tests/e2e/run_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/run_test.sh -------------------------------------------------------------------------------- /tests/e2e/sft/run_sft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/sft/run_sft.sh -------------------------------------------------------------------------------- /tests/e2e/sft/test_sp_loss_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/e2e/sft/test_sp_loss_match.py -------------------------------------------------------------------------------- /tests/gpu_utility/test_memory_buffers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/gpu_utility/test_memory_buffers.py -------------------------------------------------------------------------------- /tests/gpu_utility/test_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/gpu_utility/test_ops.py -------------------------------------------------------------------------------- /tests/gpu_utility/test_torch_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/gpu_utility/test_torch_functional.py -------------------------------------------------------------------------------- /tests/kernels/test_linear_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/kernels/test_linear_cross_entropy.py -------------------------------------------------------------------------------- /tests/kill_github_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/kill_github_tests.sh -------------------------------------------------------------------------------- /tests/models/test_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/models/test_transformer.py -------------------------------------------------------------------------------- /tests/models/test_transformers_ulysses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/models/test_transformers_ulysses.py -------------------------------------------------------------------------------- /tests/npu/run_qwen2_5_05b_grpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/npu/run_qwen2_5_05b_grpo.sh -------------------------------------------------------------------------------- /tests/npu/run_qwen2_5_32b_grpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/npu/run_qwen2_5_32b_grpo.sh -------------------------------------------------------------------------------- /tests/npu/run_qwen2_5_7b_grpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/npu/run_qwen2_5_7b_grpo.sh -------------------------------------------------------------------------------- /tests/ray_cpu/check_worker_alive/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_cpu/check_worker_alive/main.py -------------------------------------------------------------------------------- /tests/ray_cpu/test_auto_padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_cpu/test_auto_padding.py -------------------------------------------------------------------------------- /tests/ray_cpu/test_check_worker_alive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_cpu/test_check_worker_alive.py -------------------------------------------------------------------------------- /tests/ray_cpu/test_decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_cpu/test_decorator.py -------------------------------------------------------------------------------- /tests/ray_cpu/test_fused_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_cpu/test_fused_workers.py -------------------------------------------------------------------------------- /tests/ray_cpu/test_ray_local_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_cpu/test_ray_local_envs.py -------------------------------------------------------------------------------- /tests/ray_gpu/detached_worker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/detached_worker/README.md -------------------------------------------------------------------------------- /tests/ray_gpu/detached_worker/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/detached_worker/client.py -------------------------------------------------------------------------------- /tests/ray_gpu/detached_worker/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/detached_worker/run.sh -------------------------------------------------------------------------------- /tests/ray_gpu/detached_worker/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/detached_worker/server.py -------------------------------------------------------------------------------- /tests/ray_gpu/test_colocated_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/test_colocated_workers.py -------------------------------------------------------------------------------- /tests/ray_gpu/test_colocated_workers_fused.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/test_colocated_workers_fused.py -------------------------------------------------------------------------------- /tests/ray_gpu/test_data_transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/test_data_transfer.py -------------------------------------------------------------------------------- /tests/ray_gpu/test_driverfunc_to_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/test_driverfunc_to_worker.py -------------------------------------------------------------------------------- /tests/ray_gpu/test_high_level_scheduling_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/test_high_level_scheduling_api.py -------------------------------------------------------------------------------- /tests/ray_gpu/test_rvdz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/test_rvdz.py -------------------------------------------------------------------------------- /tests/ray_gpu/test_worker_group_basics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/test_worker_group_basics.py -------------------------------------------------------------------------------- /tests/ray_gpu/test_worker_group_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/ray_gpu/test_worker_group_torch.py -------------------------------------------------------------------------------- /tests/reward_score/test_sandbox_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/reward_score/test_sandbox_fusion.py -------------------------------------------------------------------------------- /tests/sandbox/test_sandbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/sandbox/test_sandbox.py -------------------------------------------------------------------------------- /tests/sanity/check_license.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/sanity/check_license.py -------------------------------------------------------------------------------- /tests/sanity/test_import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/sanity/test_import.py -------------------------------------------------------------------------------- /tests/single_controller/base/test_decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/single_controller/base/test_decorator.py -------------------------------------------------------------------------------- /tests/test_protocol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/test_protocol.py -------------------------------------------------------------------------------- /tests/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/trainer/__init__.py -------------------------------------------------------------------------------- /tests/trainer/ppo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/trainer/ppo/__init__.py -------------------------------------------------------------------------------- /tests/trainer/ppo/test_metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/trainer/ppo/test_metric_utils.py -------------------------------------------------------------------------------- /tests/utils/cpu_tests/test_fs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/cpu_tests/test_fs.py -------------------------------------------------------------------------------- /tests/utils/cpu_tests/test_import_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/cpu_tests/test_import_utils.py -------------------------------------------------------------------------------- /tests/utils/cpu_tests/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/cpu_tests/test_model.py -------------------------------------------------------------------------------- /tests/utils/cpu_tests/test_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/cpu_tests/test_module.py -------------------------------------------------------------------------------- /tests/utils/cpu_tests/test_timeout_decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/cpu_tests/test_timeout_decorator.py -------------------------------------------------------------------------------- /tests/utils/gpu_tests/checkpoint/test_fsdp_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/gpu_tests/checkpoint/test_fsdp_ckpt.py -------------------------------------------------------------------------------- /tests/utils/gpu_tests/dataset/test_multiturn_sft_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/gpu_tests/dataset/test_multiturn_sft_dataset.py -------------------------------------------------------------------------------- /tests/utils/gpu_tests/dataset/test_rl_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/gpu_tests/dataset/test_rl_dataset.py -------------------------------------------------------------------------------- /tests/utils/gpu_tests/dataset/test_rm_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/gpu_tests/dataset/test_rm_dataset.py -------------------------------------------------------------------------------- /tests/utils/gpu_tests/dataset/test_sft_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/gpu_tests/dataset/test_sft_dataset.py -------------------------------------------------------------------------------- /tests/utils/gpu_tests/test_activation_offload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/gpu_tests/test_activation_offload.py -------------------------------------------------------------------------------- /tests/utils/gpu_tests/test_flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/gpu_tests/test_flops_counter.py -------------------------------------------------------------------------------- /tests/utils/gpu_tests/test_seqlen_balancing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/gpu_tests/test_seqlen_balancing.py -------------------------------------------------------------------------------- /tests/utils/gpu_tests/test_torch_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/utils/gpu_tests/test_torch_functional.py -------------------------------------------------------------------------------- /tests/workers/rollout/async_rollout_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/async_rollout_utils.py -------------------------------------------------------------------------------- /tests/workers/rollout/run_fsdp_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/run_fsdp_vllm.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_hf_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/test_hf_rollout.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_async_rollout_w_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/test_sglang_async_rollout_w_tools.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_async_spmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/test_sglang_async_spmd.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_sglang_spmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/test_sglang_spmd.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_vllm_hf_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/test_vllm_hf_loader.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_vllm_multi_turn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/test_vllm_multi_turn.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_vllm_spmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/test_vllm_spmd.py -------------------------------------------------------------------------------- /tests/workers/rollout/test_vllm_tool_calling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/test_vllm_tool_calling.py -------------------------------------------------------------------------------- /tests/workers/rollout/utils_sglang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/tests/workers/rollout/utils_sglang.py -------------------------------------------------------------------------------- /verl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/__init__.py -------------------------------------------------------------------------------- /verl/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/README.md -------------------------------------------------------------------------------- /verl/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/checkpoint_utils/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/llama_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/checkpoint_utils/llama_loader.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/llama_loader_depracated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/checkpoint_utils/llama_loader_depracated.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/llama_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/checkpoint_utils/llama_saver.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/layers/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/layers/parallel_attention.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/layers/parallel_decoder.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/layers/parallel_linear.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/layers/parallel_mlp.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/layers/parallel_rmsnorm.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/modeling_llama_megatron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/llama/megatron/modeling_llama_megatron.py -------------------------------------------------------------------------------- /verl/models/mcore/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/mcore/__init__.py -------------------------------------------------------------------------------- /verl/models/mcore/config_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/mcore/config_converter.py -------------------------------------------------------------------------------- /verl/models/mcore/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/mcore/loader.py -------------------------------------------------------------------------------- /verl/models/mcore/model_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/mcore/model_forward.py -------------------------------------------------------------------------------- /verl/models/mcore/model_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/mcore/model_initializer.py -------------------------------------------------------------------------------- /verl/models/mcore/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/mcore/readme.md -------------------------------------------------------------------------------- /verl/models/mcore/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/mcore/registry.py -------------------------------------------------------------------------------- /verl/models/mcore/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/mcore/saver.py -------------------------------------------------------------------------------- /verl/models/mcore/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/mcore/util.py -------------------------------------------------------------------------------- /verl/models/mcore/weight_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/mcore/weight_converter.py -------------------------------------------------------------------------------- /verl/models/qwen2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/__init__.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/__init__.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/checkpoint_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/checkpoint_utils/__init__.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader_depracated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader_depracated.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/checkpoint_utils/qwen2_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/checkpoint_utils/qwen2_saver.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/layers/__init__.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/parallel_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/layers/parallel_attention.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/parallel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/layers/parallel_decoder.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/parallel_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/layers/parallel_linear.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/parallel_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/layers/parallel_mlp.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/layers/parallel_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/layers/parallel_rmsnorm.py -------------------------------------------------------------------------------- /verl/models/qwen2/megatron/modeling_qwen2_megatron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/qwen2/megatron/modeling_qwen2_megatron.py -------------------------------------------------------------------------------- /verl/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/registry.py -------------------------------------------------------------------------------- /verl/models/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/transformers/__init__.py -------------------------------------------------------------------------------- /verl/models/transformers/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/transformers/llama.py -------------------------------------------------------------------------------- /verl/models/transformers/monkey_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/transformers/monkey_patch.py -------------------------------------------------------------------------------- /verl/models/transformers/qwen2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/transformers/qwen2.py -------------------------------------------------------------------------------- /verl/models/transformers/qwen2_5_vl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/transformers/qwen2_5_vl.py -------------------------------------------------------------------------------- /verl/models/transformers/qwen2_vl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/transformers/qwen2_vl.py -------------------------------------------------------------------------------- /verl/models/weight_loader_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/models/weight_loader_registry.py -------------------------------------------------------------------------------- /verl/protocol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/protocol.py -------------------------------------------------------------------------------- /verl/single_controller/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/base/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/base/decorator.py -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/base/megatron/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/base/megatron/worker.py -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/worker_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/base/megatron/worker_group.py -------------------------------------------------------------------------------- /verl/single_controller/base/register_center/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/base/register_center/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/register_center/ray.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/base/register_center/ray.py -------------------------------------------------------------------------------- /verl/single_controller/base/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/base/worker.py -------------------------------------------------------------------------------- /verl/single_controller/base/worker_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/base/worker_group.py -------------------------------------------------------------------------------- /verl/single_controller/ray/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/ray/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/ray/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/ray/base.py -------------------------------------------------------------------------------- /verl/single_controller/ray/megatron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/single_controller/ray/megatron.py -------------------------------------------------------------------------------- /verl/third_party/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/__init__.py -------------------------------------------------------------------------------- /verl/third_party/sglang/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/sglang/__init__.py -------------------------------------------------------------------------------- /verl/third_party/sglang/parallel_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/sglang/parallel_state.py -------------------------------------------------------------------------------- /verl/third_party/vllm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/__init__.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/__init__.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/arg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/arg_utils.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/config.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/dtensor_weight_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/dtensor_weight_loaders.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/hf_weight_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/hf_weight_loader.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/llm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/llm.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/llm_engine_sp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/llm_engine_sp.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/megatron_weight_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/megatron_weight_loaders.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/model_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/model_loader.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/model_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/model_runner.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/parallel_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/parallel_state.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/spmd_gpu_executor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/spmd_gpu_executor.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/tokenizer.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_5_4/worker.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__init__.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/arg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/arg_utils.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/config.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/dtensor_weight_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/dtensor_weight_loaders.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/hf_weight_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/hf_weight_loader.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/llm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/llm.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/llm_engine_sp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/llm_engine_sp.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/megatron_weight_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/megatron_weight_loaders.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/model_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/model_loader.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/model_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/model_runner.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/parallel_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/parallel_state.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/spmd_gpu_executor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/spmd_gpu_executor.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/tokenizer.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/third_party/vllm/vllm_v_0_6_3/worker.py -------------------------------------------------------------------------------- /verl/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/tools/__init__.py -------------------------------------------------------------------------------- /verl/tools/base_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/tools/base_tool.py -------------------------------------------------------------------------------- /verl/tools/gsm8k_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/tools/gsm8k_tool.py -------------------------------------------------------------------------------- /verl/tools/schemas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/tools/schemas.py -------------------------------------------------------------------------------- /verl/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/__init__.py -------------------------------------------------------------------------------- /verl/trainer/config/evaluation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/config/evaluation.yaml -------------------------------------------------------------------------------- /verl/trainer/config/generation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/config/generation.yaml -------------------------------------------------------------------------------- /verl/trainer/config/ppo_megatron_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/config/ppo_megatron_trainer.yaml -------------------------------------------------------------------------------- /verl/trainer/config/ppo_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/config/ppo_trainer.yaml -------------------------------------------------------------------------------- /verl/trainer/config/sft_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/config/sft_trainer.yaml -------------------------------------------------------------------------------- /verl/trainer/fsdp_sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/fsdp_sft_trainer.py -------------------------------------------------------------------------------- /verl/trainer/main_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/main_eval.py -------------------------------------------------------------------------------- /verl/trainer/main_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/main_generation.py -------------------------------------------------------------------------------- /verl/trainer/main_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/main_ppo.py -------------------------------------------------------------------------------- /verl/trainer/ppo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/ppo/__init__.py -------------------------------------------------------------------------------- /verl/trainer/ppo/core_algos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/ppo/core_algos.py -------------------------------------------------------------------------------- /verl/trainer/ppo/metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/ppo/metric_utils.py -------------------------------------------------------------------------------- /verl/trainer/ppo/ray_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/ppo/ray_trainer.py -------------------------------------------------------------------------------- /verl/trainer/ppo/reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/ppo/reward.py -------------------------------------------------------------------------------- /verl/trainer/runtime_env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/trainer/runtime_env.yaml -------------------------------------------------------------------------------- /verl/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/__init__.py -------------------------------------------------------------------------------- /verl/utils/activation_offload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/activation_offload.py -------------------------------------------------------------------------------- /verl/utils/checkpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/checkpoint/__init__.py -------------------------------------------------------------------------------- /verl/utils/checkpoint/checkpoint_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/checkpoint/checkpoint_manager.py -------------------------------------------------------------------------------- /verl/utils/checkpoint/fsdp_checkpoint_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/checkpoint/fsdp_checkpoint_manager.py -------------------------------------------------------------------------------- /verl/utils/checkpoint/megatron_checkpoint_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/checkpoint/megatron_checkpoint_manager.py -------------------------------------------------------------------------------- /verl/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/config.py -------------------------------------------------------------------------------- /verl/utils/dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/dataset/README.md -------------------------------------------------------------------------------- /verl/utils/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/dataset/__init__.py -------------------------------------------------------------------------------- /verl/utils/dataset/audio_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/dataset/audio_utils.py -------------------------------------------------------------------------------- /verl/utils/dataset/multiturn_sft_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/dataset/multiturn_sft_dataset.py -------------------------------------------------------------------------------- /verl/utils/dataset/rl_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/dataset/rl_dataset.py -------------------------------------------------------------------------------- /verl/utils/dataset/rm_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/dataset/rm_dataset.py -------------------------------------------------------------------------------- /verl/utils/dataset/sft_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/dataset/sft_dataset.py -------------------------------------------------------------------------------- /verl/utils/dataset/vision_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/dataset/vision_utils.py -------------------------------------------------------------------------------- /verl/utils/debug/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/debug/__init__.py -------------------------------------------------------------------------------- /verl/utils/debug/performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/debug/performance.py -------------------------------------------------------------------------------- /verl/utils/debug/profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/debug/profile.py -------------------------------------------------------------------------------- /verl/utils/debug/trajectory_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/debug/trajectory_tracker.py -------------------------------------------------------------------------------- /verl/utils/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/device.py -------------------------------------------------------------------------------- /verl/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/distributed.py -------------------------------------------------------------------------------- /verl/utils/experimental/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/experimental/__init__.py -------------------------------------------------------------------------------- /verl/utils/experimental/torch_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/experimental/torch_functional.py -------------------------------------------------------------------------------- /verl/utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/flops_counter.py -------------------------------------------------------------------------------- /verl/utils/fs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/fs.py -------------------------------------------------------------------------------- /verl/utils/fsdp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/fsdp_utils.py -------------------------------------------------------------------------------- /verl/utils/hdfs_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/hdfs_io.py -------------------------------------------------------------------------------- /verl/utils/import_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/import_utils.py -------------------------------------------------------------------------------- /verl/utils/logger/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/logger/__init__.py -------------------------------------------------------------------------------- /verl/utils/logger/aggregate_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/logger/aggregate_logger.py -------------------------------------------------------------------------------- /verl/utils/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/logging_utils.py -------------------------------------------------------------------------------- /verl/utils/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/megatron/__init__.py -------------------------------------------------------------------------------- /verl/utils/megatron/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/megatron/memory.py -------------------------------------------------------------------------------- /verl/utils/megatron/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/megatron/optimizer.py -------------------------------------------------------------------------------- /verl/utils/megatron/pipeline_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/megatron/pipeline_parallel.py -------------------------------------------------------------------------------- /verl/utils/megatron/sequence_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/megatron/sequence_parallel.py -------------------------------------------------------------------------------- /verl/utils/megatron/tensor_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/megatron/tensor_parallel.py -------------------------------------------------------------------------------- /verl/utils/megatron_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/megatron_utils.py -------------------------------------------------------------------------------- /verl/utils/memory_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/memory_buffer.py -------------------------------------------------------------------------------- /verl/utils/metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/metric/__init__.py -------------------------------------------------------------------------------- /verl/utils/metric/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/metric/utils.py -------------------------------------------------------------------------------- /verl/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/model.py -------------------------------------------------------------------------------- /verl/utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/net_utils.py -------------------------------------------------------------------------------- /verl/utils/py_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/py_functional.py -------------------------------------------------------------------------------- /verl/utils/ray_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/ray_utils.py -------------------------------------------------------------------------------- /verl/utils/rendezvous/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/rendezvous/__init__.py -------------------------------------------------------------------------------- /verl/utils/rendezvous/ray_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/rendezvous/ray_backend.py -------------------------------------------------------------------------------- /verl/utils/reward_score/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/__init__.py -------------------------------------------------------------------------------- /verl/utils/reward_score/alr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/alr.py -------------------------------------------------------------------------------- /verl/utils/reward_score/geo3k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/geo3k.py -------------------------------------------------------------------------------- /verl/utils/reward_score/gsm8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/gsm8k.py -------------------------------------------------------------------------------- /verl/utils/reward_score/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/math.py -------------------------------------------------------------------------------- /verl/utils/reward_score/math_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/math_batch.py -------------------------------------------------------------------------------- /verl/utils/reward_score/math_dapo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/math_dapo.py -------------------------------------------------------------------------------- /verl/utils/reward_score/math_verify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/math_verify.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_code/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/prime_code/__init__.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_code/testing_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/prime_code/testing_util.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_code/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/prime_code/utils.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_math/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/prime_math/__init__.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_math/grader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/prime_math/grader.py -------------------------------------------------------------------------------- /verl/utils/reward_score/prime_math/math_normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/prime_math/math_normalize.py -------------------------------------------------------------------------------- /verl/utils/reward_score/sandbox_fusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/sandbox_fusion/__init__.py -------------------------------------------------------------------------------- /verl/utils/reward_score/sandbox_fusion/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/reward_score/sandbox_fusion/utils.py -------------------------------------------------------------------------------- /verl/utils/seqlen_balancing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/seqlen_balancing.py -------------------------------------------------------------------------------- /verl/utils/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/tokenizer.py -------------------------------------------------------------------------------- /verl/utils/torch_dtypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/torch_dtypes.py -------------------------------------------------------------------------------- /verl/utils/torch_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/torch_functional.py -------------------------------------------------------------------------------- /verl/utils/tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/tracking.py -------------------------------------------------------------------------------- /verl/utils/ulysses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/ulysses.py -------------------------------------------------------------------------------- /verl/utils/vllm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/utils/vllm_utils.py -------------------------------------------------------------------------------- /verl/version/version: -------------------------------------------------------------------------------- 1 | 0.3.1.dev 2 | -------------------------------------------------------------------------------- /verl/workers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/__init__.py -------------------------------------------------------------------------------- /verl/workers/actor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/actor/__init__.py -------------------------------------------------------------------------------- /verl/workers/actor/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/actor/base.py -------------------------------------------------------------------------------- /verl/workers/actor/dp_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/actor/dp_actor.py -------------------------------------------------------------------------------- /verl/workers/actor/megatron_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/actor/megatron_actor.py -------------------------------------------------------------------------------- /verl/workers/critic/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/critic/__init__.py -------------------------------------------------------------------------------- /verl/workers/critic/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/critic/base.py -------------------------------------------------------------------------------- /verl/workers/critic/dp_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/critic/dp_critic.py -------------------------------------------------------------------------------- /verl/workers/critic/megatron_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/critic/megatron_critic.py -------------------------------------------------------------------------------- /verl/workers/fsdp_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/fsdp_workers.py -------------------------------------------------------------------------------- /verl/workers/megatron_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/megatron_workers.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/reward_manager/__init__.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/reward_manager/batch.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/dapo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/reward_manager/dapo.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/naive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/reward_manager/naive.py -------------------------------------------------------------------------------- /verl/workers/reward_manager/prime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/reward_manager/prime.py -------------------------------------------------------------------------------- /verl/workers/reward_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/reward_model/__init__.py -------------------------------------------------------------------------------- /verl/workers/reward_model/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/reward_model/base.py -------------------------------------------------------------------------------- /verl/workers/reward_model/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/reward_model/megatron/__init__.py -------------------------------------------------------------------------------- /verl/workers/reward_model/megatron/reward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/reward_model/megatron/reward_model.py -------------------------------------------------------------------------------- /verl/workers/rollout/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/async_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/async_server.py -------------------------------------------------------------------------------- /verl/workers/rollout/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/base.py -------------------------------------------------------------------------------- /verl/workers/rollout/hf_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/hf_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/naive/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/naive/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/naive/naive_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/naive/naive_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/schemas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/schemas.py -------------------------------------------------------------------------------- /verl/workers/rollout/sglang_rollout/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/sglang_rollout/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/sglang_rollout/async_sglang_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/sglang_rollout/async_sglang_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/sglang_rollout/sglang_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/sglang_rollout/sglang_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/sglang_rollout/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/sglang_rollout/utils.py -------------------------------------------------------------------------------- /verl/workers/rollout/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/tokenizer.py -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/vllm_rollout/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/fire_vllm_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/vllm_rollout/fire_vllm_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/vllm_async_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/vllm_rollout/vllm_async_server.py -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/vllm_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/vllm_rollout/vllm_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/vllm_rollout_spmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/rollout/vllm_rollout/vllm_rollout_spmd.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/sharding_manager/__init__.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/sharding_manager/base.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/fsdp_sglang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/sharding_manager/fsdp_sglang.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/fsdp_ulysses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/sharding_manager/fsdp_ulysses.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/fsdp_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/sharding_manager/fsdp_vllm.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/megatron_sglang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/sharding_manager/megatron_sglang.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/megatron_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xid32/SoundMind/HEAD/verl/workers/sharding_manager/megatron_vllm.py --------------------------------------------------------------------------------