{
  "head": "token_classifier",
  "output_dim": 17,
  "per_token": true,
  "shapes": [
    {
      "batch": 32,
      "seq": 128,
      "function": "b32_s128"
    }
  ]
}