Index

A

ablation (NeuroSurgeon.Masking.mask_layer.MaskLayer property)

C

calculate_l0() (NeuroSurgeon.Masking.hardconcrete_layer.HardConcreteLayer method) (NeuroSurgeon.Masking.mask_layer.MaskLayer method) calculate_max_l0() (NeuroSurgeon.Masking.mask_layer.MaskLayer method) CircuitConfig (class in NeuroSurgeon.Models.model_configs) CircuitModel (class in NeuroSurgeon.Models.circuit_model) CircuitProbeConfig (class in NeuroSurgeon.Probing.probe_configs)

compute_l0_statistics() (NeuroSurgeon.Models.circuit_model.CircuitModel method) ContSparseConv1d (class in NeuroSurgeon.Masking.contsparse_layer) ContSparseConv2d (class in NeuroSurgeon.Masking.contsparse_layer) ContSparseGPTConv1D (class in NeuroSurgeon.Masking.contsparse_layer) ContSparseLayer (class in NeuroSurgeon.Masking.contsparse_layer) ContSparseLinear (class in NeuroSurgeon.Masking.contsparse_layer)

F

force_resample (NeuroSurgeon.Masking.contsparse_layer.ContSparseLayer property) (NeuroSurgeon.Masking.hardconcrete_layer.HardConcreteLayer property) (NeuroSurgeon.Masking.magprune_layer.MagPruneLayer property) forward() (NeuroSurgeon.Masking.contsparse_layer.ContSparseGPTConv1D method) (NeuroSurgeon.Masking.contsparse_layer.ContSparseLinear method) (NeuroSurgeon.Masking.hardconcrete_layer.HardConcreteGPTConv1D method) (NeuroSurgeon.Masking.hardconcrete_layer.HardConcreteLinear method) (NeuroSurgeon.Masking.magprune_layer.MagPruneGPTConv1D method) (NeuroSurgeon.Masking.magprune_layer.MagPruneLinear method) (NeuroSurgeon.Masking.mask_layer.MaskLayer method) (NeuroSurgeon.Models.circuit_model.CircuitModel method) (NeuroSurgeon.Probing.residual_update_model.ResidualUpdateModel method) (NeuroSurgeon.Probing.subnetwork_probe.SubnetworkProbe method)

from_layer() (NeuroSurgeon.Masking.contsparse_layer.ContSparseConv1d class method) (NeuroSurgeon.Masking.contsparse_layer.ContSparseConv2d class method) (NeuroSurgeon.Masking.contsparse_layer.ContSparseGPTConv1D class method) (NeuroSurgeon.Masking.contsparse_layer.ContSparseLinear class method) (NeuroSurgeon.Masking.hardconcrete_layer.HardConcreteConv1d class method) (NeuroSurgeon.Masking.hardconcrete_layer.HardConcreteConv2d class method) (NeuroSurgeon.Masking.hardconcrete_layer.HardConcreteGPTConv1D class method) (NeuroSurgeon.Masking.hardconcrete_layer.HardConcreteLinear class method) (NeuroSurgeon.Masking.magprune_layer.MagPruneConv1d class method) (NeuroSurgeon.Masking.magprune_layer.MagPruneConv2d class method) (NeuroSurgeon.Masking.magprune_layer.MagPruneGPTConv1D class method) (NeuroSurgeon.Masking.magprune_layer.MagPruneLinear class method) (NeuroSurgeon.Masking.mask_layer.MaskLayer method)

H

HardConcreteConv1d (class in NeuroSurgeon.Masking.hardconcrete_layer) HardConcreteConv2d (class in NeuroSurgeon.Masking.hardconcrete_layer)

HardConcreteGPTConv1D (class in NeuroSurgeon.Masking.hardconcrete_layer) HardConcreteLayer (class in NeuroSurgeon.Masking.hardconcrete_layer) HardConcreteLinear (class in NeuroSurgeon.Masking.hardconcrete_layer)

M

MagPruneConv1d (class in NeuroSurgeon.Masking.magprune_layer) MagPruneConv2d (class in NeuroSurgeon.Masking.magprune_layer) MagPruneGPTConv1D (class in NeuroSurgeon.Masking.magprune_layer) MagPruneLayer (class in NeuroSurgeon.Masking.magprune_layer) MagPruneLinear (class in NeuroSurgeon.Masking.magprune_layer) mask_bias (NeuroSurgeon.Masking.mask_layer.MaskLayer property) mask_init_percentage (NeuroSurgeon.Masking.hardconcrete_layer.HardConcreteLayer property) mask_init_value (NeuroSurgeon.Masking.contsparse_layer.ContSparseLayer property) mask_unit (NeuroSurgeon.Masking.mask_layer.MaskLayer property) MaskLayer (class in NeuroSurgeon.Masking.mask_layer) module NeuroSurgeon.Masking NeuroSurgeon.Masking.contsparse_layer NeuroSurgeon.Masking.hardconcrete_layer NeuroSurgeon.Masking.magprune_layer NeuroSurgeon.Masking.mask_layer NeuroSurgeon.Models NeuroSurgeon.Models.circuit_model NeuroSurgeon.Models.model_configs NeuroSurgeon.Probing.probe_configs NeuroSurgeon.Probing.residual_update_model NeuroSurgeon.Probing.subnetwork_probe

N

NeuroSurgeon.Masking module NeuroSurgeon.Masking.contsparse_layer module NeuroSurgeon.Masking.hardconcrete_layer module NeuroSurgeon.Masking.magprune_layer module NeuroSurgeon.Masking.mask_layer module NeuroSurgeon.Models module

NeuroSurgeon.Models.circuit_model module NeuroSurgeon.Models.model_configs module NeuroSurgeon.Probing.probe_configs module NeuroSurgeon.Probing.residual_update_model module NeuroSurgeon.Probing.subnetwork_probe module

P

prune_percentage (NeuroSurgeon.Masking.magprune_layer.MagPruneLayer property)

R

reset_parameters() (NeuroSurgeon.Masking.contsparse_layer.ContSparseLinear method) (NeuroSurgeon.Masking.hardconcrete_layer.HardConcreteLinear method) (NeuroSurgeon.Masking.magprune_layer.MagPruneLinear method) (NeuroSurgeon.Masking.mask_layer.MaskLayer method)

ResidualUpdateModel (class in NeuroSurgeon.Probing.residual_update_model) ResidualUpdateModelConfig (class in NeuroSurgeon.Probing.probe_configs)

S

set_ablate_mode() (NeuroSurgeon.Models.circuit_model.CircuitModel method)

SubnetworkProbe (class in NeuroSurgeon.Probing.subnetwork_probe) SubnetworkProbeConfig (class in NeuroSurgeon.Probing.probe_configs)

T

temperature (NeuroSurgeon.Masking.contsparse_layer.ContSparseLayer property) (NeuroSurgeon.Models.circuit_model.CircuitModel property) train() (NeuroSurgeon.Masking.mask_layer.MaskLayer method) (NeuroSurgeon.Models.circuit_model.CircuitModel method) (NeuroSurgeon.Probing.residual_update_model.ResidualUpdateModel method) (NeuroSurgeon.Probing.subnetwork_probe.SubnetworkProbe method)

U

use_masks (NeuroSurgeon.Masking.mask_layer.MaskLayer property)

use_masks() (NeuroSurgeon.Models.circuit_model.CircuitModel method)