Index A | B | C | E | F | G | H | I | L | M | P | Q | R | S | T | U | W A adjust_identity() (in module problm_solver.samplers) AdjustFn (in module problm_solver.samplers) alpha (problm_solver.data.Hyperparams attribute) B base_live_state (problm_solver.samplers.SamplerContext attribute) BeamSampler (class in problm_solver.samplers) branch_sampler (problm_solver.data.LLMOutputDataFull attribute) BranchSampler (class in problm_solver.samplers) C candidate_tokens_to_id_logprobs() (in module problm_solver.samplers) change_context() (problm_solver.llama_interface.Model method) context (problm_solver.data.LLMOutputDataFull attribute) context_tokens (problm_solver.samplers.SamplerContext attribute) E ensure_models_dir() (in module problm_solver.cli) ensure_probs_dir() (in module problm_solver.cli) ensure_responses_dir() (in module problm_solver.cli) eval_tokens (problm_solver.samplers.SamplerContext attribute) F future_logprob() (problm_solver.samplers.BeamSampler method) (problm_solver.samplers.BranchSampler method) (problm_solver.samplers.MetropolisSampler method) future_logprob_from_context() (problm_solver.samplers.BeamSampler method) (problm_solver.samplers.BranchSampler method) G generate_data() (problm_solver.llama_interface.Model method) generate_with_sampler() (problm_solver.llama_interface.Model method) get_adjusted_path() (in module problm_solver.cli) get_math500_results_path() (in module problm_solver.cli) get_probs_path() (in module problm_solver.cli) get_responses_path() (in module problm_solver.cli) get_rng() (problm_solver.random.RandomManager method) H Hyperparams (class in problm_solver.data) hyperparams (problm_solver.data.LLMOutputDataFull attribute) I id_logprobs_to_candidate_tokens() (in module problm_solver.samplers) L list_models() (in module problm_solver.cli) LLMNextTokenData (class in problm_solver.data) LLMOutputData (class in problm_solver.data) LLMOutputDataFull (class in problm_solver.data) LLMTokenData (class in problm_solver.data) load_live_state (problm_solver.samplers.SamplerContext attribute) load_live_state() (problm_solver.llama_interface.Model method) M main() (in module problm_solver.cli) max_proposals (problm_solver.samplers.BranchSampler property) (problm_solver.samplers.MetropolisSampler property) max_tokens (problm_solver.data.Hyperparams attribute) MetropolisSampler (class in problm_solver.samplers) Model (class in problm_solver.llama_interface) module problm_solver.analysis.probabilities problm_solver.cli problm_solver.data problm_solver.llama_interface problm_solver.random problm_solver.samplers P prev_probs (problm_solver.samplers.SamplerContext attribute) prob_of_token() (in module problm_solver.analysis.probabilities) problm_solver.analysis.probabilities module problm_solver.cli module problm_solver.data module problm_solver.llama_interface module problm_solver.random module problm_solver.samplers module Q query() (problm_solver.llama_interface.Model method) query_branch (problm_solver.samplers.SamplerContext attribute) query_branch() (problm_solver.llama_interface.Model method) query_branch_from_live (problm_solver.samplers.SamplerContext attribute) query_branch_from_live() (problm_solver.llama_interface.Model method) query_branches_from_live_batch (problm_solver.samplers.SamplerContext attribute) query_branches_from_live_batch() (problm_solver.llama_interface.Model method) query_log_probs() (problm_solver.llama_interface.Model method) query_log_probs_next_token() (problm_solver.llama_interface.Model method) query_log_probs_next_token_ids() (problm_solver.llama_interface.Model method) query_n_times() (problm_solver.llama_interface.Model method) query_next_id (problm_solver.samplers.SamplerContext attribute) query_next_ids_from_live (problm_solver.samplers.SamplerContext attribute) R RandomManager (class in problm_solver.random) read() (problm_solver.data.LLMNextTokenData method) (problm_solver.data.LLMOutputData method) (problm_solver.data.LLMOutputDataFull method) (problm_solver.data.LLMTokenData method) reset() (problm_solver.samplers.BeamSampler method) (problm_solver.samplers.BranchSampler method) (problm_solver.samplers.MetropolisSampler method) resolve_rng() (in module problm_solver.random) response_probabilities (problm_solver.data.LLMOutputDataFull attribute) response_topk (problm_solver.data.LLMOutputDataFull attribute) RNGLike (in module problm_solver.random) S sample_from_logprobs() (in module problm_solver.analysis.probabilities) sample_token() (problm_solver.llama_interface.Model method) SampleLowTempNucleus (class in problm_solver.samplers) SamplePowerDist (class in problm_solver.samplers) SamplerContext (class in problm_solver.samplers) sampling_method (problm_solver.data.LLMOutputDataFull attribute) save_live_state (problm_solver.samplers.SamplerContext attribute) save_live_state() (problm_solver.llama_interface.Model method) seed (problm_solver.random.RandomManager property) should_continue() (problm_solver.samplers.BeamSampler method) (problm_solver.samplers.BranchSampler method) (problm_solver.samplers.MetropolisSampler method) spawn_rng() (problm_solver.random.RandomManager method) step() (problm_solver.samplers.BeamSampler method) (problm_solver.samplers.BranchSampler method) (problm_solver.samplers.MetropolisSampler method) supports_token_beam (problm_solver.samplers.BeamSampler attribute) (problm_solver.samplers.BranchSampler attribute) T test_dataset_adjusted() (problm_solver.llama_interface.Model method) token_id_probs (problm_solver.samplers.SamplerContext attribute) TokenProbError top_k (problm_solver.data.Hyperparams attribute) top_p (problm_solver.data.Hyperparams attribute) U ui_gen_data() (in module problm_solver.cli) ui_generate_low_temp() (in module problm_solver.cli) ui_generate_power_mcmc() (in module problm_solver.cli) ui_get_probs() (in module problm_solver.cli) ui_math500() (in module problm_solver.cli) ui_save_data() (in module problm_solver.cli) ui_save_token_data() (in module problm_solver.cli) ui_select_function() (in module problm_solver.cli) ui_select_model() (in module problm_solver.cli) UnexpectedFunctionError W write() (problm_solver.data.LLMNextTokenData method) (problm_solver.data.LLMOutputData method) (problm_solver.data.LLMOutputDataFull method) (problm_solver.data.LLMTokenData method)