{
  "format": "chess_autocomplete_model_repo_v1",
  "model": {
    "architecture": "ChessTransformer",
    "dim": 768,
    "num_heads": 12,
    "num_blocks": 12,
    "max_half_moves": 600,
    "input_token_representation": "Discrete",
    "norm_type": "layernorm",
    "mlp_type": "swiglu",
    "loops": 1,
    "add_metadata_tokens": true,
    "metadata_dropout_p": 0.3,
    "input_size": 4171,
    "output_size": 4135
  },
  "tokenizers": {
    "input_tokenizer": "RealizableMoveTokenizer",
    "output_tokenizer": "RealizableMoveTokenizer"
  },
  "artifacts": {
    "pytorch": {
      "path": "model.safetensors",
      "format": "safetensors"
    },
    "onnx": {
      "path": "model.onnx",
      "interface": "bin_logits_v2",
      "input_name": "bin_moves",
      "output_name": "bin_logits",
      "precision": "fp32"
    },
    "onnx_bf16": {
      "path": "model-bf16-fp32compute.onnx",
      "interface": "bin_logits_v2",
      "input_name": "bin_moves",
      "output_name": "bin_logits",
      "precision": "bf16"
    },
    "onnx_int8": {
      "path": "model-int8-blk128.onnx",
      "interface": "bin_logits_v2",
      "input_name": "bin_moves",
      "output_name": "bin_logits",
      "precision": "int8"
    },
    "onnx_int8_last": {
      "path": "model-int8-blk128-last.onnx",
      "interface": "bin_logits_v1",
      "input_name": "bin_moves",
      "output_name": "bin_logits",
      "precision": "int8"
    }
  },
  "source": {
    "run_id": "chess-autocomplete-v1-91m",
    "checkpoint": "checkpoint-000599999.pt",
    "checkpoint_iter": 599999
  }
}