[{"data":1,"prerenderedAt":-1},["ShallowReactive",2],{"skill:01M6JHANR5W858S38XTYHCFRMP":3,"skill-same:01M6JHANR5W858S38XTYHCFRMP":90},{"skillNo":4,"name":5,"nameZh":6,"description":7,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":12,"repositoryNo":13,"starsCount":14,"licenseSpdxId":15,"licenseFileUrl":6,"sourceHtmlUrl":16,"categories":17,"keywords":18,"publishedAt":39,"linkOnly":40,"markdownContent":41,"markdownBody":42,"frontMatter":43,"refName":54,"commitSha":55,"skillDirectory":56,"forksCount":57,"repositoryHtmlUrl":58,"repositoryHomepageUrl":6,"ownerHtmlUrl":59,"ownerBlogUrl":6,"ownerDescription":60,"updatedAt":61,"install":62,"resources":75},"01M6JHANR5W858S38XTYHCFRMP","nnsight-remote-interpretability",null,"Provides guidance for interpreting and manipulating neural network internals using nnsight with optional NDIF remote execution. Use when needing to run interpretability experiments on massive models (70B+) without local GPU resources, or when working with any PyTorch architecture.","transaurus\u002Fstaging-davila7-claude-code-templates","transaurus","https:\u002F\u002Favatars.githubusercontent.com\u002Fu\u002F210882017?v=4","01M6JH1R6ZCRXGJCTDP5ZKZBE6","Page Turner Agent","01M6JH1RBBR2P6Y0XWDGM4B0G7",0,"MIT","https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fai-research\u002Fmechanistic-interpretability-nnsight\u002FSKILL.md",[],[19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38],"remote","Interpretability","nnsight remote interpretability","nnsight","local","for","orchestra","with","resources","neural network","experiments","Neural","use","model","execution","network","gpu","run","pytorch","architecture","2026-11-10 10:12:30",false,"---\nname: nnsight-remote-interpretability\ndescription: Provides guidance for interpreting and manipulating neural network internals using nnsight with optional NDIF remote execution. Use when needing to run interpretability experiments on massive models (70B+) without local GPU resources, or when working with any PyTorch architecture.\nversion: 1.0.0\nauthor: Orchestra Research\nlicense: MIT\ntags: [nnsight, NDIF, Remote Execution, Mechanistic Interpretability, Model Internals]\ndependencies: [nnsight>=0.5.0, torch>=2.0.0]\n---\n\n# nnsight: Transparent Access to Neural Network Internals\n\nnnsight (\u002Fɛn.saɪt\u002F) enables researchers to interpret and manipulate the internals of any PyTorch model, with the unique capability of running the same code locally on small models or remotely on massive models (70B+) via NDIF.\n\n**GitHub**: [ndif-team\u002Fnnsight](https:\u002F\u002Fgithub.com\u002Fndif-team\u002Fnnsight) (730+ stars)\n**Paper**: [NNsight and NDIF: Democratizing Access to Foundation Model Internals](https:\u002F\u002Farxiv.org\u002Fabs\u002F2407.14561) (ICLR 2025)\n\n## Key Value Proposition\n\n**Write once, run anywhere**: The same interpretability code works on GPT-2 locally or Llama-3.1-405B remotely. Just toggle `remote=True`.\n\n```python\n# Local execution (small model)\nwith model.trace(\"Hello world\"):\n    hidden = model.transformer.h[5].output[0].save()\n\n# Remote execution (massive model) - same code!\nwith model.trace(\"Hello world\", remote=True):\n    hidden = model.model.layers[40].output[0].save()\n```\n\n## When to Use nnsight\n\n**Use nnsight when you need to:**\n- Run interpretability experiments on models too large for local GPUs (70B, 405B)\n- Work with any PyTorch architecture (transformers, Mamba, custom models)\n- Perform multi-token generation interventions\n- Share activations between different prompts\n- Access full model internals without reimplementation\n\n**Consider alternatives when:**\n- You want consistent API across models → Use **TransformerLens**\n- You need declarative, shareable interventions → Use **pyvene**\n- You're training SAEs → Use **SAELens**\n- You only work with small models locally → **TransformerLens** may be simpler\n\n## Installation\n\n```bash\n# Basic installation\npip install nnsight\n\n# For vLLM support\npip install \"nnsight[vllm]\"\n```\n\nFor remote NDIF execution, sign up at [login.ndif.us](https:\u002F\u002Flogin.ndif.us) for an API key.\n\n## Core Concepts\n\n### LanguageModel Wrapper\n\n```python\nfrom nnsight import LanguageModel\n\n# Load model (uses HuggingFace under the hood)\nmodel = LanguageModel(\"openai-community\u002Fgpt2\", device_map=\"auto\")\n\n# For larger models\nmodel = LanguageModel(\"meta-llama\u002FLlama-3.1-8B\", device_map=\"auto\")\n```\n\n### Tracing Context\n\nThe `trace` context manager enables deferred execution - operations are collected into a computation graph:\n\n```python\nfrom nnsight import LanguageModel\n\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")\n\nwith model.trace(\"The Eiffel Tower is in\") as tracer:\n    # Access any module's output\n    hidden_states = model.transformer.h[5].output[0].save()\n\n    # Access attention patterns\n    attn = model.transformer.h[5].attn.attn_dropout.input[0][0].save()\n\n    # Modify activations\n    model.transformer.h[8].output[0][:] = 0  # Zero out layer 8\n\n    # Get final output\n    logits = model.output.save()\n\n# After context exits, access saved values\nprint(hidden_states.shape)  # [batch, seq, hidden]\n```\n\n### Proxy Objects\n\nInside `trace`, module accesses return Proxy objects that record operations:\n\n```python\nwith model.trace(\"Hello\"):\n    # These are all Proxy objects - operations are deferred\n    h5_out = model.transformer.h[5].output[0]  # Proxy\n    h5_mean = h5_out.mean(dim=-1)              # Proxy\n    h5_saved = h5_mean.save()                   # Save for later access\n```\n\n## Workflow 1: Activation Analysis\n\n### Step-by-Step\n\n```python\nfrom nnsight import LanguageModel\nimport torch\n\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")\n\nprompt = \"The capital of France is\"\n\nwith model.trace(prompt) as tracer:\n    # 1. Collect activations from multiple layers\n    layer_outputs = []\n    for i in range(12):  # GPT-2 has 12 layers\n        layer_out = model.transformer.h[i].output[0].save()\n        layer_outputs.append(layer_out)\n\n    # 2. Get attention patterns\n    attn_patterns = []\n    for i in range(12):\n        # Access attention weights (after softmax)\n        attn = model.transformer.h[i].attn.attn_dropout.input[0][0].save()\n        attn_patterns.append(attn)\n\n    # 3. Get final logits\n    logits = model.output.save()\n\n# 4. Analyze outside context\nfor i, layer_out in enumerate(layer_outputs):\n    print(f\"Layer {i} output shape: {layer_out.shape}\")\n    print(f\"Layer {i} norm: {layer_out.norm().item():.3f}\")\n\n# 5. Find top predictions\nprobs = torch.softmax(logits[0, -1], dim=-1)\ntop_tokens = probs.topk(5)\nfor token, prob in zip(top_tokens.indices, top_tokens.values):\n    print(f\"{model.tokenizer.decode(token)}: {prob.item():.3f}\")\n```\n\n### Checklist\n- [ ] Load model with LanguageModel wrapper\n- [ ] Use trace context for operations\n- [ ] Call `.save()` on values you need after context\n- [ ] Access saved values outside context\n- [ ] Use `.shape`, `.norm()`, etc. for analysis\n\n## Workflow 2: Activation Patching\n\n### Step-by-Step\n\n```python\nfrom nnsight import LanguageModel\nimport torch\n\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")\n\nclean_prompt = \"The Eiffel Tower is in\"\ncorrupted_prompt = \"The Colosseum is in\"\n\n# 1. Get clean activations\nwith model.trace(clean_prompt) as tracer:\n    clean_hidden = model.transformer.h[8].output[0].save()\n\n# 2. Patch clean into corrupted run\nwith model.trace(corrupted_prompt) as tracer:\n    # Replace layer 8 output with clean activations\n    model.transformer.h[8].output[0][:] = clean_hidden\n\n    patched_logits = model.output.save()\n\n# 3. Compare predictions\nparis_token = model.tokenizer.encode(\" Paris\")[0]\nrome_token = model.tokenizer.encode(\" Rome\")[0]\n\npatched_probs = torch.softmax(patched_logits[0, -1], dim=-1)\nprint(f\"Paris prob: {patched_probs[paris_token].item():.3f}\")\nprint(f\"Rome prob: {patched_probs[rome_token].item():.3f}\")\n```\n\n### Systematic Patching Sweep\n\n```python\ndef patch_layer_position(layer, position, clean_cache, corrupted_prompt):\n    \"\"\"Patch single layer\u002Fposition from clean to corrupted.\"\"\"\n    with model.trace(corrupted_prompt) as tracer:\n        # Get current activation\n        current = model.transformer.h[layer].output[0]\n\n        # Patch only specific position\n        current[:, position, :] = clean_cache[layer][:, position, :]\n\n        logits = model.output.save()\n\n    return logits\n\n# Sweep over all layers and positions\nresults = torch.zeros(12, seq_len)\nfor layer in range(12):\n    for pos in range(seq_len):\n        logits = patch_layer_position(layer, pos, clean_hidden, corrupted)\n        results[layer, pos] = compute_metric(logits)\n```\n\n## Workflow 3: Remote Execution with NDIF\n\nRun the same experiments on massive models without local GPUs.\n\n### Step-by-Step\n\n```python\nfrom nnsight import LanguageModel\n\n# 1. Load large model (will run remotely)\nmodel = LanguageModel(\"meta-llama\u002FLlama-3.1-70B\")\n\n# 2. Same code, just add remote=True\nwith model.trace(\"The meaning of life is\", remote=True) as tracer:\n    # Access internals of 70B model!\n    layer_40_out = model.model.layers[40].output[0].save()\n    logits = model.output.save()\n\n# 3. Results returned from NDIF\nprint(f\"Layer 40 shape: {layer_40_out.shape}\")\n\n# 4. Generation with interventions\nwith model.trace(remote=True) as tracer:\n    with tracer.invoke(\"What is 2+2?\"):\n        # Intervene during generation\n        model.model.layers[20].output[0][:, -1, :] *= 1.5\n\n    output = model.generate(max_new_tokens=50)\n```\n\n### NDIF Setup\n\n1. Sign up at [login.ndif.us](https:\u002F\u002Flogin.ndif.us)\n2. Get API key\n3. Set environment variable or pass to nnsight:\n\n```python\nimport os\nos.environ[\"NDIF_API_KEY\"] = \"your_key\"\n\n# Or configure directly\nfrom nnsight import CONFIG\nCONFIG.API_KEY = \"your_key\"\n```\n\n### Available Models on NDIF\n\n- Llama-3.1-8B, 70B, 405B\n- DeepSeek-R1 models\n- Various open-weight models (check [ndif.us](https:\u002F\u002Fndif.us) for current list)\n\n## Workflow 4: Cross-Prompt Activation Sharing\n\nShare activations between different inputs in a single trace.\n\n```python\nfrom nnsight import LanguageModel\n\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")\n\nwith model.trace() as tracer:\n    # First prompt\n    with tracer.invoke(\"The cat sat on the\"):\n        cat_hidden = model.transformer.h[6].output[0].save()\n\n    # Second prompt - inject cat's activations\n    with tracer.invoke(\"The dog ran through the\"):\n        # Replace with cat's activations at layer 6\n        model.transformer.h[6].output[0][:] = cat_hidden\n        dog_with_cat = model.output.save()\n\n# The dog prompt now has cat's internal representations\n```\n\n## Workflow 5: Gradient-Based Analysis\n\nAccess gradients during backward pass.\n\n```python\nfrom nnsight import LanguageModel\nimport torch\n\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")\n\nwith model.trace(\"The quick brown fox\") as tracer:\n    # Save activations and enable gradient\n    hidden = model.transformer.h[5].output[0].save()\n    hidden.retain_grad()\n\n    logits = model.output\n\n    # Compute loss on specific token\n    target_token = model.tokenizer.encode(\" jumps\")[0]\n    loss = -logits[0, -1, target_token]\n\n    # Backward pass\n    loss.backward()\n\n# Access gradients\ngrad = hidden.grad\nprint(f\"Gradient shape: {grad.shape}\")\nprint(f\"Gradient norm: {grad.norm().item():.3f}\")\n```\n\n**Note**: Gradient access not supported for vLLM or remote execution.\n\n## Common Issues & Solutions\n\n### Issue: Module path differs between models\n```python\n# GPT-2 structure\nmodel.transformer.h[5].output[0]\n\n# LLaMA structure\nmodel.model.layers[5].output[0]\n\n# Solution: Check model structure\nprint(model._model)  # See actual module names\n```\n\n### Issue: Forgetting to save\n```python\n# WRONG: Value not accessible outside trace\nwith model.trace(\"Hello\"):\n    hidden = model.transformer.h[5].output[0]  # Not saved!\n\nprint(hidden)  # Error or wrong value\n\n# RIGHT: Call .save()\nwith model.trace(\"Hello\"):\n    hidden = model.transformer.h[5].output[0].save()\n\nprint(hidden)  # Works!\n```\n\n### Issue: Remote timeout\n```python\n# For long operations, increase timeout\nwith model.trace(\"prompt\", remote=True, timeout=300) as tracer:\n    # Long operation...\n```\n\n### Issue: Memory with many saved activations\n```python\n# Only save what you need\nwith model.trace(\"prompt\"):\n    # Don't save everything\n    for i in range(100):\n        model.transformer.h[i].output[0].save()  # Memory heavy!\n\n    # Better: save specific layers\n    key_layers = [0, 5, 11]\n    for i in key_layers:\n        model.transformer.h[i].output[0].save()\n```\n\n### Issue: vLLM gradient limitation\n```python\n# vLLM doesn't support gradients\n# Use standard execution for gradient analysis\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")  # Not vLLM\n```\n\n## Key API Reference\n\n| Method\u002FProperty | Purpose |\n|-----------------|---------|\n| `model.trace(prompt, remote=False)` | Start tracing context |\n| `proxy.save()` | Save value for access after trace |\n| `proxy[:]` | Slice\u002Findex proxy (assignment patches) |\n| `tracer.invoke(prompt)` | Add prompt within trace |\n| `model.generate(...)` | Generate with interventions |\n| `model.output` | Final model output logits |\n| `model._model` | Underlying HuggingFace model |\n\n## Comparison with Other Tools\n\n| Feature | nnsight | TransformerLens | pyvene |\n|---------|---------|-----------------|--------|\n| Any architecture | Yes | Transformers only | Yes |\n| Remote execution | Yes (NDIF) | No | No |\n| Consistent API | No | Yes | Yes |\n| Deferred execution | Yes | No | No |\n| HuggingFace native | Yes | Reimplemented | Yes |\n| Shareable configs | No | No | Yes |\n\n## Reference Documentation\n\nFor detailed API documentation, tutorials, and advanced usage, see the `references\u002F` folder:\n\n| File | Contents |\n|------|----------|\n| [references\u002FREADME.md](references\u002FREADME.md) | Overview and quick start guide |\n| [references\u002Fapi.md](references\u002Fapi.md) | Complete API reference for LanguageModel, tracing, proxy objects |\n| [references\u002Ftutorials.md](references\u002Ftutorials.md) | Step-by-step tutorials for local and remote interpretability |\n\n## External Resources\n\n### Tutorials\n- [Getting Started](https:\u002F\u002Fnnsight.net\u002Fstart\u002F)\n- [Features Overview](https:\u002F\u002Fnnsight.net\u002Ffeatures\u002F)\n- [Remote Execution](https:\u002F\u002Fnnsight.net\u002Fnotebooks\u002Ffeatures\u002Fremote_execution\u002F)\n- [Applied Tutorials](https:\u002F\u002Fnnsight.net\u002Fapplied_tutorials\u002F)\n\n### Official Documentation\n- [Official Docs](https:\u002F\u002Fnnsight.net\u002Fdocumentation\u002F)\n- [NDIF Info](https:\u002F\u002Fndif.us\u002F)\n- [Community Forum](https:\u002F\u002Fdiscuss.ndif.us\u002F)\n\n### Papers\n- [NNsight and NDIF Paper](https:\u002F\u002Farxiv.org\u002Fabs\u002F2407.14561) - Fiotto-Kaufman et al. (ICLR 2025)\n\n## Architecture Support\n\nnnsight works with any PyTorch model:\n- **Transformers**: GPT-2, LLaMA, Mistral, etc.\n- **State Space Models**: Mamba\n- **Vision Models**: ViT, CLIP\n- **Custom architectures**: Any nn.Module\n\nThe key is knowing the module structure to access the right components.\n","# nnsight: Transparent Access to Neural Network Internals\n\nnnsight (\u002Fɛn.saɪt\u002F) enables researchers to interpret and manipulate the internals of any PyTorch model, with the unique capability of running the same code locally on small models or remotely on massive models (70B+) via NDIF.\n\n**GitHub**: [ndif-team\u002Fnnsight](https:\u002F\u002Fgithub.com\u002Fndif-team\u002Fnnsight) (730+ stars)\n**Paper**: [NNsight and NDIF: Democratizing Access to Foundation Model Internals](https:\u002F\u002Farxiv.org\u002Fabs\u002F2407.14561) (ICLR 2025)\n\n## Key Value Proposition\n\n**Write once, run anywhere**: The same interpretability code works on GPT-2 locally or Llama-3.1-405B remotely. Just toggle `remote=True`.\n\n```python\n# Local execution (small model)\nwith model.trace(\"Hello world\"):\n    hidden = model.transformer.h[5].output[0].save()\n\n# Remote execution (massive model) - same code!\nwith model.trace(\"Hello world\", remote=True):\n    hidden = model.model.layers[40].output[0].save()\n```\n\n## When to Use nnsight\n\n**Use nnsight when you need to:**\n- Run interpretability experiments on models too large for local GPUs (70B, 405B)\n- Work with any PyTorch architecture (transformers, Mamba, custom models)\n- Perform multi-token generation interventions\n- Share activations between different prompts\n- Access full model internals without reimplementation\n\n**Consider alternatives when:**\n- You want consistent API across models → Use **TransformerLens**\n- You need declarative, shareable interventions → Use **pyvene**\n- You're training SAEs → Use **SAELens**\n- You only work with small models locally → **TransformerLens** may be simpler\n\n## Installation\n\n```bash\n# Basic installation\npip install nnsight\n\n# For vLLM support\npip install \"nnsight[vllm]\"\n```\n\nFor remote NDIF execution, sign up at [login.ndif.us](https:\u002F\u002Flogin.ndif.us) for an API key.\n\n## Core Concepts\n\n### LanguageModel Wrapper\n\n```python\nfrom nnsight import LanguageModel\n\n# Load model (uses HuggingFace under the hood)\nmodel = LanguageModel(\"openai-community\u002Fgpt2\", device_map=\"auto\")\n\n# For larger models\nmodel = LanguageModel(\"meta-llama\u002FLlama-3.1-8B\", device_map=\"auto\")\n```\n\n### Tracing Context\n\nThe `trace` context manager enables deferred execution - operations are collected into a computation graph:\n\n```python\nfrom nnsight import LanguageModel\n\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")\n\nwith model.trace(\"The Eiffel Tower is in\") as tracer:\n    # Access any module's output\n    hidden_states = model.transformer.h[5].output[0].save()\n\n    # Access attention patterns\n    attn = model.transformer.h[5].attn.attn_dropout.input[0][0].save()\n\n    # Modify activations\n    model.transformer.h[8].output[0][:] = 0  # Zero out layer 8\n\n    # Get final output\n    logits = model.output.save()\n\n# After context exits, access saved values\nprint(hidden_states.shape)  # [batch, seq, hidden]\n```\n\n### Proxy Objects\n\nInside `trace`, module accesses return Proxy objects that record operations:\n\n```python\nwith model.trace(\"Hello\"):\n    # These are all Proxy objects - operations are deferred\n    h5_out = model.transformer.h[5].output[0]  # Proxy\n    h5_mean = h5_out.mean(dim=-1)              # Proxy\n    h5_saved = h5_mean.save()                   # Save for later access\n```\n\n## Workflow 1: Activation Analysis\n\n### Step-by-Step\n\n```python\nfrom nnsight import LanguageModel\nimport torch\n\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")\n\nprompt = \"The capital of France is\"\n\nwith model.trace(prompt) as tracer:\n    # 1. Collect activations from multiple layers\n    layer_outputs = []\n    for i in range(12):  # GPT-2 has 12 layers\n        layer_out = model.transformer.h[i].output[0].save()\n        layer_outputs.append(layer_out)\n\n    # 2. Get attention patterns\n    attn_patterns = []\n    for i in range(12):\n        # Access attention weights (after softmax)\n        attn = model.transformer.h[i].attn.attn_dropout.input[0][0].save()\n        attn_patterns.append(attn)\n\n    # 3. Get final logits\n    logits = model.output.save()\n\n# 4. Analyze outside context\nfor i, layer_out in enumerate(layer_outputs):\n    print(f\"Layer {i} output shape: {layer_out.shape}\")\n    print(f\"Layer {i} norm: {layer_out.norm().item():.3f}\")\n\n# 5. Find top predictions\nprobs = torch.softmax(logits[0, -1], dim=-1)\ntop_tokens = probs.topk(5)\nfor token, prob in zip(top_tokens.indices, top_tokens.values):\n    print(f\"{model.tokenizer.decode(token)}: {prob.item():.3f}\")\n```\n\n### Checklist\n- [ ] Load model with LanguageModel wrapper\n- [ ] Use trace context for operations\n- [ ] Call `.save()` on values you need after context\n- [ ] Access saved values outside context\n- [ ] Use `.shape`, `.norm()`, etc. for analysis\n\n## Workflow 2: Activation Patching\n\n### Step-by-Step\n\n```python\nfrom nnsight import LanguageModel\nimport torch\n\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")\n\nclean_prompt = \"The Eiffel Tower is in\"\ncorrupted_prompt = \"The Colosseum is in\"\n\n# 1. Get clean activations\nwith model.trace(clean_prompt) as tracer:\n    clean_hidden = model.transformer.h[8].output[0].save()\n\n# 2. Patch clean into corrupted run\nwith model.trace(corrupted_prompt) as tracer:\n    # Replace layer 8 output with clean activations\n    model.transformer.h[8].output[0][:] = clean_hidden\n\n    patched_logits = model.output.save()\n\n# 3. Compare predictions\nparis_token = model.tokenizer.encode(\" Paris\")[0]\nrome_token = model.tokenizer.encode(\" Rome\")[0]\n\npatched_probs = torch.softmax(patched_logits[0, -1], dim=-1)\nprint(f\"Paris prob: {patched_probs[paris_token].item():.3f}\")\nprint(f\"Rome prob: {patched_probs[rome_token].item():.3f}\")\n```\n\n### Systematic Patching Sweep\n\n```python\ndef patch_layer_position(layer, position, clean_cache, corrupted_prompt):\n    \"\"\"Patch single layer\u002Fposition from clean to corrupted.\"\"\"\n    with model.trace(corrupted_prompt) as tracer:\n        # Get current activation\n        current = model.transformer.h[layer].output[0]\n\n        # Patch only specific position\n        current[:, position, :] = clean_cache[layer][:, position, :]\n\n        logits = model.output.save()\n\n    return logits\n\n# Sweep over all layers and positions\nresults = torch.zeros(12, seq_len)\nfor layer in range(12):\n    for pos in range(seq_len):\n        logits = patch_layer_position(layer, pos, clean_hidden, corrupted)\n        results[layer, pos] = compute_metric(logits)\n```\n\n## Workflow 3: Remote Execution with NDIF\n\nRun the same experiments on massive models without local GPUs.\n\n### Step-by-Step\n\n```python\nfrom nnsight import LanguageModel\n\n# 1. Load large model (will run remotely)\nmodel = LanguageModel(\"meta-llama\u002FLlama-3.1-70B\")\n\n# 2. Same code, just add remote=True\nwith model.trace(\"The meaning of life is\", remote=True) as tracer:\n    # Access internals of 70B model!\n    layer_40_out = model.model.layers[40].output[0].save()\n    logits = model.output.save()\n\n# 3. Results returned from NDIF\nprint(f\"Layer 40 shape: {layer_40_out.shape}\")\n\n# 4. Generation with interventions\nwith model.trace(remote=True) as tracer:\n    with tracer.invoke(\"What is 2+2?\"):\n        # Intervene during generation\n        model.model.layers[20].output[0][:, -1, :] *= 1.5\n\n    output = model.generate(max_new_tokens=50)\n```\n\n### NDIF Setup\n\n1. Sign up at [login.ndif.us](https:\u002F\u002Flogin.ndif.us)\n2. Get API key\n3. Set environment variable or pass to nnsight:\n\n```python\nimport os\nos.environ[\"NDIF_API_KEY\"] = \"your_key\"\n\n# Or configure directly\nfrom nnsight import CONFIG\nCONFIG.API_KEY = \"your_key\"\n```\n\n### Available Models on NDIF\n\n- Llama-3.1-8B, 70B, 405B\n- DeepSeek-R1 models\n- Various open-weight models (check [ndif.us](https:\u002F\u002Fndif.us) for current list)\n\n## Workflow 4: Cross-Prompt Activation Sharing\n\nShare activations between different inputs in a single trace.\n\n```python\nfrom nnsight import LanguageModel\n\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")\n\nwith model.trace() as tracer:\n    # First prompt\n    with tracer.invoke(\"The cat sat on the\"):\n        cat_hidden = model.transformer.h[6].output[0].save()\n\n    # Second prompt - inject cat's activations\n    with tracer.invoke(\"The dog ran through the\"):\n        # Replace with cat's activations at layer 6\n        model.transformer.h[6].output[0][:] = cat_hidden\n        dog_with_cat = model.output.save()\n\n# The dog prompt now has cat's internal representations\n```\n\n## Workflow 5: Gradient-Based Analysis\n\nAccess gradients during backward pass.\n\n```python\nfrom nnsight import LanguageModel\nimport torch\n\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")\n\nwith model.trace(\"The quick brown fox\") as tracer:\n    # Save activations and enable gradient\n    hidden = model.transformer.h[5].output[0].save()\n    hidden.retain_grad()\n\n    logits = model.output\n\n    # Compute loss on specific token\n    target_token = model.tokenizer.encode(\" jumps\")[0]\n    loss = -logits[0, -1, target_token]\n\n    # Backward pass\n    loss.backward()\n\n# Access gradients\ngrad = hidden.grad\nprint(f\"Gradient shape: {grad.shape}\")\nprint(f\"Gradient norm: {grad.norm().item():.3f}\")\n```\n\n**Note**: Gradient access not supported for vLLM or remote execution.\n\n## Common Issues & Solutions\n\n### Issue: Module path differs between models\n```python\n# GPT-2 structure\nmodel.transformer.h[5].output[0]\n\n# LLaMA structure\nmodel.model.layers[5].output[0]\n\n# Solution: Check model structure\nprint(model._model)  # See actual module names\n```\n\n### Issue: Forgetting to save\n```python\n# WRONG: Value not accessible outside trace\nwith model.trace(\"Hello\"):\n    hidden = model.transformer.h[5].output[0]  # Not saved!\n\nprint(hidden)  # Error or wrong value\n\n# RIGHT: Call .save()\nwith model.trace(\"Hello\"):\n    hidden = model.transformer.h[5].output[0].save()\n\nprint(hidden)  # Works!\n```\n\n### Issue: Remote timeout\n```python\n# For long operations, increase timeout\nwith model.trace(\"prompt\", remote=True, timeout=300) as tracer:\n    # Long operation...\n```\n\n### Issue: Memory with many saved activations\n```python\n# Only save what you need\nwith model.trace(\"prompt\"):\n    # Don't save everything\n    for i in range(100):\n        model.transformer.h[i].output[0].save()  # Memory heavy!\n\n    # Better: save specific layers\n    key_layers = [0, 5, 11]\n    for i in key_layers:\n        model.transformer.h[i].output[0].save()\n```\n\n### Issue: vLLM gradient limitation\n```python\n# vLLM doesn't support gradients\n# Use standard execution for gradient analysis\nmodel = LanguageModel(\"gpt2\", device_map=\"auto\")  # Not vLLM\n```\n\n## Key API Reference\n\n| Method\u002FProperty | Purpose |\n|-----------------|---------|\n| `model.trace(prompt, remote=False)` | Start tracing context |\n| `proxy.save()` | Save value for access after trace |\n| `proxy[:]` | Slice\u002Findex proxy (assignment patches) |\n| `tracer.invoke(prompt)` | Add prompt within trace |\n| `model.generate(...)` | Generate with interventions |\n| `model.output` | Final model output logits |\n| `model._model` | Underlying HuggingFace model |\n\n## Comparison with Other Tools\n\n| Feature | nnsight | TransformerLens | pyvene |\n|---------|---------|-----------------|--------|\n| Any architecture | Yes | Transformers only | Yes |\n| Remote execution | Yes (NDIF) | No | No |\n| Consistent API | No | Yes | Yes |\n| Deferred execution | Yes | No | No |\n| HuggingFace native | Yes | Reimplemented | Yes |\n| Shareable configs | No | No | Yes |\n\n## Reference Documentation\n\nFor detailed API documentation, tutorials, and advanced usage, see the `references\u002F` folder:\n\n| File | Contents |\n|------|----------|\n| [references\u002FREADME.md](references\u002FREADME.md) | Overview and quick start guide |\n| [references\u002Fapi.md](references\u002Fapi.md) | Complete API reference for LanguageModel, tracing, proxy objects |\n| [references\u002Ftutorials.md](references\u002Ftutorials.md) | Step-by-step tutorials for local and remote interpretability |\n\n## External Resources\n\n### Tutorials\n- [Getting Started](https:\u002F\u002Fnnsight.net\u002Fstart\u002F)\n- [Features Overview](https:\u002F\u002Fnnsight.net\u002Ffeatures\u002F)\n- [Remote Execution](https:\u002F\u002Fnnsight.net\u002Fnotebooks\u002Ffeatures\u002Fremote_execution\u002F)\n- [Applied Tutorials](https:\u002F\u002Fnnsight.net\u002Fapplied_tutorials\u002F)\n\n### Official Documentation\n- [Official Docs](https:\u002F\u002Fnnsight.net\u002Fdocumentation\u002F)\n- [NDIF Info](https:\u002F\u002Fndif.us\u002F)\n- [Community Forum](https:\u002F\u002Fdiscuss.ndif.us\u002F)\n\n### Papers\n- [NNsight and NDIF Paper](https:\u002F\u002Farxiv.org\u002Fabs\u002F2407.14561) - Fiotto-Kaufman et al. (ICLR 2025)\n\n## Architecture Support\n\nnnsight works with any PyTorch model:\n- **Transformers**: GPT-2, LLaMA, Mistral, etc.\n- **State Space Models**: Mamba\n- **Vision Models**: ViT, CLIP\n- **Custom architectures**: Any nn.Module\n\nThe key is knowing the module structure to access the right components.\n",{"name":5,"tags":44,"author":49,"license":15,"version":50,"description":7,"dependencies":51},[22,45,46,47,48],"NDIF","Remote Execution","Mechanistic Interpretability","Model Internals","Orchestra Research","1.0.0",[52,53],"nnsight>=0.5.0","torch>=2.0.0","main","34fa6acf794b9a6310d298cbb8fb8c9420a5edcf","cli-tool\u002Fcomponents\u002Fskills\u002Fai-research\u002Fmechanistic-interpretability-nnsight",2,"https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates","https:\u002F\u002Fgithub.com\u002Ftransaurus","Automated works of Page Turner","2026-03-28 10:12:59",{"source":8,"skillName":5,"commands":63,"prompt":73,"zipAvailable":74},[64,67,70],{"tool":65,"command":66},"npx","npx skills add transaurus\u002Fstaging-davila7-claude-code-templates --skill nnsight-remote-interpretability",{"tool":68,"command":69},"bunx","bunx skills add transaurus\u002Fstaging-davila7-claude-code-templates --skill nnsight-remote-interpretability",{"tool":71,"command":72},"pnpm","pnpm dlx skills add transaurus\u002Fstaging-davila7-claude-code-templates --skill nnsight-remote-interpretability","安装技能，技能地址：https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fai-research\u002Fmechanistic-interpretability-nnsight\u002FSKILL.md",true,[76,82,86],{"path":77,"resourceType":78,"resourceTypeDesc":79,"sizeBytes":80,"sourceHtmlUrl":81},"references\u002Fapi.md","REFERENCE","参考文档",7160,"https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fai-research\u002Fmechanistic-interpretability-nnsight\u002Freferences\u002Fapi.md",{"path":83,"resourceType":78,"resourceTypeDesc":79,"sizeBytes":84,"sourceHtmlUrl":85},"references\u002FREADME.md",1913,"https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fai-research\u002Fmechanistic-interpretability-nnsight\u002Freferences\u002FREADME.md",{"path":87,"resourceType":78,"resourceTypeDesc":79,"sizeBytes":88,"sourceHtmlUrl":89},"references\u002Ftutorials.md",7970,"https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fai-research\u002Fmechanistic-interpretability-nnsight\u002Freferences\u002Ftutorials.md",[91,121,142,164,184,210],{"skillNo":92,"name":93,"nameZh":6,"description":94,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":12,"repositoryNo":13,"starsCount":14,"licenseSpdxId":15,"licenseFileUrl":6,"sourceHtmlUrl":95,"categories":96,"keywords":102,"publishedAt":39},"01M6JMRJR3BE0HGWFWDZSB853S","zapier-make-patterns","No-code automation democratizes workflow building. Zapier and Make (formerly Integromat) let non-developers automate business processes without writing code. But no-code doesn't mean no-complexity - these platforms have their own patterns, pitfalls, and breaking points.  This skill covers when to use which platform, how to build reliable automations, and when to graduate to code-based solutions. Key insight: Zapier optimizes for simplicity and integrations (7000+ apps), Make optimizes for power","https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fworkflow-automation\u002Fzapier-make-patterns\u002FSKILL.md",[97],{"categoryNo":98,"categoryName":99,"taxonomyType":100,"taxonomyTypeDesc":101},"01JBSEEDCAT000000000000023","Skill与智能体开发","SCENARIO","使用场景",[103,104,105,106,107,108,109,110,111,112,31,113,114,115,116,117,118,119,24,120],"zapier make patterns","patterns","make","zapier","writing","workflow","automation","code","integrations","business","power","complexity","platform","apps","how to","No-Code","which","spawner",{"skillNo":122,"name":123,"nameZh":6,"description":124,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":12,"repositoryNo":13,"starsCount":14,"licenseSpdxId":15,"licenseFileUrl":6,"sourceHtmlUrl":125,"categories":126,"keywords":127,"publishedAt":39},"01M6JMR8YK6SWPCTG1MCQTMN9S","yeet","Use only when the user explicitly asks to stage, commit, push, and open a GitHub pull request in one flow using the GitHub CLI (`gh`).","https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fworkflow-automation\u002Fyeet\u002FSKILL.md",[],[123,128,129,130,131,132,133,134,135,136,137,138,31,139,140,141],"cli","github","commit","pull request","openai","pull","push","flow","open","gh","user","only","stage","github cli",{"skillNo":143,"name":144,"nameZh":6,"description":145,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":12,"repositoryNo":13,"starsCount":14,"licenseSpdxId":15,"licenseFileUrl":6,"sourceHtmlUrl":146,"categories":147,"keywords":149,"publishedAt":39},"01M6JMQNFE64ZASQK90HARRXW8","workflow-automation","Workflow automation is the infrastructure that makes AI agents reliable. Without durable execution, a network hiccup during a 10-step payment flow means lost money and angry customers. With it, workflows resume exactly where they left off.  This skill covers the platforms (n8n, Temporal, Inngest) and patterns (sequential, parallel, orchestrator-worker) that turn brittle scripts into production-grade automation.  Key insight: The platforms make different tradeoffs. n8n optimizes for accessibility","https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fworkflow-automation\u002Fworkflow-automation\u002FSKILL.md",[148],{"categoryNo":98,"categoryName":99,"taxonomyType":100,"taxonomyTypeDesc":101},[108,109,150,151,152,153,154,33,155,156,157,158,159,160,161,162,163,26,24,120],"Workflow Automation","ai agents","accessibility","workflows","apache","inngest","parallel","orchestrator","resume","n8n","payment","worker","scripts","temporal",{"skillNo":165,"name":166,"nameZh":6,"description":167,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":12,"repositoryNo":13,"starsCount":14,"licenseSpdxId":15,"licenseFileUrl":6,"sourceHtmlUrl":168,"categories":169,"keywords":170,"publishedAt":39},"01M6JMQDYJ0XETERPHGDR3EVGN","trigger-dev","Trigger.dev expert for background jobs, AI workflows, and reliable async execution with excellent developer experience and TypeScript-first design. Use when: trigger.dev, trigger dev, background task, ai background job, long running task.","https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fworkflow-automation\u002Ftrigger-dev\u002FSKILL.md",[],[171,172,173,174,175,153,176,154,33,177,31,178,179,180,181,182,183,26,24,120],"dev","trigger dev","trigger","design","typescript","async execution","first","async","developer experience","expert","task","background jobs","long",{"skillNo":185,"name":186,"nameZh":6,"description":187,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":12,"repositoryNo":13,"starsCount":14,"licenseSpdxId":15,"licenseFileUrl":6,"sourceHtmlUrl":188,"categories":189,"keywords":193,"publishedAt":39},"01M6JMQ5J6KK3GSX9FJCWWK6G6","planning-with-files","Transforms workflow to use Manus-style persistent markdown files for planning, progress tracking, and knowledge storage. Use when starting complex tasks, multi-step projects, research tasks, or when the user mentions planning, organizing work, tracking progress, or wants structured output.","https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fworkflow-automation\u002Fplanning-with-files\u002FSKILL.md",[190],{"categoryNo":191,"categoryName":192,"taxonomyType":100,"taxonomyTypeDesc":101},"01JBSEEDCAT000000000000021","资料检索与研究",[194,195,196,26,197,108,198,199,200,201,202,203,31,204,205,206,207,208,209,24],"planning with files","planning","files","with files","storage","tasks","markdown","knowledge","projects","work","manus","structured output","progress","transforms","multi","output",{"skillNo":211,"name":212,"nameZh":6,"description":213,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":12,"repositoryNo":13,"starsCount":14,"licenseSpdxId":15,"licenseFileUrl":6,"sourceHtmlUrl":214,"categories":215,"keywords":220,"publishedAt":39},"01M6JMPWAT60RXXNEC4V9TRBRS","n8n-workflow-patterns","Proven workflow architectural patterns from real n8n workflows. Use when building new workflows, designing workflow structure, choosing workflow patterns, planning workflow architecture, or asking about webhook processing, HTTP API integration, database operations, AI agent workflows, or scheduled tasks.","https:\u002F\u002Fgithub.com\u002Ftransaurus\u002Fstaging-davila7-claude-code-templates\u002Fblob\u002Fmain\u002Fcli-tool\u002Fcomponents\u002Fskills\u002Fworkflow-automation\u002Fn8n\u002Fn8n-workflow-patterns\u002FSKILL.md",[216,219],{"categoryNo":217,"categoryName":218,"taxonomyType":100,"taxonomyTypeDesc":101},"01JBSEEDCAT000000000000010","后端与接口",{"categoryNo":98,"categoryName":99,"taxonomyType":100,"taxonomyTypeDesc":101},[108,221,159,222,104,223,224,153,38,199,195,225,226,227,228,229,230,231,232,233],"n8n workflow patterns","workflow patterns","n8n workflow","ai agent","database","api","planning workflow","structure","database operations","webhook","processing","http api","scheduled"]