Functions — github.com/Agent-RL/ReCall

hub / github.com/Agent-RL/ReCall / functions

Functions1,539 in github.com/Agent-RL/ReCall

⨍ Functions 1,539◇ Types & classes 237↳ Endpoints 28

↓ 1 callersFunction_fetch_tp_shard_tensor_gate_up

fetch gate_up tensor in tp shards

src/verl/models/llama/megatron/checkpoint_utils/llama_loader.py:142

↓ 1 callersFunction_fetch_tp_shard_tensor_qkv

fetch tensor in tp shards across mp_group

src/verl/models/llama/megatron/checkpoint_utils/llama_loader.py:168

↓ 1 callersFunction_fetch_tp_shard_tensor_vocab

fetch tensor in tp shards

src/verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader.py:106

↓ 1 callersFunction_fetch_tp_shard_tensor_vocab

fetch tensor in tp shards

src/verl/models/llama/megatron/checkpoint_utils/llama_loader.py:110

↓ 1 callersFunction_fix_a_slash_b

(string)

src/verl/utils/reward_score/prime_math/math_normalize.py:89

↓ 1 callersFunction_fix_fracs

(string)

src/verl/utils/reward_score/prime_math/math_normalize.py:57

↓ 1 callersFunction_fix_sqrt

(string)

src/verl/utils/reward_score/prime_math/math_normalize.py:114

↓ 1 callersFunction_flatten_dict

(raw: Dict[str, Any], *, sep: str)

src/verl/utils/tracking.py:173

↓ 1 callersMethod_forward_head

(self, hidden_states)

src/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:307

↓ 1 callersMethod_forward_head

(self, hidden_states)

src/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:610

↓ 1 callersMethod_forward_head

(self, hidden_states)

src/verl/models/llama/megatron/modeling_llama_megatron.py:307

↓ 1 callersMethod_forward_head

(self, hidden_states)

src/verl/models/llama/megatron/modeling_llama_megatron.py:565

↓ 1 callersMethod_forward_micro_batch

(self, micro_batch)

src/verl/workers/fsdp_workers.py:1025

↓ 1 callersMethod_generate_async

(self, input_list: List, batch_size=None, return_scores=False, **params)

src/flashrag/generator/openai_generator.py:83

↓ 1 callersMethod_generate_minibatch

(self, prompts: DataProto)

src/verl/workers/rollout/hf_rollout.py:51

↓ 1 callersMethod_get_batch_response

(self, input_list: List[List], batch_size, **params)

src/flashrag/generator/openai_generator.py:74

↓ 1 callersMethod_get_final_config

(self)

src/flashrag/config/config.py:87

↓ 1 callersMethod_get_free_port

(self)

src/verl/single_controller/base/worker.py:58

↓ 1 callersMethod_get_internal_config

(self)

src/flashrag/config/config.py:80

↓ 1 callersMethod_get_node_ip

(self)

src/verl/single_controller/base/worker.py:41

↓ 1 callersFunction_get_parallel_model_architecture_from_config

(config: PretrainedConfig, value=False)

src/verl/utils/model.py:266

↓ 1 callersMethod_get_response

(self, messages: List, **params)

src/flashrag/generator/openai_generator.py:66

↓ 1 callersMethod_init_device

(self)

src/flashrag/config/config.py:103

↓ 1 callersMethod_init_head

(self, config: Qwen2Config)

src/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:295

↓ 1 callersMethod_init_head

(self, config)

src/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:549

↓ 1 callersMethod_init_head

(self, config)

src/verl/models/llama/megatron/modeling_llama_megatron.py:295

↓ 1 callersMethod_init_head

(self, config)

src/verl/models/llama/megatron/modeling_llama_megatron.py:553

↓ 1 callersMethod_init_rope

(self)

src/verl/models/qwen2/megatron/layers/parallel_attention.py:208

↓ 1 callersMethod_init_rope

(self)

src/verl/models/llama/megatron/layers/parallel_attention.py:236

↓ 1 callersMethod_init_with_detached_workers

(self, worker_names)

src/verl/single_controller/ray/base.py:232

↓ 1 callersMethod_init_with_resource_pool

(self, resource_pool, ray_cls_with_init, bin_pack, detached)

src/verl/single_controller/ray/base.py:237

↓ 1 callersFunction_inject_implicit_mixed_number

Automatically make a mixed number evalable e.g. 7 3/4 => 7+3/4

src/verl/utils/reward_score/prime_math/__init__.py:126

↓ 1 callersFunction_inner_call_method

(_method)

src/verl/utils/reward_score/prime_code/testing_util.py:633

↓ 1 callersFunction_is_float

(num: str)

src/verl/utils/reward_score/prime_math/__init__.py:92

↓ 1 callersFunction_is_int

(x: float)

src/verl/utils/reward_score/prime_math/__init__.py:100

↓ 1 callersMethod_is_worker_alive

(self, worker)

src/verl/single_controller/base/worker_group.py:112

↓ 1 callersFunction_last_boxed_only_string

(string)

src/verl/utils/reward_score/prime_math/__init__.py:333

↓ 1 callersMethod_lexical_unit

(self, sents)

src/flashrag/refiner/selective_context_compressor.py:153

↓ 1 callersMethod_load_checkpoint

(self)

src/verl/trainer/ppo/ray_trainer.py:729

↓ 1 callersMethod_load_data

Load data from the provided dataset_path or directly download the file(TODO).

src/flashrag/dataset/dataset.py:111

↓ 1 callersMethod_load_embedding

(self, embedding_path, corpus_size, hidden_size)

src/flashrag/retriever/index_builder.py:216

↓ 1 callersMethod_load_model

(self)

src/flashrag/generator/multimodal_generator.py:70

↓ 1 callersMethod_load_model

r"""Load model and tokenizer for generator.

src/flashrag/generator/generator.py:363

↓ 1 callersMethod_load_params_to_cuda

(self, pp_rank, to_empty=False)

src/verl/workers/sharding_manager/megatron_vllm.py:113

↓ 1 callersFunction_make_causal_mask

Make causal mask used for bi-directional self-attention.

src/verl/utils/torch_functional.py:492

↓ 1 callersFunction_make_causal_mask

Make causal mask used for bi-directional self-attention.

src/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:48

↓ 1 callersFunction_make_causal_mask

Make causal mask used for bi-directional self-attention.

src/verl/models/llama/megatron/modeling_llama_megatron.py:48

↓ 1 callersFunction_map_each_response

(l)

src/verl/workers/rollout/sglang_rollout/sglang_rollout.py:63

↓ 1 callersFunction_materialize_futures

(*args, **kwargs)

src/verl/single_controller/base/decorator.py:384

↓ 1 callersMethod_maybe_log_val_generations

Log a table of validation samples to the configured logger (wandb or swanlab)

src/verl/trainer/ppo/ray_trainer.py:481

↓ 1 callersFunction_megatron_calc_layer_map

Calculate the mapping of global layer_idx to local layer_idx Returns: layer_map (Dict: int -> tuple(int, int, int)): mapping f

src/verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader_depracated.py:21

↓ 1 callersFunction_megatron_calc_layer_map

Calculate the mapping of global layer_idx to local layer_idx Returns: layer_map (Dict: int -> tuple(int, int, int)): mapping f

src/verl/models/qwen2/megatron/checkpoint_utils/qwen2_loader.py:21

↓ 1 callersFunction_megatron_calc_layer_map

Calculate the mapping of global layer_idx to local layer_idx Returns: layer_map (Dict: int -> tuple(int, int, int)): mapping f

src/verl/models/qwen2/megatron/checkpoint_utils/qwen2_saver.py:39

↓ 1 callersFunction_megatron_calc_layer_map

Calculate the mapping of global layer_idx to local layer_idx Returns: layer_map (Dict: int -> tuple(int, int, int)): mapping f

src/verl/models/mcore/saver.py:52

↓ 1 callersFunction_megatron_calc_layer_map

Calculate the mapping of global layer_idx to local layer_idx Returns: layer_map (Dict: int -> tuple(int, int, int)): mapping f

src/verl/models/mcore/loader.py:22

↓ 1 callersFunction_megatron_calc_layer_map

Calculate the mapping of global layer_idx to local layer_idx Returns: layer_map (Dict: int -> tuple(int, int, int)): mapping f

src/verl/models/llama/megatron/checkpoint_utils/llama_saver.py:39

↓ 1 callersFunction_megatron_calc_layer_map

Calculate the mapping of global layer_idx to local layer_idx Returns: layer_map (Dict: int -> tuple(int, int, int)): mapping f

src/verl/models/llama/megatron/checkpoint_utils/llama_loader.py:23

↓ 1 callersFunction_megatron_calc_layer_map

Calculate the mapping of global layer_idx to local layer_idx Returns: layer_map (Dict: int -> tuple(int, int, int)): mapping f

src/verl/models/llama/megatron/checkpoint_utils/llama_loader_depracated.py:23

↓ 1 callersMethod_merge_external_config

(self)

src/flashrag/config/config.py:73

↓ 1 callersFunction_mkdir

hdfs mkdir

src/verl/utils/hdfs_io.py:75

↓ 1 callersMethod_normalize_config_bsz

(self)

src/verl/trainer/fsdp_sft_trainer.py:111

↓ 1 callersMethod_optimizer_step

(self)

src/verl/workers/critic/dp_critic.py:115

↓ 1 callersMethod_optimizer_step

(self)

src/verl/workers/actor/dp_actor.py:161

↓ 1 callersFunction_parse_latex

Attempts to parse latex to an expression sympy can read.

src/verl/utils/reward_score/prime_math/__init__.py:74

↓ 1 callersFunction_post_process_outputs

(tokenizer, output)

src/verl/workers/rollout/sglang_rollout/sglang_rollout.py:61

↓ 1 callersFunction_pre_process_inputs

(pad_token_id, prompt_token_ids: torch.Tensor)

src/verl/workers/rollout/sglang_rollout/sglang_rollout.py:52

↓ 1 callersFunction_pre_process_inputs

(pad_token_id, prompt_token_ids: torch.Tensor)

src/verl/workers/rollout/vllm_rollout/fire_vllm_rollout.py:50

↓ 1 callersFunction_pre_process_inputs

(pad_token_id, prompt_token_ids: torch.Tensor)

src/verl/workers/rollout/vllm_rollout/vllm_rollout.py:49

↓ 1 callersMethod_prepare_decoder_attention_mask

(self, attention_mask, input_shape, inputs_embeds)

src/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:101

↓ 1 callersMethod_prepare_decoder_attention_mask

(self, attention_mask, input_shape, inputs_embeds)

src/verl/models/llama/megatron/modeling_llama_megatron.py:101

↓ 1 callersMethod_prepare_dir

(self)

src/flashrag/config/config.py:238

↓ 1 callersMethod_prepare_model

(self)

src/flashrag/refiner/selective_context_compressor.py:71

↓ 1 callersMethod_prepare_phrase_tokenizer

(self)

src/flashrag/refiner/selective_context_compressor.py:60

↓ 1 callersMethod_read_files_and_process

(self)

src/verl/utils/dataset/multiturn_sft_dataset.py:61

↓ 1 callersMethod_read_files_and_tokenize

(self)

src/verl/utils/dataset/rm_dataset.py:85

↓ 1 callersMethod_read_files_and_tokenize

(self)

src/verl/utils/dataset/sft_dataset.py:76

↓ 1 callersFunction_remove_right_units

(string)

src/verl/utils/reward_score/prime_math/math_normalize.py:104

↓ 1 callersMethod_save_cache

(self)

src/flashrag/retriever/retriever.py:161

↓ 1 callersMethod_save_checkpoint

(self)

src/verl/trainer/ppo/ray_trainer.py:683

↓ 1 callersMethod_save_embedding

(self, all_embeddings)

src/flashrag/retriever/index_builder.py:220

↓ 1 callersMethod_search

(self, query: str, num: int = None, return_score=False)

src/flashrag/retriever/retriever.py:271

↓ 1 callersMethod_set_additional_key

(self)

src/flashrag/config/config.py:121

↓ 1 callersMethod_set_seed

(self)

src/flashrag/config/config.py:256

↓ 1 callersFunction_str_to_int

(x: str)

src/verl/utils/reward_score/prime_math/__init__.py:120

↓ 1 callersFunction_strip_string

(string)

src/verl/utils/reward_score/prime_math/math_normalize.py:129

↓ 1 callersMethod_switch_chat_template

(self, data: DataProto)

src/verl/workers/fsdp_workers.py:1095

↓ 1 callersFunction_sympy_parse

Parses an expression with sympy.

src/verl/utils/reward_score/prime_math/__init__.py:65

↓ 1 callersFunction_transform_params_to_json_serializable

(x, convert_list_to_dict: bool)

src/verl/utils/tracking.py:153

↓ 1 callersMethod_validate_config

(self)

src/verl/trainer/ppo/ray_trainer.py:297

↓ 1 callersMethod_validate_config

Validate config options not implemented for Megatron backend

src/verl/workers/critic/megatron_critic.py:69

↓ 1 callersMethod_validate_config

Validate config options not implemented for Megatron backend

src/verl/workers/actor/megatron_actor.py:138

↓ 1 callersFunctionacc_reward

(predict_str: str, ground_truth: str)

src/verl/utils/reward_score/geo3k.py:25

↓ 1 callersMethodadd_source

(self, result: Union[list, tuple], retriever)

src/flashrag/retriever/retriever.py:635

↓ 1 callersFunctionall_gather_tensor

(local_tensor: Tensor, group: Optional[dist.ProcessGroup] = None, async_op: bool = False)

src/verl/utils/ulysses.py:154

↓ 1 callersMethodallgather_params

allgather params of all pp ranks. Return a list of handles

src/verl/workers/sharding_manager/megatron_vllm.py:139

↓ 1 callersFunctionapply_checkpoint_wrapper

Wrap each block of the encoder to enable checkpointing.

src/flashrag/generator/fid.py:125

↓ 1 callersFunctionapply_kl_penalty

(data: DataProto, kl_ctrl: core_algos.AdaptiveKLController, kl_penalty='kl')

src/verl/trainer/ppo/ray_trainer.py:141

↓ 1 callersFunctionapply_rotary_pos_emb

(q, k, cos, sin, position_ids)

src/verl/models/llama/megatron/layers/parallel_attention.py:155

↓ 1 callersFunctionapply_rotary_pos_emb_rmpad_flash

(q, k, cos, sin, cu_seqlens, max_seqlen)

src/verl/models/qwen2/megatron/layers/parallel_attention.py:304

↓ 1 callersFunctionapply_rotary_pos_emb_rmpad_flash

(q, k, cos, sin, cu_seqlens, max_seqlen)

src/verl/models/llama/megatron/layers/parallel_attention.py:360

← previous next →401–500 of 1,539, ranked by callers