{
  "schemaVersion": 1,
  "lessonId": "transformers-02",
  "contentHash": "d62730a9b6f73ecf78a5b1ea0d799061469f8f60316f7715e6cd70ed70a0f606",
  "rendererHash": "a1bd5b648f4731ae5d5f0a24811c2540d1a0a132149ce589d17c4113509cbedf",
  "kind": "executed",
  "generatedAt": "2026-10-20T10:45:52.168439+00:00",
  "provenance": "recorded_cpu_computation",
  "environment": {
    "backend": "cpu",
    "deviceCount": 1,
    "jax": "0.9.2",
    "numpy": "2.4.4",
    "matplotlib": "3.10.8"
  },
  "stdout": "Causal means: [2.       3.       4.666667 7.5     ]\n",
  "data": {
    "kind": "heatmap",
    "values": [
      [
        1.0,
        0.0,
        0.0,
        0.0
      ],
      [
        0.5,
        0.5,
        0.0,
        0.0
      ],
      [
        0.3333333432674408,
        0.3333333432674408,
        0.3333333432674408,
        0.0
      ],
      [
        0.25,
        0.25,
        0.25,
        0.25
      ]
    ],
    "rows": [
      "query 0",
      "query 1",
      "query 2",
      "query 3"
    ],
    "columns": [
      "key 0",
      "key 1",
      "key 2",
      "key 3"
    ],
    "unit": "attention weight"
  }
}
