Methoddo_updates(cls, frozen_noiser_params, noiser_params, params, base_keys, fitnesses, iterinfos, es_map)
src/hyperscalees/noiser/eggroll_baseline_subtraction.py:150
Methoddo_updates(cls, frozen_noiser_params, noiser_params, params, base_keys, fitnesses, iterinfos, es_map)
src/hyperscalees/noiser/eggroll.py:129
Methoddo_updates(cls, frozen_noiser_params, noiser_params, params, base_keys, fitnesses, iterinfos, es_map)
src/hyperscalees/noiser/base_noiser.py:36
Methoddo_updates(cls, frozen_noiser_params, noiser_params, params, base_keys, fitnesses, iterinfos, es_map)
src/hyperscalees/noiser/alteggroll.py:128
Methodget_noisy_standard(cls, frozen_noiser_params, noiser_params, param, base_key, iterinfo)
src/hyperscalees/noiser/open_es.py:78
Methodget_noisy_standard(cls, frozen_noiser_params, noiser_params, param, base_key, iterinfo)
src/hyperscalees/noiser/eggroll_baseline_subtraction.py:114
Methodget_noisy_standard(cls, frozen_noiser_params, noiser_params, param, base_key, iterinfo)
src/hyperscalees/noiser/eggroll.py:98
Methodget_noisy_standard(cls, frozen_noiser_params, noiser_params, param, base_key, iterinfo)
src/hyperscalees/noiser/base_noiser.py:28
Methodget_noisy_standard(cls, frozen_noiser_params, noiser_params, param, base_key, iterinfo)
src/hyperscalees/noiser/alteggroll.py:97
Methodinner_loop(cls, r, w, k, v, a, b, s, length, new_starts)
src/hyperscalees/models/llm/rwkv7.py:200
Methodinner_loop(cls, r, w, k, v, a, b, s, length, new_starts)
src/hyperscalees/models/llm/rwkv7.py:257
Methodinner_loop(cls, r, w, k, v, a, b, s, length, new_starts)
src/hyperscalees/models/llm/rwkv7.py:282
Methodinner_loop(cls, r, k, v, w, time_first, s, length, new_starts)
src/hyperscalees/models/llm/qrwkv6.py:205
Methodrand_init(cls, key, n_embd, space_gymnax, use_bias, dtype, *args, **kwargs)
src/hyperscalees/models/rl.py:61
Methodrand_init(cls, key, n_embd, obs_space, act_space, n_layers, use_bias, activation, have_critic, dtype, *args, **kwargs)
src/hyperscalees/models/rl.py:94
Methodrand_init(cls, key, shape, scale, raw_value, dtype, *args, **kwargs)
src/hyperscalees/models/common.py:68
Methodrand_init(cls, key, in_dim, out_dim, dtype, *args, **kwargs)
src/hyperscalees/models/common.py:99
Methodrand_init(cls, key, in_dim, out_dim, dtype, *args, **kwargs)
src/hyperscalees/models/common.py:113
Methodrand_init(cls, key, in_dim, out_dim, use_bias, dtype, *args, **kwargs)
src/hyperscalees/models/common.py:127
Methodrand_init(cls, key, in_dim, out_dim, hidden_dims, use_bias, activation, dtype, *args, **kwargs)
src/hyperscalees/models/common.py:147
Methodrandomize_weights(cls, key, n_layer, n_embd, vocab_size, config, dtype)
src/hyperscalees/models/llm/llm.py:71
Functionsingle_example_loss(params, old_params, noiser_params, is_input_token, tokens, advantage)
llm_experiments/do_grpo.py:182
Functionsingle_example_loss(params, old_params, noiser_params, is_input_token, tokens, advantage)
llm_experiments/do_grpo_multi_gpu.py:200