{
  "schemaVersion": 1,
  "lessonId": "transforms-03",
  "contentHash": "c306f3ef08ecee0f35915e560ca61c38fd59adc47cce3ac28ddb6b1336b694da",
  "rendererHash": "a1bd5b648f4731ae5d5f0a24811c2540d1a0a132149ce589d17c4113509cbedf",
  "kind": "executed",
  "generatedAt": "2026-10-20T10:45:12.424996+00:00",
  "provenance": "recorded_cpu_computation",
  "environment": {
    "backend": "cpu",
    "deviceCount": 1,
    "jax": "0.9.2",
    "numpy": "2.4.4",
    "matplotlib": "3.10.8"
  },
  "stdout": "[1. 1. 8.]\n",
  "data": {
    "kind": "bar",
    "labels": [
      "row 0",
      "row 1",
      "row 2"
    ],
    "ylabel": "prediction",
    "series": [
      {
        "label": "vmap output",
        "y": [
          1.0,
          1.0,
          8.0
        ]
      }
    ]
  },
  "diagram": {
    "title": "Mapped rows and shared weights",
    "prediction": "What should happen if you duplicate an input row while keeping the same shared weights?",
    "reading": "Each row enters the same single-example rule with one shared weight vector. The scalar results are stacked in row order. This is a mathematical mapping, not a hardware-thread allocation. Reordering observations should reorder outputs identically.",
    "alt": "Each row enters the same single-example rule with one shared weight vector. The scalar results are stacked in row order. This is a mathematical mapping, not a hardware-thread allocation. Reordering observations should reorder outputs identically.",
    "scope": "Architecture and dataflow mechanism diagram.",
    "height": 4,
    "panel": {
      "kind": "table",
      "headers": [
        "Mapped input",
        "Shared argument",
        "Single-example result"
      ],
      "rows": [
        [
          "row 0",
          "same weights",
          "1"
        ],
        [
          "row 1",
          "same weights",
          "1"
        ],
        [
          "row 2",
          "same weights",
          "8"
        ],
        [
          "in_axes = 0",
          "in_axes = None",
          "stack \u2192 [1, 1, 8]"
        ]
      ]
    }
  }
}