Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/DLYuanGod/MegaTrain
/ endpoints
Endpoints
136 in github.com/DLYuanGod/MegaTrain
⨍
Functions
5,181
◇
Types & classes
739
↳
Endpoints
136
Route
_default_compute_score
deprecated("verl.utils.reward_score.default_compute_score")
verl/verl/utils/reward_score/__init__.py:None
Route
_inner_call_method
patch("builtins.open", mock_open(read_data=inputs))
verl/verl/utils/reward_score/prime_code/testing_util.py:None
Route
_query_collect_info
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/single_controller/base/worker.py:None
Route
_query_dispatch_info
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/single_controller/base/worker.py:None
Route
add
register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)
verl/tests/single_controller/test_colocated_workers_fused.py:None
Route
add
register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)
verl/tests/single_controller/test_split_resource_pool.py:None
Route
add
register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)
verl/tests/single_controller/test_colocated_workers.py:None
Route
add
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/tests/single_controller/test_fused_workers_on_cpu.py:None
Route
add
register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)
verl/tests/single_controller/test_auto_padding_on_cpu.py:None
Route
async_calls_finalize_fn_exec
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/megatron_workers.py:None
Route
async_dp_compute
register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO, blocking=False)
verl/tests/single_controller/test_decorator_on_cpu.py:None
Route
check_weights
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/tests/checkpoint_engine/test_utils.py:None
Route
clear_cpu_model
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/experimental/separation/engine_workers.py:None
Route
compute_log_prob
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="actor"))
verl/verl/workers/megatron_workers.py:None
Route
compute_log_prob
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="actor"))
verl/verl/workers/fsdp_workers.py:None
Route
compute_log_prob
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="actor"))
verl/verl/workers/engine_workers.py:None
Route
compute_policy_loss
deprecated("verl.trainer.ppo.core_algos.compute_policy_loss_vanilla")
verl/verl/trainer/ppo/core_algos.py:None
Route
compute_ref_log_prob
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="actor"))
verl/verl/workers/megatron_workers.py:None
Route
compute_ref_log_prob
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="actor"))
verl/verl/workers/fsdp_workers.py:None
Route
compute_ref_log_prob
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="ref"))
verl/verl/workers/engine_workers.py:None
Route
compute_values
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="critic"))
verl/verl/workers/megatron_workers.py:None
Route
compute_values
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="critic"))
verl/verl/workers/fsdp_workers.py:None
Route
do_nothing
register(dispatch_mode=Dispatch.DP_COMPUTE, blocking=False)
verl/tests/single_controller/test_data_transfer.py:None
Route
dp_compute
register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)
verl/tests/single_controller/test_decorator_on_cpu.py:None
Route
dp_compute_td
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="train"), blocking=False)
verl/tests/single_controller/test_decorator_on_cpu.py:None
Route
dump_memory_snapshot
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/megatron_workers.py:None
Route
dump_memory_snapshot
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/fsdp_workers.py:None
Route
env_interact_step
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="env"), blocking=False)
verl/verl/experimental/vla/workers/env/env_worker.py:None
Route
execute_checkpoint_engine
register(dispatch_mode=Dispatch.DP_COMPUTE, blocking=False)
verl/tests/checkpoint_engine/test_utils.py:None
Route
execute_checkpoint_engine
register(dispatch_mode=Dispatch.DP_COMPUTE, blocking=False)
verl/verl/workers/engine_workers.py:None
Route
execute_checkpoint_engine
register(dispatch_mode=Dispatch.DP_COMPUTE, blocking=False)
verl/verl/checkpoint_engine/base.py:None
Route
execute_func_rank_zero
register(dispatch_mode=Dispatch.ALL_TO_ALL, execute_mode=Execute.RANK_ZERO)
verl/verl/single_controller/base/worker.py:None
Route
execute_with_func_generator
register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO_WITH_FUNC)
verl/verl/single_controller/base/worker.py:None
Route
finish_rollout
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/experimental/vla/workers/env/env_worker.py:None
Route
foo
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/tests/single_controller/test_fused_workers_on_cpu.py:None
Route
foo
register(dispatch_mode=Dispatch.ONE_TO_ALL, blocking=False)
verl/tests/single_controller/check_worker_alive/main.py:None
Route
foo_all_to_all
register(Dispatch.ALL_TO_ALL, blocking=False)
verl/tests/single_controller/test_worker_group_basics.py:None
Route
foo_custom
register(dispatch_mode={"dispatch_fn": two_to_all_dispatch_fn, "collect_fn": collect_all_to_all})
verl/tests/single_controller/test_worker_group_basics.py:None
Route
foo_one_to_all
register(Dispatch.ONE_TO_ALL, blocking=False)
verl/tests/single_controller/test_worker_group_basics.py:None
Route
foo_rank_zero
register(dispatch_mode=Dispatch.ALL_TO_ALL, execute_mode=Execute.RANK_ZERO)
verl/tests/single_controller/test_worker_group_basics.py:None
Route
generate_data_proto
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="infer"))
verl/tests/single_controller/test_device_mesh_register.py:None
Route
generate_nested_tensor
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="infer"))
verl/tests/single_controller/test_device_mesh_register.py:None
Route
generate_sequences
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="rollout"))
verl/verl/workers/megatron_workers.py:None
Route
generate_sequences
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="rollout"))
verl/verl/workers/fsdp_workers.py:None
Route
generate_sequences
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="rollout"), blocking=False)
verl/verl/experimental/vla/fsdp_workers.py:None
Route
generate_tensordict
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="infer"))
verl/tests/single_controller/test_device_mesh_register.py:None
Route
get
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/tests/single_controller/test_nested_worker.py:None
Route
get_all_state_ids
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/experimental/vla/workers/env/env_worker.py:None
Route
get_replica_rank
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/checkpoint_engine/base.py:None
Route
get_tensors
register(Dispatch.ONE_TO_ALL)
verl/tests/single_controller/test_ray_collectives.py:None
Route
infer_batch
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="train"), blocking=False)
verl/verl/workers/engine_workers.py:None
Route
init
register(Dispatch.ONE_TO_ALL)
verl/tests/single_controller/test_ray_collectives.py:None
Route
init_model
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/tests/single_controller/detached_worker/server.py:None
Route
init_model
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/megatron_workers.py:None
Route
init_model
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/fsdp_workers.py:None
Route
init_model
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/engine_workers.py:None
Route
init_model
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/experimental/vla/fsdp_workers.py:None
Route
init_simulator
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/experimental/vla/workers/env/env_worker.py:None
Route
init_worker
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/experimental/vla/workers/env/env_worker.py:None
Route
is_leader_rank
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/checkpoint_engine/base.py:None
Route
load_checkpoint
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/megatron_workers.py:None
Route
load_checkpoint
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/fsdp_workers.py:None
Route
load_checkpoint
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/engine_workers.py:None
Route
load_extern_type
deprecated(replacement="load_module(file_path); getattr(module, type_name)")
verl/verl/utils/import_utils.py:None
Route
load_pretrained_model
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/megatron_workers.py:None
Route
on_reqid_submitted
on(Input.Submitted, "#reqid-box")
verl/scripts/rollout_viewer.py:None
Route
on_search_submitted
on(Input.Submitted, "#search-box")
verl/scripts/rollout_viewer.py:None
Route
receive_tensors
register(Dispatch.ONE_TO_ALL, blocking=False)
verl/tests/single_controller/test_ray_collectives.py:None
Route
reset
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/engine_workers.py:None
Route
reset_envs_to_state_ids
register(dispatch_mode=make_nd_compute_dataproto_dispatch_fn(mesh_name="env"), blocking=False)
verl/verl/experimental/vla/workers/env/env_worker.py:None
Route
restore_model_from_cpu
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/experimental/separation/engine_workers.py:None
Route
retrieve_endpoint
app.post("/retrieve")
verl/examples/sglang_multiturn/search_r1_like/local_dense_retriever/retrieval_server.py:None
Route
save_checkpoint
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/megatron_workers.py:None
Route
save_checkpoint
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/fsdp_workers.py:None
Route
save_checkpoint
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/engine_workers.py:None
Route
save_merged_weights
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/scripts/megatron_merge_lora.py:None
Route
save_model_to_cpu
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/experimental/separation/engine_workers.py:None
Route
send_tensors
register(Dispatch.ONE_TO_ALL, blocking=False)
verl/tests/single_controller/test_ray_collectives.py:None
Route
set_loss_fn
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/engine_workers.py:None
Route
start_profile
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/utils/profiler/profile.py:None
Route
start_profile
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/megatron_workers.py:None
Route
start_profile
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/fsdp_workers.py:None
Route
stop_profile
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/utils/profiler/profile.py:None
Route
stop_profile
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/megatron_workers.py:None
Route
stop_profile
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/workers/fsdp_workers.py:None
Route
sub
register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)
verl/tests/single_controller/test_colocated_workers_fused.py:None
Route
sub
register(dispatch_mode=Dispatch.DP_COMPUTE_PROTO)
verl/tests/single_controller/test_colocated_workers.py:None
Route
sub
register(dispatch_mode=Dispatch.ALL_TO_ALL)
verl/tests/single_controller/test_fused_workers_on_cpu.py:None
Route
switch_to_rollout
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/experimental/vla/fsdp_workers.py:None
Route
switch_to_train
register(dispatch_mode=Dispatch.ONE_TO_ALL)
verl/verl/experimental/vla/fsdp_workers.py:None
Route
test_attn_implementation_passed_to_autoconfig
patch("transformers.AutoConfig.from_pretrained")
verl/tests/workers/test_fsdp_attn_implementation.py:None
Route
test_attn_implementation_passed_to_model
patch("transformers.AutoConfig.from_pretrained")
verl/tests/workers/test_fsdp_attn_implementation.py:None
Route
test_collect_merged_lora_params
pytest.mark.parametrize("lora_targets", (["q_proj", "v_proj"], "all-linear"))
verl/tests/utils/test_fsdp_lora_merge.py:None
Route
test_compute_timing_metrics
patch("verl.trainer.ppo.metric_utils._compute_response_info")
verl/tests/trainer/ppo/test_metric_utils_on_cpu.py:None
Route
test_continuous_score_consistency
pytest.mark.skipif(not os.environ.get("SANDBOX_FUSION_URL"), reason="SANDBOX_FUSION_URL environment variable n
verl/tests/utils/reward_score/test_sandbox_on_cpu.py:None
Route
test_critic_attn_implementation_override_functionality
patch("transformers.AutoConfig.from_pretrained")
verl/tests/workers/critic/test_special_dp_critic.py:None
Route
test_critic_attn_implementation_passed_to_autoconfig
patch("transformers.AutoConfig.from_pretrained")
verl/tests/workers/test_fsdp_attn_implementation.py:None
Route
test_discrete_mode
patch("verl.utils.profiler.torch_profile.get_torch_profiler")
verl/tests/utils/test_torch_profile.py:None
Route
test_get_npu_versions_cann_path_not_exists
patch("subprocess.run")
verl/tests/utils/test_check_ipc_version_support_on_npu.py:None
Route
test_get_npu_versions_info_file_not_exists
patch("subprocess.run")
verl/tests/utils/test_check_ipc_version_support_on_npu.py:None
next →
1–100 of 136, ranked by callers