Functions
torch.Tensor	_cosine_distance_rows (torch.Tensor a, torch.Tensor b)
list[dict[str, Any]]	_layer_drift_rows (dict[int, torch.Tensor] base_acts, dict[int, torch.Tensor] ft_acts, int n_layers)
list[dict[str, Any]]	_per_probe_rows (dict[int, torch.Tensor] base_acts, dict[int, torch.Tensor] ft_acts, list[dict[str, Any]] layer_rows, list[str] probes)
dict[str, Any]	_finalize_payload (*, str model_id, str mode, str activation_mode, str checkpoint_name, str checkpoint_path, list[str] prompts, list[dict[str, Any]] layer_profile, list[dict[str, Any]] per_probe, int\|None step=None)
dict[str, Any]	run_llm_residual_drift (str model_id, str\|Path checkpoint_path, list[str] prompts, *, str checkpoint_name="checkpoint")
dict[str, Any]	run_residual_drift (str model_id, str\|Path checkpoint_path, list[str]\|None prompts=None, *, str\|None checkpoint_name=None)
dict[str, Any]	run_residual_drift_from_args (dict[str, Any] args)

Function Documentation

◆ _cosine_distance_rows()

torch.Tensor _cosine_distance_rows	(	torch.Tensor	a,
		torch.Tensor	b )

protected

Per-row cosine distance. a, b: (n, d).

Definition at line 19 of file residual_drift.py.

dict[str, Any] _finalize_payload	(	*	,
		str	model_id,
		str	mode,
		str	activation_mode,
		str	checkpoint_name,
		str	checkpoint_path,
		list[str]	prompts,
		list[dict[str, Any]]	layer_profile,
		list[dict[str, Any]]	per_probe,
		int \| None	step = None )

protected

Definition at line 81 of file residual_drift.py.

list[dict[str, Any]] _layer_drift_rows	(	dict[int, torch.Tensor]	base_acts,
		dict[int, torch.Tensor]	ft_acts,
		int	n_layers )

protected

Definition at line 26 of file residual_drift.py.

list[dict[str, Any]] _per_probe_rows	(	dict[int, torch.Tensor]	base_acts,
		dict[int, torch.Tensor]	ft_acts,
		list[dict[str, Any]]	layer_rows,
		list[str]	probes )