{
  "model": "TabICLv2",
  "source": {
    "package": "tabicl 2.2.0",
    "repo": "jingang/TabICL",
    "revision": "4dcd344ece2c00be9e831fdd35bed57b5ad83e19",
    "checkpoints": {
      "classifier": "tabicl-classifier-v2-20260212.ckpt",
      "regressor": "tabicl-regressor-v2-20260212.ckpt"
    },
    "license": "BSD-3-Clause"
  },
  "opset": 18,
  "exporter": "torch.onnx (dynamo) 2.14.0+cpu",
  "limits": {
    "maxClasses": 10,
    "numQuantiles": 999,
    "featureGroupSize": 3,
    "cellDim": 128,
    "rowDim": 512
  },
  "files": [
    {
      "task": "classifier",
      "stage": "col",
      "variant": "f16w",
      "file": "classifier-col.f16w.onnx",
      "bytes": 1894277,
      "sha256": "a94e3998566acf9513e27ff38d5e8f7b9873bd8751d86192fd308ec426e7dd1c",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "X",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            3
          ]
        },
        {
          "name": "y",
          "type": "float32",
          "shape": [
            "batch",
            "digits",
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "E",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            128
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "icl",
      "variant": "f16w",
      "file": "classifier-icl.f16w.onnx",
      "bytes": 52917443,
      "sha256": "21791e2a7303c4d48611201f42f59f2ad69cf6ec5e6f528785234b889934e1a8",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "R",
          "type": "float32",
          "shape": [
            "batch",
            "rows",
            512
          ]
        },
        {
          "name": "y",
          "type": "float32",
          "shape": [
            "batch",
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "out",
          "type": "float32",
          "shape": [
            "batch",
            "test",
            10
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "row",
      "variant": "f16w",
      "file": "classifier-row.f16w.onnx",
      "bytes": 855980,
      "sha256": "177bffd02a588f58e25d0ed0269221214863109fc89a21966f1b9c9673bf8601",
      "cpuNodes": [
        "node_Concat_7"
      ],
      "inputs": [
        {
          "name": "E",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            128
          ]
        }
      ],
      "outputs": [
        {
          "name": "R",
          "type": "float32",
          "shape": [
            "batch",
            "rows",
            512
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "col",
      "variant": "fp32",
      "file": "classifier-col.fp32.onnx",
      "bytes": 3626570,
      "sha256": "bb7221f89b5d73f4adc1b337bf0eaba7ab362c2494c33ce1f0f3549a2fa2b020",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "X",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            3
          ]
        },
        {
          "name": "y",
          "type": "float32",
          "shape": [
            "batch",
            "digits",
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "E",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            128
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "icl",
      "variant": "fp32",
      "file": "classifier-icl.fp32.onnx",
      "bytes": 105366407,
      "sha256": "7af95b18d1b45c15ebe26b89f0e066130ef6ee9adc4a3263bc8250f52b372260",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "R",
          "type": "float32",
          "shape": [
            "batch",
            "rows",
            512
          ]
        },
        {
          "name": "y",
          "type": "float32",
          "shape": [
            "batch",
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "out",
          "type": "float32",
          "shape": [
            "batch",
            "test",
            10
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "row",
      "variant": "fp32",
      "file": "classifier-row.fp32.onnx",
      "bytes": 1641650,
      "sha256": "316b5f0ce0aa2559731693aae9f126d941e4e1b01b9d9323e35df095b27be397",
      "cpuNodes": [
        "node_Concat_7"
      ],
      "inputs": [
        {
          "name": "E",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            128
          ]
        }
      ],
      "outputs": [
        {
          "name": "R",
          "type": "float32",
          "shape": [
            "batch",
            "rows",
            512
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "col",
      "variant": "f16w",
      "file": "regressor-col.f16w.onnx",
      "bytes": 1883270,
      "sha256": "d434c1c37b5e0f53e8052869ba93682f42da4f3eb0f1154bda95b40ab6807436",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "X",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            3
          ]
        },
        {
          "name": "y",
          "type": "float32",
          "shape": [
            "batch",
            "digits",
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "E",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            128
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "icl",
      "variant": "f16w",
      "file": "regressor-icl.f16w.onnx",
      "bytes": 54884074,
      "sha256": "e2f25c737dee1524687c758487e85c38768984b5ec027ffb318da08fb54c454d",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "R",
          "type": "float32",
          "shape": [
            "batch",
            "rows",
            512
          ]
        },
        {
          "name": "y",
          "type": "float32",
          "shape": [
            "batch",
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "out",
          "type": "float32",
          "shape": [
            "batch",
            "test",
            999
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "row",
      "variant": "f16w",
      "file": "regressor-row.f16w.onnx",
      "bytes": 851555,
      "sha256": "280e72146067baaa2516ffa8483be2c368073fa4171f765a08ac2e8e4c801d8b",
      "cpuNodes": [
        "node_Concat_7"
      ],
      "inputs": [
        {
          "name": "E",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            128
          ]
        }
      ],
      "outputs": [
        {
          "name": "R",
          "type": "float32",
          "shape": [
            "batch",
            "rows",
            512
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "col",
      "variant": "fp32",
      "file": "regressor-col.fp32.onnx",
      "bytes": 3613048,
      "sha256": "2ea2113646278399b1e58beb57d3e46d4496ee17e9c3871bafd4985dc92f4300",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "X",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            3
          ]
        },
        {
          "name": "y",
          "type": "float32",
          "shape": [
            "batch",
            "digits",
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "E",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            128
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "icl",
      "variant": "fp32",
      "file": "regressor-icl.fp32.onnx",
      "bytes": 109348315,
      "sha256": "17bb4ca71d88aea44b3f8886e376c41d382ded9ef71aafc594ae3b736a3b8916",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "R",
          "type": "float32",
          "shape": [
            "batch",
            "rows",
            512
          ]
        },
        {
          "name": "y",
          "type": "float32",
          "shape": [
            "batch",
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "out",
          "type": "float32",
          "shape": [
            "batch",
            "test",
            999
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "row",
      "variant": "fp32",
      "file": "regressor-row.fp32.onnx",
      "bytes": 1637225,
      "sha256": "e03db9cdec62860e538afe61c6e1a6f7533a72756066da3660c508efb1f06559",
      "cpuNodes": [
        "node_Concat_7"
      ],
      "inputs": [
        {
          "name": "E",
          "type": "float32",
          "shape": [
            "batch",
            "columns",
            "rows",
            128
          ]
        }
      ],
      "outputs": [
        {
          "name": "R",
          "type": "float32",
          "shape": [
            "batch",
            "rows",
            512
          ]
        }
      ]
    }
  ]
}