├── LICENSE ├── README.md ├── datasets ├── download_nyt.py ├── masked_time_entity │ ├── test_masked.jsonl │ └── train_masked.jsonl ├── news_desk.py ├── process_nyt.py └── val_pre_error.py ├── docs ├── Makefile ├── README.md ├── _static │ └── logo.png ├── advance │ ├── dpo_extension.rst │ ├── fsdp_extension.rst │ ├── megatron_extension.rst │ └── placement.rst ├── conf.py ├── examples │ ├── config.rst │ ├── gsm8k_example.rst │ └── ppo_code_architecture.rst ├── experiment │ └── ppo.rst ├── faq │ └── faq.rst ├── index.rst ├── preparation │ ├── prepare_data.rst │ └── reward_function.rst ├── requirements-docs.txt ├── start │ ├── install.rst │ └── quickstart.rst └── workers │ ├── fsdp_workers.rst │ ├── megatron_workers.rst │ └── ray_trainer.rst ├── examples ├── data_preprocess │ ├── arth.py │ ├── countdown.py │ ├── create_time_reasoning_final_datasets.py │ ├── event_order_nyt.py │ ├── full_hh_rlhf.py │ ├── gsm8k.py │ ├── hellaswag.py │ ├── math_dataset.py │ ├── multiply.py │ ├── time_prediction_nyt.py │ └── time_reasoning_nyt.py ├── generation │ └── run_deepseek_v2_lite_math.sh ├── grpo_trainer │ ├── run_deepseek7b_llm.sh │ ├── run_deepseek7b_llm_seq_balance.sh │ ├── run_qwen2-7b.sh │ └── run_qwen2-7b_seq_balance.sh ├── ppo_trainer │ ├── run_deepseek7b_llm.sh │ ├── run_deepseek7b_llm_sp2.sh │ ├── run_deepseek_full_hh_rlhf.sh │ ├── run_deepseek_math_gsm8k_megatron.sh │ ├── run_deepseek_megatron.sh │ ├── run_gemma.sh │ ├── run_qwen2-7b.sh │ ├── run_qwen2-7b_rm.sh │ ├── run_qwen2-7b_rm_seq_balance.sh │ ├── run_qwen2-7b_seq_balance.sh │ ├── run_qwen2.5-32b.sh │ └── verl_getting_started.ipynb ├── ray │ └── tutorial.ipynb ├── sft │ └── gsm8k │ │ ├── run_deepseek_6b7.sh │ │ ├── run_gemma_2b.sh │ │ └── run_gemma_7b.sh └── split_placement │ ├── README.md │ ├── config │ └── ppo_trainer_split.yaml │ ├── main_ppo_split.py │ ├── run_deepseek7b_llm.sh │ └── split_monkey_patch.py ├── figures ├── Stage1_result.png ├── Stage1_training.png ├── Stage2_result.png ├── Stage3_result.png ├── examples_graph.pdf ├── output_examples.png ├── time-r1.pdf └── time-r1_overview.png ├── future_news_generation ├── analyze_generation_diversity_monthly.py ├── analyze_monthly_similarity.py ├── analyze_news_desk_distribution.py ├── news_generator.py ├── stage3.sh ├── topic_analysis │ ├── category_trends.png │ └── overall_distribution_pie.png └── v3_generation_4_prediction.py ├── patches └── megatron_v4.patch ├── preliminary ├── original_ability_stas.py ├── test_original_ability.py └── time_entity_statistics.py ├── requirements.txt ├── scripts ├── stage1_no_dynamic_reward.sh ├── stage1_phase1.sh ├── stage1_phase2.sh ├── stage1_phase3.sh └── stage2.sh ├── setup.py ├── verl.egg-info ├── PKG-INFO ├── SOURCES.txt ├── dependency_links.txt ├── requires.txt └── top_level.txt └── verl ├── __init__.py ├── __pycache__ ├── __init__.cpython-39.pyc └── protocol.cpython-39.pyc ├── models ├── README.md ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ └── registry.cpython-39.pyc ├── llama │ ├── __init__.py │ └── megatron │ │ ├── __init__.py │ │ ├── checkpoint_utils │ │ ├── __init__.py │ │ ├── llama_loader.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 ├── registry.py ├── transformers │ ├── __init__.py │ ├── llama.py │ ├── monkey_patch.py │ └── qwen2.py └── weight_loader_registry.py ├── protocol.py ├── single_controller ├── __init__.py ├── __pycache__ │ └── __init__.cpython-39.pyc ├── base │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── decorator.cpython-39.pyc │ │ ├── worker.cpython-39.pyc │ │ └── worker_group.cpython-39.pyc │ ├── decorator.py │ ├── megatron │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ ├── worker.cpython-39.pyc │ │ │ └── worker_group.cpython-39.pyc │ │ ├── worker.py │ │ └── worker_group.py │ ├── register_center │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-39.pyc │ │ │ └── ray.cpython-39.pyc │ │ └── ray.py │ ├── worker.py │ └── worker_group.py ├── ray │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── base.cpython-39.pyc │ │ └── megatron.cpython-39.pyc │ ├── base.py │ └── megatron.py └── version │ └── version ├── third_party ├── __init__.py ├── __pycache__ │ └── __init__.cpython-39.pyc └── vllm │ ├── __init__.py │ ├── __pycache__ │ └── __init__.cpython-39.pyc │ ├── vllm_v_0_3_1 │ ├── __init__.py │ ├── arg_utils.py │ ├── config.py │ ├── llm.py │ ├── llm_engine_sp.py │ ├── model_loader.py │ ├── model_runner.py │ ├── parallel_state.py │ ├── tokenizer.py │ ├── weight_loaders.py │ └── worker.py │ ├── vllm_v_0_4_2 │ ├── __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_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 │ ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── arg_utils.cpython-39.pyc │ ├── config.cpython-39.pyc │ ├── dtensor_weight_loaders.cpython-39.pyc │ ├── hf_weight_loader.cpython-39.pyc │ ├── llm.cpython-39.pyc │ ├── llm_engine_sp.cpython-39.pyc │ ├── megatron_weight_loaders.cpython-39.pyc │ ├── model_loader.cpython-39.pyc │ ├── model_runner.cpython-39.pyc │ ├── parallel_state.cpython-39.pyc │ ├── spmd_gpu_executor.cpython-39.pyc │ ├── tokenizer.cpython-39.pyc │ └── worker.cpython-39.pyc │ ├── 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 ├── trainer ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ └── main_ppo.cpython-39.pyc ├── 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 ├── main_ppo_s1_p1.py ├── main_ppo_s1_p2.py ├── main_ppo_s1_p3.py ├── main_ppo_s2.py ├── ppo │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── core_algos.cpython-39.pyc │ │ └── ray_trainer.cpython-39.pyc │ ├── core_algos.py │ ├── ray_trainer.py │ ├── ray_trainer_s1.py │ └── ray_trainer_s2.py └── runtime_env.yaml ├── utils ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── flops_counter.cpython-39.pyc │ ├── fs.cpython-39.pyc │ ├── fsdp_utils.cpython-39.pyc │ ├── hdfs_io.cpython-39.pyc │ ├── import_utils.cpython-39.pyc │ ├── logging_utils.cpython-39.pyc │ ├── model.cpython-39.pyc │ ├── py_functional.cpython-39.pyc │ ├── seqlen_balancing.cpython-39.pyc │ ├── tokenizer.cpython-39.pyc │ ├── torch_dtypes.cpython-39.pyc │ ├── torch_functional.cpython-39.pyc │ ├── tracking.cpython-39.pyc │ └── ulysses.cpython-39.pyc ├── config.py ├── dataset │ ├── README.md │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── rl_dataset.cpython-39.pyc │ │ ├── rm_dataset.cpython-39.pyc │ │ └── sft_dataset.cpython-39.pyc │ ├── rl_dataset.py │ ├── rm_dataset.py │ └── sft_dataset.py ├── debug │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ └── performance.cpython-39.pyc │ ├── performance.py │ └── trajectory_tracker.py ├── distributed.py ├── flops_counter.py ├── fs.py ├── fsdp_utils.py ├── hdfs_io.py ├── import_utils.py ├── logger │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ └── aggregate_logger.cpython-39.pyc │ └── aggregate_logger.py ├── logging_utils.py ├── megatron │ ├── __init__.py │ ├── memory.py │ ├── optimizer.py │ ├── optimizer_config.py │ ├── pipeline_parallel.py │ ├── sequence_parallel.py │ └── tensor_parallel.py ├── megatron_utils.py ├── memory_buffer.py ├── model.py ├── py_functional.py ├── ray_utils.py ├── rendezvous │ ├── __init__.py │ └── ray_backend.py ├── reward_score │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ ├── countdown.cpython-39.pyc │ │ ├── gsm8k.cpython-39.pyc │ │ ├── math.cpython-39.pyc │ │ ├── multiply.cpython-39.pyc │ │ ├── news.cpython-39.pyc │ │ ├── time_prediction.cpython-39.pyc │ │ └── time_reasoning.cpython-39.pyc │ ├── countdown.py │ ├── gsm8k.py │ ├── math.py │ ├── multiply.py │ ├── time_prediction.py │ ├── time_reasoning.py │ └── time_reasoning_fixed_alpha.py ├── seqlen_balancing.py ├── tokenizer.py ├── torch_dtypes.py ├── torch_functional.py ├── tracking.py └── ulysses.py ├── version └── version └── workers ├── __init__.py ├── __pycache__ ├── __init__.cpython-39.pyc └── fsdp_workers.cpython-39.pyc ├── actor ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── base.cpython-39.pyc │ └── dp_actor.cpython-39.pyc ├── 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_model ├── __init__.py ├── base.py └── megatron │ ├── __init__.py │ └── reward_model.py ├── rollout ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-39.pyc │ ├── base.cpython-39.pyc │ ├── hf_rollout.cpython-39.pyc │ └── tokenizer.cpython-39.pyc ├── base.py ├── hf_rollout.py ├── naive │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-39.pyc │ │ └── naive_rollout.cpython-39.pyc │ └── naive_rollout.py ├── tokenizer.py └── vllm_rollout │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-39.pyc │ └── vllm_rollout.cpython-39.pyc │ └── vllm_rollout.py └── sharding_manager ├── __init__.py ├── __pycache__ ├── __init__.cpython-39.pyc ├── base.cpython-39.pyc ├── fsdp_ulysses.cpython-39.pyc └── fsdp_vllm.cpython-39.pyc ├── base.py ├── fsdp_ulysses.py ├── fsdp_vllm.py └── megatron_vllm.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/README.md -------------------------------------------------------------------------------- /datasets/download_nyt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/datasets/download_nyt.py -------------------------------------------------------------------------------- /datasets/masked_time_entity/test_masked.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/datasets/masked_time_entity/test_masked.jsonl -------------------------------------------------------------------------------- /datasets/masked_time_entity/train_masked.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/datasets/masked_time_entity/train_masked.jsonl -------------------------------------------------------------------------------- /datasets/news_desk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/datasets/news_desk.py -------------------------------------------------------------------------------- /datasets/process_nyt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/datasets/process_nyt.py -------------------------------------------------------------------------------- /datasets/val_pre_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/datasets/val_pre_error.py -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/_static/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/_static/logo.png -------------------------------------------------------------------------------- /docs/advance/dpo_extension.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/advance/dpo_extension.rst -------------------------------------------------------------------------------- /docs/advance/fsdp_extension.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/advance/fsdp_extension.rst -------------------------------------------------------------------------------- /docs/advance/megatron_extension.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/advance/megatron_extension.rst -------------------------------------------------------------------------------- /docs/advance/placement.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/advance/placement.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/examples/config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/examples/config.rst -------------------------------------------------------------------------------- /docs/examples/gsm8k_example.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/examples/gsm8k_example.rst -------------------------------------------------------------------------------- /docs/examples/ppo_code_architecture.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/examples/ppo_code_architecture.rst -------------------------------------------------------------------------------- /docs/experiment/ppo.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/experiment/ppo.rst -------------------------------------------------------------------------------- /docs/faq/faq.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/faq/faq.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/preparation/prepare_data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/preparation/prepare_data.rst -------------------------------------------------------------------------------- /docs/preparation/reward_function.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/preparation/reward_function.rst -------------------------------------------------------------------------------- /docs/requirements-docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/requirements-docs.txt -------------------------------------------------------------------------------- /docs/start/install.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/start/install.rst -------------------------------------------------------------------------------- /docs/start/quickstart.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/start/quickstart.rst -------------------------------------------------------------------------------- /docs/workers/fsdp_workers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/workers/fsdp_workers.rst -------------------------------------------------------------------------------- /docs/workers/megatron_workers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/workers/megatron_workers.rst -------------------------------------------------------------------------------- /docs/workers/ray_trainer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/docs/workers/ray_trainer.rst -------------------------------------------------------------------------------- /examples/data_preprocess/arth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/arth.py -------------------------------------------------------------------------------- /examples/data_preprocess/countdown.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/countdown.py -------------------------------------------------------------------------------- /examples/data_preprocess/create_time_reasoning_final_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/create_time_reasoning_final_datasets.py -------------------------------------------------------------------------------- /examples/data_preprocess/event_order_nyt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/event_order_nyt.py -------------------------------------------------------------------------------- /examples/data_preprocess/full_hh_rlhf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/full_hh_rlhf.py -------------------------------------------------------------------------------- /examples/data_preprocess/gsm8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/gsm8k.py -------------------------------------------------------------------------------- /examples/data_preprocess/hellaswag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/hellaswag.py -------------------------------------------------------------------------------- /examples/data_preprocess/math_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/math_dataset.py -------------------------------------------------------------------------------- /examples/data_preprocess/multiply.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/multiply.py -------------------------------------------------------------------------------- /examples/data_preprocess/time_prediction_nyt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/time_prediction_nyt.py -------------------------------------------------------------------------------- /examples/data_preprocess/time_reasoning_nyt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/data_preprocess/time_reasoning_nyt.py -------------------------------------------------------------------------------- /examples/generation/run_deepseek_v2_lite_math.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/generation/run_deepseek_v2_lite_math.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek7b_llm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/grpo_trainer/run_deepseek7b_llm.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_deepseek7b_llm_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/grpo_trainer/run_deepseek7b_llm_seq_balance.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/grpo_trainer/run_qwen2-7b.sh -------------------------------------------------------------------------------- /examples/grpo_trainer/run_qwen2-7b_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/grpo_trainer/run_qwen2-7b_seq_balance.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_deepseek7b_llm.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek7b_llm_sp2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_deepseek7b_llm_sp2.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek_full_hh_rlhf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_deepseek_full_hh_rlhf.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek_math_gsm8k_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_deepseek_math_gsm8k_megatron.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_deepseek_megatron.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_deepseek_megatron.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_gemma.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_gemma.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_qwen2-7b.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_rm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_qwen2-7b_rm.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_rm_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_qwen2-7b_rm_seq_balance.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2-7b_seq_balance.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_qwen2-7b_seq_balance.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/run_qwen2.5-32b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/run_qwen2.5-32b.sh -------------------------------------------------------------------------------- /examples/ppo_trainer/verl_getting_started.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ppo_trainer/verl_getting_started.ipynb -------------------------------------------------------------------------------- /examples/ray/tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/ray/tutorial.ipynb -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_deepseek_6b7.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/sft/gsm8k/run_deepseek_6b7.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_gemma_2b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/sft/gsm8k/run_gemma_2b.sh -------------------------------------------------------------------------------- /examples/sft/gsm8k/run_gemma_7b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/sft/gsm8k/run_gemma_7b.sh -------------------------------------------------------------------------------- /examples/split_placement/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/split_placement/README.md -------------------------------------------------------------------------------- /examples/split_placement/config/ppo_trainer_split.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/split_placement/config/ppo_trainer_split.yaml -------------------------------------------------------------------------------- /examples/split_placement/main_ppo_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/split_placement/main_ppo_split.py -------------------------------------------------------------------------------- /examples/split_placement/run_deepseek7b_llm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/split_placement/run_deepseek7b_llm.sh -------------------------------------------------------------------------------- /examples/split_placement/split_monkey_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/examples/split_placement/split_monkey_patch.py -------------------------------------------------------------------------------- /figures/Stage1_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/figures/Stage1_result.png -------------------------------------------------------------------------------- /figures/Stage1_training.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/figures/Stage1_training.png -------------------------------------------------------------------------------- /figures/Stage2_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/figures/Stage2_result.png -------------------------------------------------------------------------------- /figures/Stage3_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/figures/Stage3_result.png -------------------------------------------------------------------------------- /figures/examples_graph.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/figures/examples_graph.pdf -------------------------------------------------------------------------------- /figures/output_examples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/figures/output_examples.png -------------------------------------------------------------------------------- /figures/time-r1.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/figures/time-r1.pdf -------------------------------------------------------------------------------- /figures/time-r1_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/figures/time-r1_overview.png -------------------------------------------------------------------------------- /future_news_generation/analyze_generation_diversity_monthly.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/future_news_generation/analyze_generation_diversity_monthly.py -------------------------------------------------------------------------------- /future_news_generation/analyze_monthly_similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/future_news_generation/analyze_monthly_similarity.py -------------------------------------------------------------------------------- /future_news_generation/analyze_news_desk_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/future_news_generation/analyze_news_desk_distribution.py -------------------------------------------------------------------------------- /future_news_generation/news_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/future_news_generation/news_generator.py -------------------------------------------------------------------------------- /future_news_generation/stage3.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/future_news_generation/stage3.sh -------------------------------------------------------------------------------- /future_news_generation/topic_analysis/category_trends.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/future_news_generation/topic_analysis/category_trends.png -------------------------------------------------------------------------------- /future_news_generation/topic_analysis/overall_distribution_pie.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/future_news_generation/topic_analysis/overall_distribution_pie.png -------------------------------------------------------------------------------- /future_news_generation/v3_generation_4_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/future_news_generation/v3_generation_4_prediction.py -------------------------------------------------------------------------------- /patches/megatron_v4.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/patches/megatron_v4.patch -------------------------------------------------------------------------------- /preliminary/original_ability_stas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/preliminary/original_ability_stas.py -------------------------------------------------------------------------------- /preliminary/test_original_ability.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/preliminary/test_original_ability.py -------------------------------------------------------------------------------- /preliminary/time_entity_statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/preliminary/time_entity_statistics.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/stage1_no_dynamic_reward.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/scripts/stage1_no_dynamic_reward.sh -------------------------------------------------------------------------------- /scripts/stage1_phase1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/scripts/stage1_phase1.sh -------------------------------------------------------------------------------- /scripts/stage1_phase2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/scripts/stage1_phase2.sh -------------------------------------------------------------------------------- /scripts/stage1_phase3.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/scripts/stage1_phase3.sh -------------------------------------------------------------------------------- /scripts/stage2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/scripts/stage2.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/setup.py -------------------------------------------------------------------------------- /verl.egg-info/PKG-INFO: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl.egg-info/PKG-INFO -------------------------------------------------------------------------------- /verl.egg-info/SOURCES.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl.egg-info/SOURCES.txt -------------------------------------------------------------------------------- /verl.egg-info/dependency_links.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /verl.egg-info/requires.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl.egg-info/requires.txt -------------------------------------------------------------------------------- /verl.egg-info/top_level.txt: -------------------------------------------------------------------------------- 1 | tests 2 | verl 3 | -------------------------------------------------------------------------------- /verl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/__init__.py -------------------------------------------------------------------------------- /verl/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/__pycache__/protocol.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/__pycache__/protocol.cpython-39.pyc -------------------------------------------------------------------------------- /verl/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/README.md -------------------------------------------------------------------------------- /verl/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/__init__.py -------------------------------------------------------------------------------- /verl/models/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/models/__pycache__/registry.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/__pycache__/registry.cpython-39.pyc -------------------------------------------------------------------------------- /verl/models/llama/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/checkpoint_utils/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/llama_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/checkpoint_utils/llama_loader.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/checkpoint_utils/llama_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/checkpoint_utils/llama_saver.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/layers/__init__.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/layers/parallel_attention.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/layers/parallel_decoder.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/layers/parallel_linear.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/layers/parallel_mlp.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/layers/parallel_rmsnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/layers/parallel_rmsnorm.py -------------------------------------------------------------------------------- /verl/models/llama/megatron/modeling_llama_megatron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/llama/megatron/modeling_llama_megatron.py -------------------------------------------------------------------------------- /verl/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/registry.py -------------------------------------------------------------------------------- /verl/models/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/transformers/__init__.py -------------------------------------------------------------------------------- /verl/models/transformers/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/transformers/llama.py -------------------------------------------------------------------------------- /verl/models/transformers/monkey_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/transformers/monkey_patch.py -------------------------------------------------------------------------------- /verl/models/transformers/qwen2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/transformers/qwen2.py -------------------------------------------------------------------------------- /verl/models/weight_loader_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/models/weight_loader_registry.py -------------------------------------------------------------------------------- /verl/protocol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/protocol.py -------------------------------------------------------------------------------- /verl/single_controller/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/base/__pycache__/decorator.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/__pycache__/decorator.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/base/__pycache__/worker.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/__pycache__/worker.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/base/__pycache__/worker_group.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/__pycache__/worker_group.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/base/decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/decorator.py -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/megatron/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/megatron/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/__pycache__/worker.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/megatron/__pycache__/worker.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/__pycache__/worker_group.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/megatron/__pycache__/worker_group.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/megatron/worker.py -------------------------------------------------------------------------------- /verl/single_controller/base/megatron/worker_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/megatron/worker_group.py -------------------------------------------------------------------------------- /verl/single_controller/base/register_center/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/register_center/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/base/register_center/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/register_center/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/base/register_center/__pycache__/ray.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/register_center/__pycache__/ray.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/base/register_center/ray.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/register_center/ray.py -------------------------------------------------------------------------------- /verl/single_controller/base/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/worker.py -------------------------------------------------------------------------------- /verl/single_controller/base/worker_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/base/worker_group.py -------------------------------------------------------------------------------- /verl/single_controller/ray/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/ray/__init__.py -------------------------------------------------------------------------------- /verl/single_controller/ray/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/ray/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/ray/__pycache__/base.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/ray/__pycache__/base.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/ray/__pycache__/megatron.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/ray/__pycache__/megatron.cpython-39.pyc -------------------------------------------------------------------------------- /verl/single_controller/ray/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/ray/base.py -------------------------------------------------------------------------------- /verl/single_controller/ray/megatron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/single_controller/ray/megatron.py -------------------------------------------------------------------------------- /verl/single_controller/version/version: -------------------------------------------------------------------------------- 1 | 0.0.2 -------------------------------------------------------------------------------- /verl/third_party/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/__init__.py -------------------------------------------------------------------------------- /verl/third_party/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/__init__.py -------------------------------------------------------------------------------- /verl/third_party/vllm/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/__init__.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/arg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/arg_utils.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/config.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/llm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/llm.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/llm_engine_sp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/llm_engine_sp.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/model_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/model_loader.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/model_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/model_runner.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/parallel_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/parallel_state.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/tokenizer.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/weight_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/weight_loaders.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_3_1/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_3_1/worker.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/__init__.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/arg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/arg_utils.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/config.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/dtensor_weight_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/dtensor_weight_loaders.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/hf_weight_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/hf_weight_loader.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/llm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/llm.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/llm_engine_sp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/llm_engine_sp.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/megatron_weight_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/megatron_weight_loaders.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/model_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/model_loader.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/model_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/model_runner.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/parallel_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/parallel_state.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/spmd_gpu_executor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/spmd_gpu_executor.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/tokenizer.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_4_2/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_4_2/worker.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_5_4/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__init__.py -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/arg_utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/arg_utils.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/config.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/config.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/dtensor_weight_loaders.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/dtensor_weight_loaders.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/hf_weight_loader.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/hf_weight_loader.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/llm.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/llm.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/llm_engine_sp.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/llm_engine_sp.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/megatron_weight_loaders.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/megatron_weight_loaders.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/model_loader.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/model_loader.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/model_runner.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/model_runner.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/parallel_state.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/parallel_state.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/spmd_gpu_executor.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/spmd_gpu_executor.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/tokenizer.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/tokenizer.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/__pycache__/worker.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/__pycache__/worker.cpython-39.pyc -------------------------------------------------------------------------------- /verl/third_party/vllm/vllm_v_0_6_3/arg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/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/ulab-uiuc/Time-R1/HEAD/verl/third_party/vllm/vllm_v_0_6_3/worker.py -------------------------------------------------------------------------------- /verl/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/__init__.py -------------------------------------------------------------------------------- /verl/trainer/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/trainer/__pycache__/main_ppo.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/__pycache__/main_ppo.cpython-39.pyc -------------------------------------------------------------------------------- /verl/trainer/config/evaluation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/config/evaluation.yaml -------------------------------------------------------------------------------- /verl/trainer/config/generation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/config/generation.yaml -------------------------------------------------------------------------------- /verl/trainer/config/ppo_megatron_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/config/ppo_megatron_trainer.yaml -------------------------------------------------------------------------------- /verl/trainer/config/ppo_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/config/ppo_trainer.yaml -------------------------------------------------------------------------------- /verl/trainer/config/sft_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/config/sft_trainer.yaml -------------------------------------------------------------------------------- /verl/trainer/fsdp_sft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/fsdp_sft_trainer.py -------------------------------------------------------------------------------- /verl/trainer/main_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/main_eval.py -------------------------------------------------------------------------------- /verl/trainer/main_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/main_generation.py -------------------------------------------------------------------------------- /verl/trainer/main_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/main_ppo.py -------------------------------------------------------------------------------- /verl/trainer/main_ppo_s1_p1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/main_ppo_s1_p1.py -------------------------------------------------------------------------------- /verl/trainer/main_ppo_s1_p2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/main_ppo_s1_p2.py -------------------------------------------------------------------------------- /verl/trainer/main_ppo_s1_p3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/main_ppo_s1_p3.py -------------------------------------------------------------------------------- /verl/trainer/main_ppo_s2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/main_ppo_s2.py -------------------------------------------------------------------------------- /verl/trainer/ppo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/ppo/__init__.py -------------------------------------------------------------------------------- /verl/trainer/ppo/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/ppo/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/trainer/ppo/__pycache__/core_algos.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/ppo/__pycache__/core_algos.cpython-39.pyc -------------------------------------------------------------------------------- /verl/trainer/ppo/__pycache__/ray_trainer.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/ppo/__pycache__/ray_trainer.cpython-39.pyc -------------------------------------------------------------------------------- /verl/trainer/ppo/core_algos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/ppo/core_algos.py -------------------------------------------------------------------------------- /verl/trainer/ppo/ray_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/ppo/ray_trainer.py -------------------------------------------------------------------------------- /verl/trainer/ppo/ray_trainer_s1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/ppo/ray_trainer_s1.py -------------------------------------------------------------------------------- /verl/trainer/ppo/ray_trainer_s2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/ppo/ray_trainer_s2.py -------------------------------------------------------------------------------- /verl/trainer/runtime_env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/trainer/runtime_env.yaml -------------------------------------------------------------------------------- /verl/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__init__.py -------------------------------------------------------------------------------- /verl/utils/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/flops_counter.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/flops_counter.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/fs.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/fs.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/fsdp_utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/fsdp_utils.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/hdfs_io.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/hdfs_io.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/import_utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/import_utils.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/logging_utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/logging_utils.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/model.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/py_functional.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/py_functional.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/seqlen_balancing.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/seqlen_balancing.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/tokenizer.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/tokenizer.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/torch_dtypes.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/torch_dtypes.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/torch_functional.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/torch_functional.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/tracking.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/tracking.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/__pycache__/ulysses.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/__pycache__/ulysses.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/config.py -------------------------------------------------------------------------------- /verl/utils/dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/dataset/README.md -------------------------------------------------------------------------------- /verl/utils/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/dataset/__init__.py -------------------------------------------------------------------------------- /verl/utils/dataset/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/dataset/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/dataset/__pycache__/rl_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/dataset/__pycache__/rl_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/dataset/__pycache__/rm_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/dataset/__pycache__/rm_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/dataset/__pycache__/sft_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/dataset/__pycache__/sft_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/dataset/rl_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/dataset/rl_dataset.py -------------------------------------------------------------------------------- /verl/utils/dataset/rm_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/dataset/rm_dataset.py -------------------------------------------------------------------------------- /verl/utils/dataset/sft_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/dataset/sft_dataset.py -------------------------------------------------------------------------------- /verl/utils/debug/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/debug/__init__.py -------------------------------------------------------------------------------- /verl/utils/debug/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/debug/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/debug/__pycache__/performance.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/debug/__pycache__/performance.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/debug/performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/debug/performance.py -------------------------------------------------------------------------------- /verl/utils/debug/trajectory_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/debug/trajectory_tracker.py -------------------------------------------------------------------------------- /verl/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/distributed.py -------------------------------------------------------------------------------- /verl/utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/flops_counter.py -------------------------------------------------------------------------------- /verl/utils/fs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/fs.py -------------------------------------------------------------------------------- /verl/utils/fsdp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/fsdp_utils.py -------------------------------------------------------------------------------- /verl/utils/hdfs_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/hdfs_io.py -------------------------------------------------------------------------------- /verl/utils/import_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/import_utils.py -------------------------------------------------------------------------------- /verl/utils/logger/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/logger/__init__.py -------------------------------------------------------------------------------- /verl/utils/logger/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/logger/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/logger/__pycache__/aggregate_logger.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/logger/__pycache__/aggregate_logger.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/logger/aggregate_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/logger/aggregate_logger.py -------------------------------------------------------------------------------- /verl/utils/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/logging_utils.py -------------------------------------------------------------------------------- /verl/utils/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/megatron/__init__.py -------------------------------------------------------------------------------- /verl/utils/megatron/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/megatron/memory.py -------------------------------------------------------------------------------- /verl/utils/megatron/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/megatron/optimizer.py -------------------------------------------------------------------------------- /verl/utils/megatron/optimizer_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/megatron/optimizer_config.py -------------------------------------------------------------------------------- /verl/utils/megatron/pipeline_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/megatron/pipeline_parallel.py -------------------------------------------------------------------------------- /verl/utils/megatron/sequence_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/megatron/sequence_parallel.py -------------------------------------------------------------------------------- /verl/utils/megatron/tensor_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/megatron/tensor_parallel.py -------------------------------------------------------------------------------- /verl/utils/megatron_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/megatron_utils.py -------------------------------------------------------------------------------- /verl/utils/memory_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/memory_buffer.py -------------------------------------------------------------------------------- /verl/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/model.py -------------------------------------------------------------------------------- /verl/utils/py_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/py_functional.py -------------------------------------------------------------------------------- /verl/utils/ray_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/ray_utils.py -------------------------------------------------------------------------------- /verl/utils/rendezvous/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/rendezvous/__init__.py -------------------------------------------------------------------------------- /verl/utils/rendezvous/ray_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/rendezvous/ray_backend.py -------------------------------------------------------------------------------- /verl/utils/reward_score/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/__init__.py -------------------------------------------------------------------------------- /verl/utils/reward_score/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/reward_score/__pycache__/countdown.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/__pycache__/countdown.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/reward_score/__pycache__/gsm8k.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/__pycache__/gsm8k.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/reward_score/__pycache__/math.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/__pycache__/math.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/reward_score/__pycache__/multiply.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/__pycache__/multiply.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/reward_score/__pycache__/news.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/__pycache__/news.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/reward_score/__pycache__/time_prediction.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/__pycache__/time_prediction.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/reward_score/__pycache__/time_reasoning.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/__pycache__/time_reasoning.cpython-39.pyc -------------------------------------------------------------------------------- /verl/utils/reward_score/countdown.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/countdown.py -------------------------------------------------------------------------------- /verl/utils/reward_score/gsm8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/gsm8k.py -------------------------------------------------------------------------------- /verl/utils/reward_score/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/math.py -------------------------------------------------------------------------------- /verl/utils/reward_score/multiply.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/multiply.py -------------------------------------------------------------------------------- /verl/utils/reward_score/time_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/time_prediction.py -------------------------------------------------------------------------------- /verl/utils/reward_score/time_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/time_reasoning.py -------------------------------------------------------------------------------- /verl/utils/reward_score/time_reasoning_fixed_alpha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/reward_score/time_reasoning_fixed_alpha.py -------------------------------------------------------------------------------- /verl/utils/seqlen_balancing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/seqlen_balancing.py -------------------------------------------------------------------------------- /verl/utils/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/tokenizer.py -------------------------------------------------------------------------------- /verl/utils/torch_dtypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/torch_dtypes.py -------------------------------------------------------------------------------- /verl/utils/torch_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/torch_functional.py -------------------------------------------------------------------------------- /verl/utils/tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/tracking.py -------------------------------------------------------------------------------- /verl/utils/ulysses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/utils/ulysses.py -------------------------------------------------------------------------------- /verl/version/version: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /verl/workers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/__init__.py -------------------------------------------------------------------------------- /verl/workers/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/__pycache__/fsdp_workers.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/__pycache__/fsdp_workers.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/actor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/actor/__init__.py -------------------------------------------------------------------------------- /verl/workers/actor/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/actor/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/actor/__pycache__/base.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/actor/__pycache__/base.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/actor/__pycache__/dp_actor.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/actor/__pycache__/dp_actor.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/actor/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/actor/base.py -------------------------------------------------------------------------------- /verl/workers/actor/dp_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/actor/dp_actor.py -------------------------------------------------------------------------------- /verl/workers/actor/megatron_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/actor/megatron_actor.py -------------------------------------------------------------------------------- /verl/workers/critic/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/critic/__init__.py -------------------------------------------------------------------------------- /verl/workers/critic/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/critic/base.py -------------------------------------------------------------------------------- /verl/workers/critic/dp_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/critic/dp_critic.py -------------------------------------------------------------------------------- /verl/workers/critic/megatron_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/critic/megatron_critic.py -------------------------------------------------------------------------------- /verl/workers/fsdp_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/fsdp_workers.py -------------------------------------------------------------------------------- /verl/workers/megatron_workers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/megatron_workers.py -------------------------------------------------------------------------------- /verl/workers/reward_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/reward_model/__init__.py -------------------------------------------------------------------------------- /verl/workers/reward_model/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/reward_model/base.py -------------------------------------------------------------------------------- /verl/workers/reward_model/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/reward_model/megatron/__init__.py -------------------------------------------------------------------------------- /verl/workers/reward_model/megatron/reward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/reward_model/megatron/reward_model.py -------------------------------------------------------------------------------- /verl/workers/rollout/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/rollout/__pycache__/base.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/__pycache__/base.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/rollout/__pycache__/hf_rollout.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/__pycache__/hf_rollout.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/rollout/__pycache__/tokenizer.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/__pycache__/tokenizer.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/rollout/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/base.py -------------------------------------------------------------------------------- /verl/workers/rollout/hf_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/hf_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/naive/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/naive/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/naive/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/naive/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/rollout/naive/__pycache__/naive_rollout.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/naive/__pycache__/naive_rollout.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/rollout/naive/naive_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/naive/naive_rollout.py -------------------------------------------------------------------------------- /verl/workers/rollout/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/tokenizer.py -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/vllm_rollout/__init__.py -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/vllm_rollout/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/__pycache__/vllm_rollout.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/vllm_rollout/__pycache__/vllm_rollout.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/rollout/vllm_rollout/vllm_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/rollout/vllm_rollout/vllm_rollout.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/sharding_manager/__init__.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/sharding_manager/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/sharding_manager/__pycache__/base.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/sharding_manager/__pycache__/base.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/sharding_manager/__pycache__/fsdp_ulysses.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/sharding_manager/__pycache__/fsdp_ulysses.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/sharding_manager/__pycache__/fsdp_vllm.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/sharding_manager/__pycache__/fsdp_vllm.cpython-39.pyc -------------------------------------------------------------------------------- /verl/workers/sharding_manager/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/sharding_manager/base.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/fsdp_ulysses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/sharding_manager/fsdp_ulysses.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/fsdp_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/sharding_manager/fsdp_vllm.py -------------------------------------------------------------------------------- /verl/workers/sharding_manager/megatron_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/Time-R1/HEAD/verl/workers/sharding_manager/megatron_vllm.py --------------------------------------------------------------------------------