[{"data":1,"prerenderedAt":-1},["ShallowReactive",2],{"skill:01M6KQAG7VPDAD13Q4JC2V8F6V":3,"skill-same:01M6KQAG7VPDAD13Q4JC2V8F6V":88},{"skillNo":4,"name":5,"nameZh":6,"description":7,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":6,"repositoryNo":12,"starsCount":13,"licenseSpdxId":14,"licenseFileUrl":6,"sourceHtmlUrl":15,"categories":16,"keywords":17,"publishedAt":39,"linkOnly":40,"markdownContent":41,"markdownBody":42,"frontMatter":43,"refName":54,"commitSha":55,"skillDirectory":56,"forksCount":13,"repositoryHtmlUrl":57,"repositoryHomepageUrl":6,"ownerHtmlUrl":58,"ownerBlogUrl":6,"ownerDescription":6,"updatedAt":59,"install":60,"resources":73},"01M6KQAG7VPDAD13Q4JC2V8F6V","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.","endman100\u002Fskill-router-nexus","endman100","https:\u002F\u002Favatars.githubusercontent.com\u002Fu\u002F31997019?v=4","01M6KPDX17BYDWW4RRB0TNGGMX","01M6KPDX64Y2QBEWVFG5W5PWPF",0,"MIT","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FAI-Research\u002Fnnsight\u002FSKILL.md",[],[18,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","when","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":14,"version":50,"description":7,"dependencies":51},[21,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","f19b254e8975d0bfddb394686706106930d7964d","AI-Research\u002Fnnsight","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus","https:\u002F\u002Fgithub.com\u002Fendman100","2026-11-11 18:19:10",{"source":8,"skillName":5,"commands":61,"prompt":71,"zipAvailable":72},[62,65,68],{"tool":63,"command":64},"npx","npx skills add endman100\u002Fskill-router-nexus --skill nnsight-remote-interpretability",{"tool":66,"command":67},"bunx","bunx skills add endman100\u002Fskill-router-nexus --skill nnsight-remote-interpretability",{"tool":69,"command":70},"pnpm","pnpm dlx skills add endman100\u002Fskill-router-nexus --skill nnsight-remote-interpretability","安装技能，技能地址：https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FAI-Research\u002Fnnsight\u002FSKILL.md",true,[74,80,84],{"path":75,"resourceType":76,"resourceTypeDesc":77,"sizeBytes":78,"sourceHtmlUrl":79},"references\u002Fapi.md","REFERENCE","参考文档",7160,"https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FAI-Research\u002Fnnsight\u002Freferences\u002Fapi.md",{"path":81,"resourceType":76,"resourceTypeDesc":77,"sizeBytes":82,"sourceHtmlUrl":83},"references\u002FREADME.md",1913,"https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FAI-Research\u002Fnnsight\u002Freferences\u002FREADME.md",{"path":85,"resourceType":76,"resourceTypeDesc":77,"sizeBytes":86,"sourceHtmlUrl":87},"references\u002Ftutorials.md",7970,"https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FAI-Research\u002Fnnsight\u002Freferences\u002Ftutorials.md",[89,113,149,171,180,204],{"skillNo":90,"name":91,"nameZh":6,"description":92,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":6,"repositoryNo":12,"starsCount":13,"licenseSpdxId":14,"licenseFileUrl":93,"sourceHtmlUrl":94,"categories":95,"keywords":101,"publishedAt":39},"01M6MPCT5BXVER8HQZTW3QE9AS","create-master","基于佛教经典文献，生成特定高僧大德的 AI 教学角色","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FMetaphysics-and-Spirituality\u002FMaster-skill\u002FLICENSE","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FMetaphysics-and-Spirituality\u002FMaster-skill\u002FSKILL.md",[96],{"categoryNo":97,"categoryName":98,"taxonomyType":99,"taxonomyTypeDesc":100},"01JBSEEDCAT000000000000039","教师与教育","PROFESSION","适用职业",[102,103,104,105,106,107,108,109,110,111,112],"create master","master","create","生成","true","webfetch","edit","write","read","grep","Bash",{"skillNo":114,"name":115,"nameZh":6,"description":116,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":6,"repositoryNo":12,"starsCount":13,"licenseSpdxId":14,"licenseFileUrl":117,"sourceHtmlUrl":118,"categories":119,"keywords":131,"publishedAt":39},"01M6M43AKE95Y64E1JRXEF5QQ3","writing-dna-skill","从至少 20 篇完整文章中蒸馏可复用的写作 DNA，分析语言、文章结构、选题逻辑、素材策略、认知框架和视觉风格，并生成 Writing-DNA.md；按该 DNA 写作时会读取全部蒸馏产物和 5 篇相关原文。用于中英文作者、账号、品牌或出版物的风格分析与一致性写作。 Distill reusable Writing DNA from at least 20 complete articles for Chinese or English authors, publications, brands, and accounts; use for language, structure, topic logic, source strategy, cognitive-frame, and visual-style analysis.","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FWriting-and-Content\u002Fwriting-dna-skill\u002FLICENSE","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FWriting-and-Content\u002Fwriting-dna-skill\u002FSKILL.md",[120,125,128],{"categoryNo":121,"categoryName":122,"taxonomyType":123,"taxonomyTypeDesc":124},"01JBSEEDCAT000000000000018","写作与文案","SCENARIO","使用场景",{"categoryNo":126,"categoryName":127,"taxonomyType":123,"taxonomyTypeDesc":124},"01JBSEEDCAT000000000000023","Skill与智能体开发",{"categoryNo":129,"categoryName":130,"taxonomyType":99,"taxonomyTypeDesc":100},"01JBSEEDCAT000000000000036","内容创作者",[132,133,134,135,136,137,138,139,140,141,31,142,143,105,144,145,146,24,147,148],"writing dna skill","DNA","writing","strategy","english","logic","visual","structure","chinese","reusable","style","distill","analysis","frame","source","at","complete",{"skillNo":150,"name":151,"nameZh":6,"description":152,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":6,"repositoryNo":12,"starsCount":13,"licenseSpdxId":14,"licenseFileUrl":153,"sourceHtmlUrl":154,"categories":155,"keywords":158,"publishedAt":39},"01M6M428TTCGQ4ME3KY99VN037","stop-slop","Remove AI writing patterns from prose. Use when drafting, editing, or reviewing text to eliminate predictable AI tells.","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FWriting-and-Content\u002Fstop-slop\u002FLICENSE","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FWriting-and-Content\u002Fstop-slop\u002FSKILL.md",[156,157],{"categoryNo":121,"categoryName":122,"taxonomyType":123,"taxonomyTypeDesc":124},{"categoryNo":129,"categoryName":130,"taxonomyType":99,"taxonomyTypeDesc":100},[159,160,161,162,23,163,164,165,166,31,167,168,169,24,170,134],"slop","stop slop","stop","drafting","drafts","https","patterns","prose","remove","editing","text","content",{"skillNo":172,"name":173,"nameZh":6,"description":174,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":6,"repositoryNo":12,"starsCount":13,"licenseSpdxId":14,"licenseFileUrl":175,"sourceHtmlUrl":176,"categories":177,"keywords":178,"publishedAt":39},"01M6M41X0B7EBX42TK435DMGGR","shuorenhua","检查和清理中英文文本里的 AI 套路，适用于“去 AI 味”“说人话”“自然一点”“别像模板”“先标问题”这类改写和审稿需求；按场景控制力度，同时保留事实、术语、语域和责任主体。","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FWriting-and-Content\u002Fshuorenhua\u002FLICENSE","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FWriting-and-Content\u002Fshuorenhua\u002FSKILL.md",[],[173,179],"文本",{"skillNo":181,"name":182,"nameZh":6,"description":183,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":6,"repositoryNo":12,"starsCount":13,"licenseSpdxId":14,"licenseFileUrl":184,"sourceHtmlUrl":185,"categories":186,"keywords":187,"publishedAt":39},"01M6M3ZD12WBEW5KXKDM4N6DFG","naming","Name products, SaaS, brands, open source projects, bots, and apps. Use when the user needs to name something, find a brand name, or pick a product name. Metaphor-driven process that produces memorable, meaningful names and avoids AI slop.","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FWriting-and-Content\u002Fnaming-main\u002FLICENSE","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FWriting-and-Content\u002Fnaming-main\u002FSKILL.md",[],[182,188,189,190,191,192,31,193,110,146,194,195,196,197,112,198,199,200,201,23,202,203,159],"repo","websearch","product name","apps","bots","view","brand","pick","gh","projects","open","saas","open source","product","describe","ai slop",{"skillNo":205,"name":206,"nameZh":6,"description":207,"descriptionZh":6,"repositoryFullName":8,"ownerLogin":9,"ownerAvatarUrl":10,"ownerAccountNo":11,"ownerDisplayName":6,"repositoryNo":12,"starsCount":13,"licenseSpdxId":14,"licenseFileUrl":208,"sourceHtmlUrl":209,"categories":210,"keywords":212,"publishedAt":39},"01M6M3YVXAFAAWJ7F6JMPFREYE","humanizer-zh","去除文本中的 AI 生成痕迹。适用于编辑或审阅文本，使其听起来更自然、更像人类书写。\n基于维基百科的\"AI 写作特征\"综合指南。检测并修复以下模式：夸大的象征意义、\n宣传性语言、以 -ing 结尾的肤浅分析、模糊的归因、破折号过度使用、三段式法则、\nAI 词汇、否定式排比、过多的连接性短语。","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FWriting-and-Content\u002Fhumanizer-zh\u002FLICENSE","https:\u002F\u002Fgithub.com\u002Fendman100\u002Fskill-router-nexus\u002Fblob\u002Fmain\u002FWriting-and-Content\u002Fhumanizer-zh\u002FSKILL.md",[211],{"categoryNo":121,"categoryName":122,"taxonomyType":123,"taxonomyTypeDesc":124},[213,214,215,110,109,108,134,105,179,216,159,161],"humanizer","zh","humanizer zh","翻译"]