causal_trace.py File Reference

Go to the source code of this file.

Namespaces
namespace	aquin
namespace	aquin.compute
namespace	aquin.compute.causal_trace

Functions
torch.Tensor	aquin.compute.causal_trace._project_unembed (model, torch.Tensor resid)
str	aquin.compute.causal_trace._format_prompt (HookedTransformer model, str prompt)
	aquin.compute.causal_trace._get_embeds (tokens, str model_id="llama-3.2-1b")
	aquin.compute.causal_trace._corrupt_all (tokens, noise_scale, str model_id="llama-3.2-1b")
	aquin.compute.causal_trace._corrupt_single_pos (tokens, pos, noise_scale, str model_id="llama-3.2-1b")
	aquin.compute.causal_trace.run_trace (str prompt, str target_token, str model_id="llama-3.2-1b", float noise_scale=3.0, int n_noise_runs=10)
	aquin.compute.causal_trace.run_prompt_attribution (prompt, response, prompt_tokens, response_tokens, sig_prompt_tis, sig_response_tis, str model_id="llama-3.2-1b", float noise_scale=3.0, int n_noise_runs=5)
	aquin.compute.causal_trace.run_logit_lens (str prompt, str model_id="llama-3.2-1b", int top_k=5)
	aquin.compute.causal_trace._resolve_chat_model (str model_id, Any|None model=None)
torch.device	aquin.compute.causal_trace._generation_device (Any model)
str	aquin.compute.causal_trace._decode_chat_completion (Any model, torch.Tensor input_ids, torch.Tensor output_ids)
str	aquin.compute.causal_trace.run_chat (str prompt, str model_id="llama-3.2-1b", int max_new_tokens=200, float temperature=0.7, Any|None model=None)
	aquin.compute.causal_trace.stream_chat (str prompt, str model_id="llama-3.2-1b", int max_new_tokens=200, float temperature=0.7)

Variables
	aquin.compute.causal_trace.DEVICE = resolve_compute_device()
	aquin.compute.causal_trace.DTYPE = default_dtype_for_device(DEVICE)