Index

A

add_dot_products_() (devinterp.optim.metrics.Metrics method) add_note() (devinterp.slt.lm_loss.NonFiniteLogitsError method) add_param_group() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) (devinterp.optim.sgnht.SGNHT method)

add_sum_squared_() (devinterp.optim.metrics.Metrics method) aggregate() (devinterp.optim.metrics.Metrics static method) arrays (devinterp.slt.writing.ZarrWriter attribute) arrays_meta (devinterp.slt.zarr_schema.ZarrSchema attribute)

B

batch_corrcoef() (in module devinterp.slt.covariance) batch_cov() (in module devinterp.slt.covariance) batch_size (devinterp.slt.config.SamplerConfig attribute)

bif() (in module devinterp.slt.bif) bounding_box_size (devinterp.slt.config.SamplerConfig attribute) buffer_device (devinterp.slt.writing.ZarrWriter attribute)

C

chain_buffer_size (devinterp.slt.writing.ZarrWriter attribute) CompositePreconditioner (class in devinterp.optim.preconditioner) CompositePrior (class in devinterp.optim.prior) compute_bif() (in module devinterp.slt.bif) compute_llc() (in module devinterp.slt.llc) compute_loss() (devinterp.slt.observables.Observable method) compute_per_token_loss() (in module devinterp.slt.lm_loss)

compute_susceptibilities() (in module devinterp.slt.susceptibilities) context_length (devinterp.slt.observables.Observable attribute) copy() (devinterp.slt.config.SamplerConfig method) count() (devinterp.optim.preconditioner.PreconditionerCoefs method) CountSketch (class in devinterp.optim.sketch) create_hierarchy() (devinterp.slt.zarr_schema.ZarrSchema method) create_param_masks() (in module devinterp.slt.weight_restrictions)

D

DataArraySpec (class in devinterp.slt.zarr_schema) DeterministicShuffledSampler (class in devinterp.slt.observables) devinterp module devinterp.optim module devinterp.optim.metrics module devinterp.optim.preconditioner module devinterp.optim.prior module devinterp.optim.sgld module devinterp.optim.sgmcmc module devinterp.optim.sgnht module devinterp.optim.sketch module devinterp.optim.utils module devinterp.slt module devinterp.slt.bif module devinterp.slt.config module

devinterp.slt.covariance module devinterp.slt.llc module devinterp.slt.lm_loss module devinterp.slt.observables module devinterp.slt.sampler module devinterp.slt.sampling module devinterp.slt.susceptibilities module devinterp.slt.weight_restrictions module devinterp.slt.writing module devinterp.slt.zarr_schema module devinterp.utils module, [1] dims (devinterp.slt.zarr_schema.DataArraySpec attribute) distance (devinterp.optim.metrics.Metrics attribute) DOT_FIELDS (devinterp.optim.metrics.Metrics attribute) dot_grad_noise (devinterp.optim.metrics.Metrics attribute) dot_grad_prior (devinterp.optim.metrics.Metrics attribute) dot_prior_noise (devinterp.optim.metrics.Metrics attribute) dtype_str (devinterp.slt.zarr_schema.DataArraySpec attribute)

E

epoch_mode (devinterp.slt.config.SamplerConfig attribute)

G

GaussianPrior (class in devinterp.optim.prior) get_coefficients() (devinterp.optim.preconditioner.Preconditioner method) get_metrics() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) get_params() (devinterp.optim.sgmcmc.SGMCMC method) get_sketches() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method)

grad() (devinterp.optim.prior.GaussianPrior method) (devinterp.optim.prior.Prior method) grad_coef (devinterp.optim.preconditioner.PreconditionerCoefs attribute) grad_correction (devinterp.optim.preconditioner.PreconditionerCoefs attribute) gradient_accumulation_steps (devinterp.slt.config.SamplerConfig attribute) group_attrs (devinterp.slt.zarr_schema.ZarrSchema attribute)

H

hash_indices (devinterp.optim.sketch.CountSketch attribute)

hash_signs (devinterp.optim.sketch.CountSketch attribute)

I

IdentityPreconditioner (class in devinterp.optim.preconditioner) index() (devinterp.optim.preconditioner.PreconditionerCoefs method) init_noise (devinterp.slt.config.SamplerConfig attribute) init_seed (devinterp.slt.config.SamplerConfig attribute)

initialize() (devinterp.optim.prior.GaussianPrior method) (devinterp.optim.prior.Prior method) input_ids (devinterp.slt.observables.Observable attribute) iter_group_metrics() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method)

K

key (devinterp.optim.prior.CompositePrior attribute) (devinterp.optim.prior.GaussianPrior attribute) (devinterp.optim.prior.Prior attribute) (devinterp.optim.prior.UniformPrior attribute)

L

llc() (in module devinterp.slt.llc) llc_weight_decay (devinterp.slt.config.SamplerConfig attribute) lm_cross_entropy_loss() (in module devinterp.slt.lm_loss) lm_forward_logits() (in module devinterp.slt.lm_loss) load_state_dict() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) (devinterp.optim.sgnht.SGNHT method)

localization (devinterp.optim.metrics.Metrics attribute) (devinterp.optim.sketch.SketchBuffer attribute) (devinterp.slt.config.SamplerConfig attribute) lr (devinterp.slt.config.SamplerConfig attribute)

M

make_evaluate_fn() (in module devinterp.slt.lm_loss) MaskPreconditioner (class in devinterp.optim.preconditioner) match_sampling_input_ids_across_chains (devinterp.slt.config.SamplerConfig attribute) Metrics (class in devinterp.optim.metrics) MicroCallback (class in devinterp.slt.sampler) model_computed_fields (devinterp.slt.config.SamplerConfig attribute) model_config (devinterp.slt.config.SamplerConfig attribute) model_construct() (devinterp.slt.config.SamplerConfig class method) model_copy() (devinterp.slt.config.SamplerConfig method) model_dump() (devinterp.slt.config.SamplerConfig method) model_dump_json() (devinterp.slt.config.SamplerConfig method) model_extra (devinterp.slt.config.SamplerConfig property) model_fields (devinterp.slt.config.SamplerConfig attribute) model_fields_set (devinterp.slt.config.SamplerConfig property) model_json_schema() (devinterp.slt.config.SamplerConfig class method) model_parametrized_name() (devinterp.slt.config.SamplerConfig class method) model_post_init() (devinterp.slt.config.SamplerConfig method) model_rebuild() (devinterp.slt.config.SamplerConfig class method) model_validate() (devinterp.slt.config.SamplerConfig class method) model_validate_json() (devinterp.slt.config.SamplerConfig class method) model_validate_strings() (devinterp.slt.config.SamplerConfig class method) module devinterp devinterp.optim devinterp.optim.metrics devinterp.optim.preconditioner devinterp.optim.prior devinterp.optim.sgld devinterp.optim.sgmcmc devinterp.optim.sgnht devinterp.optim.sketch devinterp.optim.utils devinterp.slt devinterp.slt.bif devinterp.slt.config devinterp.slt.covariance devinterp.slt.llc devinterp.slt.lm_loss devinterp.slt.observables devinterp.slt.sampler devinterp.slt.sampling devinterp.slt.susceptibilities devinterp.slt.weight_restrictions devinterp.slt.writing devinterp.slt.zarr_schema devinterp.utils, [1]

N

n_beta (devinterp.slt.config.SamplerConfig attribute) n_samples (devinterp.slt.observables.Observable attribute) NHTPreconditioning (class in devinterp.optim.preconditioner) noise (devinterp.optim.metrics.Metrics attribute) (devinterp.optim.sketch.SketchBuffer attribute) noise_coef (devinterp.optim.preconditioner.PreconditionerCoefs attribute) noise_level (devinterp.slt.config.SamplerConfig attribute)

NonFiniteLogitsError NORM_FIELDS (devinterp.optim.metrics.Metrics attribute) num_burnin_steps (devinterp.slt.config.SamplerConfig attribute) num_chains (devinterp.slt.config.SamplerConfig attribute) num_draws (devinterp.slt.config.SamplerConfig attribute) num_init_loss_batches (devinterp.slt.config.SamplerConfig attribute) num_steps_bw_draws (devinterp.slt.config.SamplerConfig attribute) numel (devinterp.optim.metrics.Metrics attribute)

O

obs_id (devinterp.slt.observables.Observable attribute)

Observable (class in devinterp.slt.observables) overall_coef (devinterp.optim.preconditioner.PreconditionerCoefs attribute)

P

Preconditioner (class in devinterp.optim.preconditioner) PreconditionerCoefs (class in devinterp.optim.preconditioner) preview_weight_restriction() (in module devinterp.slt.weight_restrictions)

Prior (class in devinterp.optim.prior) prior (devinterp.optim.metrics.Metrics property) prior_coef (devinterp.optim.preconditioner.PreconditionerCoefs attribute) push() (devinterp.slt.writing.ZarrWriter method)

Q

QUANTITIES (devinterp.optim.sketch.SketchBuffer attribute)

R

register_load_state_dict_post_hook() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) (devinterp.optim.sgnht.SGNHT method) register_load_state_dict_pre_hook() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) (devinterp.optim.sgnht.SGNHT method) register_state_dict_post_hook() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) (devinterp.optim.sgnht.SGNHT method) register_state_dict_pre_hook() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) (devinterp.optim.sgnht.SGNHT method)

register_step_post_hook() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) (devinterp.optim.sgnht.SGNHT method) register_step_pre_hook() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) (devinterp.optim.sgnht.SGNHT method) rmsprop_sgld() (devinterp.optim.sgmcmc.SGMCMC class method) RMSpropPreconditioner (class in devinterp.optim.preconditioner)

S

sample() (in module devinterp.slt.sampling) sample_single_chain() (in module devinterp.slt.sampler) SamplerConfig (class in devinterp.slt.config) sampling_method (devinterp.slt.config.SamplerConfig attribute) sampling_method_kwargs (devinterp.slt.config.SamplerConfig attribute) save_metrics (devinterp.slt.config.SamplerConfig attribute) scaled_grad (devinterp.optim.metrics.Metrics attribute) (devinterp.optim.sketch.SketchBuffer attribute) scatter_into_() (devinterp.optim.sketch.CountSketch method) SGLD (class in devinterp.optim.sgld) sgld() (devinterp.optim.sgmcmc.SGMCMC class method) SGMCMC (class in devinterp.optim.sgmcmc)

SGNHT (class in devinterp.optim.sgnht) sgnht() (devinterp.optim.sgmcmc.SGMCMC class method) shape (devinterp.slt.zarr_schema.DataArraySpec attribute) shuffle (devinterp.slt.config.SamplerConfig attribute) sketch() (devinterp.optim.sketch.CountSketch method) SketchBuffer (class in devinterp.optim.sketch) sqrt_norms_() (devinterp.optim.metrics.Metrics method) state_dict() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) (devinterp.optim.sgnht.SGNHT method) step() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) susceptibilities() (in module devinterp.slt.susceptibilities)

T

to() (devinterp.optim.metrics.Metrics method)

to_obs_id() (in module devinterp.slt.observables) tokenize_and_concatenate() (in module devinterp.utils), [1]

U

UniformPrior (class in devinterp.optim.prior)

unscaled_grad (devinterp.optim.metrics.Metrics attribute) (devinterp.optim.sketch.SketchBuffer attribute)

W

weight_decay (devinterp.optim.metrics.Metrics attribute) (devinterp.optim.sketch.SketchBuffer attribute)

with_traceback() (devinterp.slt.lm_loss.NonFiniteLogitsError method) write() (devinterp.slt.writing.ZarrWriter method)

X

xr_corrcoef_with_torch_backend() (in module devinterp.slt.covariance)

Z

ZarrSchema (class in devinterp.slt.zarr_schema) ZarrWriter (class in devinterp.slt.writing) zero_() (devinterp.optim.metrics.Metrics method)

zero_grad() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgmcmc.SGMCMC method) (devinterp.optim.sgnht.SGNHT method)