pyvene.models.intervenable_base

pyvene.models.intervenable_base#

Functions

build_intervenable_model(config, model, **kwargs)

Factory design pattern for different types of intervenable models.

Classes

BaseModel(config, model, backend, **kwargs)

Base model class for sharing static vars and methods.

IntervenableModel(config, model, **kwargs)

Intervenable model via pyvene native backend (hook-based).

IntervenableModelOutput([original_outputs, ...])

Output of the IntervenableModel, including original outputs, intervened outputs, and collected activations.

IntervenableNdifModel(config, model, **kwargs)

Intervenable model via ndif backend.