{
  "head": "entity_pool",
  "output_dim": 4,
  "per_token": false,
  "shapes": [
    {
      "batch": 16,
      "seq": 64,
      "function": "b16_s64"
    },
    {
      "batch": 16,
      "seq": 128,
      "function": "b16_s128"
    },
    {
      "batch": 16,
      "seq": 256,
      "function": "b16_s256"
    }
  ]
}