[
  {
    "metadataOutputVersion" : "3.0",
    "shortDescription" : "v9.2.0.lealii",
    "outputSchema" : [
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2)",
        "shortDescription" : "",
        "shape" : "[2]",
        "name" : "actionId",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32)",
        "shortDescription" : "",
        "shape" : "[]",
        "name" : "actionCandidates",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2)",
        "shortDescription" : "",
        "shape" : "[2]",
        "name" : "shadowActionId",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32)",
        "shortDescription" : "",
        "shape" : "[]",
        "name" : "shadowActionCandidates",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 2 × 50 × 15)",
        "shortDescription" : "",
        "shape" : "[2, 50, 15]",
        "name" : "rankings",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 3000)",
        "shortDescription" : "",
        "shape" : "[3000]",
        "name" : "diagnostic",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 10 × 3)",
        "shortDescription" : "",
        "shape" : "[10, 3]",
        "name" : "anonymizedHistory",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32)",
        "shortDescription" : "",
        "shape" : "[]",
        "name" : "forcedPrompt",
        "type" : "MultiArray"
      }
    ],
    "version" : "v9.2.0",
    "modelParameters" : [

    ],
    "specificationVersion" : 6,
    "storagePrecision" : "Float32",
    "mlProgramOperationTypeHistogram" : {
      "LessEqual" : 4,
      "Greater" : 87,
      "Cast" : 188,
      "ReduceL2Norm" : 4,
      "Argsort" : 15,
      "SliceByIndex" : 95,
      "Log" : 5,
      "Concat" : 23,
      "ScatterNd" : 6,
      "ReduceMax" : 18,
      "Inverse" : 7,
      "GatherNd" : 2,
      "Pad" : 1,
      "ReduceMin" : 6,
      "Cumsum" : 4,
      "Shape" : 6,
      "Linear" : 10,
      "Gather" : 3,
      "NotEqual" : 6,
      "Squeeze" : 11,
      "Equal" : 21,
      "RealDiv" : 18,
      "LogicalAnd" : 5,
      "GreaterEqual" : 10,
      "Less" : 37,
      "NonZero" : 8,
      "Exp" : 17,
      "Mul" : 364,
      "Abs" : 23,
      "Scatter" : 12,
      "ExpandDims" : 89,
      "ReduceSum" : 53,
      "Minimum" : 1,
      "Pow" : 13,
      "GatherAlongAxis" : 12,
      "Sub" : 97,
      "FloorDiv" : 12,
      "Reshape" : 28,
      "Transpose" : 33,
      "Clip" : 2,
      "Add" : 117,
      "ReduceArgmax" : 5,
      "Cos" : 4,
      "Tile" : 6
    },
    "computePrecision" : "Mixed (Float32, Int32)",
    "stateSchema" : [

    ],
    "isUpdatable" : "0",
    "availability" : {
      "macOS" : "12.0",
      "tvOS" : "15.0",
      "visionOS" : "1.0",
      "watchOS" : "8.0",
      "iOS" : "15.0",
      "macCatalyst" : "15.0"
    },
    "modelType" : {
      "name" : "MLModelType_mlProgram"
    },
    "inputSchema" : [
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 9 × 50 × 15)",
        "shortDescription" : "",
        "shape" : "[9, 50, 15]",
        "name" : "tuples",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 4)",
        "shortDescription" : "",
        "shape" : "[4]",
        "name" : "deviceContext",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1000 × 8)",
        "shortDescription" : "",
        "shape" : "[1000, 8]",
        "name" : "candidateInteractions",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1000 × 2)",
        "shortDescription" : "",
        "shape" : "[1000, 2]",
        "name" : "tupleInteractions_candidates",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1000 × 7)",
        "shortDescription" : "",
        "shape" : "[1000, 7]",
        "name" : "tupleInteractions_alignment",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 10 × 3)",
        "shortDescription" : "",
        "shape" : "[10, 3]",
        "name" : "similarityScores",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 50 × 2)",
        "shortDescription" : "",
        "shape" : "[50, 2]",
        "name" : "riskLevel",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 50 × 15)",
        "shortDescription" : "",
        "shape" : "[50, 15]",
        "name" : "candidate_risk",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 4)",
        "shortDescription" : "",
        "shape" : "[4]",
        "name" : "forcedPromptRate",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 15)",
        "shortDescription" : "",
        "shape" : "[15]",
        "name" : "parameterName",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 15)",
        "shortDescription" : "",
        "shape" : "[15]",
        "name" : "alreadyPrompted",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 15)",
        "shortDescription" : "",
        "shape" : "[15]",
        "name" : "isResolved",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1)",
        "shortDescription" : "",
        "shape" : "[1]",
        "name" : "component",
        "type" : "MultiArray"
      }
    ],
    "userDefinedMetadata" : {
      "com.github.apple.coremltools.source_dialect" : "TorchScript",
      "com.github.apple.coremltools.version" : "8.2.2",
      "com.github.apple.coremltools.source" : "torch==2.6.0",
      "mldb_token" : "mldb-wrxg5d7ao8",
      "tag" : "mldb-wrxg5d7ao8"
    },
    "generatedClassName" : "PlanResolutionModel",
    "method" : "predict"
  }
]