{
  "architecture_name": "mitra",
  "config": {
    "dim": 512,
    "dim_output": 10,
    "n_layers": 12,
    "n_heads": 4,
    "task": "CLASSIFICATION"
  }
}