Index

A

activation (devinterp.mechinterp.activations.ActivationProbe attribute) ActivationProbe (class in devinterp.mechinterp.activations) add_module() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) add_param_group() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgnht.SGNHT method)

append() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) apply() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

B

BetweenLayerCovarianceAccumulator (class in devinterp.slt.cov) bfloat16() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

buffers() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

C

call_super_init (devinterp.mechinterp.hooks.HookedList attribute) (devinterp.mechinterp.hooks.HookedSequential attribute) children() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) collect_cache() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) compile() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

configure_sweep() (devinterp.vis_utils.EpsilonBetaAnalyzer method) count() (devinterp.utils.Outputs method) CovarianceAccumulator (class in devinterp.slt.cov) cpu() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) cuda() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

D

device (devinterp.slt.mala.MalaAcceptanceRate attribute) devinterp module devinterp.backends module devinterp.backends.default module devinterp.backends.default.slt module devinterp.backends.default.slt.sampler module devinterp.backends.tpu module devinterp.backends.tpu.slt module devinterp.backends.tpu.slt.sampler module devinterp.mechinterp module devinterp.mechinterp.activations module devinterp.mechinterp.hooks module devinterp.optim module devinterp.optim.sgld module devinterp.optim.sgnht module

devinterp.slt module devinterp.slt.callback module devinterp.slt.cov module devinterp.slt.gradient module devinterp.slt.llc module devinterp.slt.loss module devinterp.slt.mala module devinterp.slt.norms module devinterp.slt.sampler module devinterp.slt.trace module devinterp.slt.wbic module devinterp.utils module devinterp.vis_utils module double() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) dump_patches (devinterp.mechinterp.hooks.HookedList attribute) (devinterp.mechinterp.hooks.HookedSequential attribute)

E

EpsilonBetaAnalyzer (class in devinterp.vis_utils) eval() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

extend() (devinterp.mechinterp.hooks.HookedList method) extra_repr() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

F

finalize() (devinterp.slt.callback.SamplerCallback method) (devinterp.slt.gradient.GradientDistribution method) (devinterp.slt.loss.OnlineLossStatistics method) (devinterp.slt.mala.MalaAcceptanceRate method) (devinterp.slt.norms.GradientNorm method) (devinterp.slt.norms.NoiseNorm method) (devinterp.slt.norms.WeightNorm method) (devinterp.slt.trace.OnlineTraceStatistics method)

float() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) forward() (devinterp.mechinterp.hooks.HookedList method)

G

get_buffer() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) get_extra_state() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) get_parameter() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) get_results() (devinterp.slt.callback.SamplerCallback method) (devinterp.slt.cov.BetweenLayerCovarianceAccumulator method) (devinterp.slt.cov.CovarianceAccumulator method) (devinterp.slt.cov.WithinHeadCovarianceAccumulator method) (devinterp.slt.gradient.GradientDistribution method) (devinterp.slt.llc.LLCEstimator method) (devinterp.slt.llc.OnlineLLCEstimator method) (devinterp.slt.loss.OnlineLossStatistics method) (devinterp.slt.norms.GradientNorm method) (devinterp.slt.norms.NoiseNorm method) (devinterp.slt.norms.WeightNorm method) (devinterp.slt.trace.OnlineTraceStatistics method) (devinterp.slt.wbic.OnlineWBICEstimator method)

get_submodule() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) GradientDistribution (class in devinterp.slt.gradient) GradientNorm (class in devinterp.slt.norms)

H

half() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

hook() (in module devinterp.mechinterp.hooks) HookedList (class in devinterp.mechinterp.hooks) HookedSequential (class in devinterp.mechinterp.hooks)

I

index() (devinterp.utils.Outputs method) insert() (devinterp.mechinterp.hooks.HookedList method)

ipu() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

L

layer_location (devinterp.mechinterp.activations.ActivationProbe attribute) learning_rate (devinterp.slt.mala.MalaAcceptanceRate attribute) LLCEstimator (class in devinterp.slt.llc) load_state_dict() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgnht.SGNHT method)

loss (devinterp.utils.Outputs attribute) loss_hist_by_draw() (devinterp.slt.loss.OnlineLossStatistics method)

M

mala_acceptance_probability() (in module devinterp.slt.mala) MalaAcceptanceRate (class in devinterp.slt.mala) model (devinterp.mechinterp.activations.ActivationProbe attribute) module devinterp devinterp.backends devinterp.backends.default devinterp.backends.default.slt devinterp.backends.default.slt.sampler devinterp.backends.tpu devinterp.backends.tpu.slt devinterp.backends.tpu.slt.sampler devinterp.mechinterp devinterp.mechinterp.activations devinterp.mechinterp.hooks devinterp.optim devinterp.optim.sgld devinterp.optim.sgnht devinterp.slt devinterp.slt.callback devinterp.slt.cov devinterp.slt.gradient devinterp.slt.llc devinterp.slt.loss devinterp.slt.mala devinterp.slt.norms devinterp.slt.sampler devinterp.slt.trace devinterp.slt.wbic devinterp.utils devinterp.vis_utils

modules() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

N

named_buffers() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) named_children() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) named_modules() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

named_parameters() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) nbeta (devinterp.slt.mala.MalaAcceptanceRate attribute) neuron_location (devinterp.mechinterp.activations.ActivationProbe attribute) NoiseNorm (class in devinterp.slt.norms) num_chains (devinterp.slt.mala.MalaAcceptanceRate attribute) num_draws (devinterp.slt.mala.MalaAcceptanceRate attribute)

O

OnlineLLCEstimator (class in devinterp.slt.llc) OnlineLossStatistics (class in devinterp.slt.loss) OnlineTraceStatistics (class in devinterp.slt.trace) OnlineWBICEstimator (class in devinterp.slt.wbic)

OptimizerPostHook (devinterp.optim.sgld.SGLD attribute) (devinterp.optim.sgnht.SGNHT attribute) OptimizerPreHook (devinterp.optim.sgld.SGLD attribute) (devinterp.optim.sgnht.SGNHT attribute) Outputs (class in devinterp.utils)

P

parameters() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) plot() (devinterp.slt.gradient.GradientDistribution method) (devinterp.vis_utils.EpsilonBetaAnalyzer method)

prepare_input() (in module devinterp.utils) prepend_dict() (in module devinterp.mechinterp.hooks)

R

register_backward_hook() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) register_buffer() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) register_forward_hook() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) register_forward_pre_hook() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) register_full_backward_hook() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) register_full_backward_pre_hook() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) register_load_state_dict_post_hook() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgnht.SGNHT method) register_load_state_dict_pre_hook() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgnht.SGNHT method)

register_module() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) register_parameter() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) register_state_dict_post_hook() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgnht.SGNHT method) register_state_dict_pre_hook() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgnht.SGNHT method) register_step_post_hook() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgnht.SGNHT method) register_step_pre_hook() (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgnht.SGNHT method) requires_grad_() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) run_with_cache() (devinterp.mechinterp.hooks.HookedSequential method)

S

sample() (in module devinterp.backends.default.slt.sampler) (in module devinterp.backends.tpu.slt.sampler) sample_at_draw() (devinterp.slt.trace.OnlineTraceStatistics method) sample_single_chain() (in module devinterp.backends.tpu.slt.sampler) SamplerCallback (class in devinterp.slt.callback) save_fig() (devinterp.vis_utils.EpsilonBetaAnalyzer method) set_extra_state() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) set_seed() (in module devinterp.utils)

SGLD (class in devinterp.optim.sgld) SGNHT (class in devinterp.optim.sgnht) share_memory() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) state_dict() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgnht.SGNHT method) step() (devinterp.optim.sgld.SGLD method) sweep() (devinterp.vis_utils.EpsilonBetaAnalyzer method)

T

T_destination (devinterp.mechinterp.hooks.HookedList attribute) (devinterp.mechinterp.hooks.HookedSequential attribute) to() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) to_empty() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

train() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) training (devinterp.mechinterp.hooks.HookedList attribute) (devinterp.mechinterp.hooks.HookedSequential attribute) type() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

W

WeightNorm (class in devinterp.slt.norms)

WithinHeadCovarianceAccumulator (class in devinterp.slt.cov)

X

xpu() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method)

Z

zero_grad() (devinterp.mechinterp.hooks.HookedList method) (devinterp.mechinterp.hooks.HookedSequential method) (devinterp.optim.sgld.SGLD method) (devinterp.optim.sgnht.SGNHT method)