residual_drift.py File Reference

Go to the source code of this file.

Namespaces
namespace	aquin
namespace	aquin.compute
namespace	aquin.compute.residual_drift

Functions
torch.Tensor	aquin.compute.residual_drift._cosine_distance_rows (torch.Tensor a, torch.Tensor b)
list[dict[str, Any]]	aquin.compute.residual_drift._layer_drift_rows (dict[int, torch.Tensor] base_acts, dict[int, torch.Tensor] ft_acts, int n_layers)
list[dict[str, Any]]	aquin.compute.residual_drift._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]	aquin.compute.residual_drift._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]	aquin.compute.residual_drift.run_llm_residual_drift (str model_id, str|Path checkpoint_path, list[str] prompts, *, str checkpoint_name="checkpoint")
dict[str, Any]	aquin.compute.residual_drift.run_residual_drift (str model_id, str|Path checkpoint_path, list[str]|None prompts=None, *, str|None checkpoint_name=None)
dict[str, Any]	aquin.compute.residual_drift.run_residual_drift_from_args (dict[str, Any] args)