{
  "model": "KumoRelational",
  "source": {
    "package": "structured-data-models @ 2b1696f4fdc2",
    "repo": "nvidia/Kumo-Relational",
    "revision": "v1.0.1",
    "checkpoints": {
      "classifier": "classifier.pt",
      "regressor": "regressor.pt"
    },
    "license": "OpenMDW-1.1 (weights; fine-tuned from TabICLv2, BSD-3-Clause)",
    "notices": [
      "LICENSE",
      "THIRD-PARTY-NOTICES.txt"
    ]
  },
  "opset": 18,
  "exporter": "torch.onnx (dynamo) 2.14.0+cpu",
  "limits": {
    "maxClasses": 10,
    "numQuantiles": 999,
    "featureGroupSize": 3,
    "cellDim": 128,
    "rowDim": 512,
    "maxKeys": 20000,
    "temperature": 0.9
  },
  "tasks": {
    "classifier": {
      "numClasses": 10,
      "outDim": 10,
      "emptyRowEmbedding": [
        -0.6515481472015381,
        -0.6578977108001709,
        -0.6777721643447876,
        -0.7822152376174927,
        1.7381236553192139,
        4.868432521820068,
        0.021998055279254913,
        0.2011304348707199,
        -1.1834341287612915,
        -0.6197394728660583,
        -0.26040998101234436,
        1.8957850933074951,
        0.1128038763999939,
        -0.5565066337585449,
        0.2857104241847992,
        -0.1305471807718277,
        -0.02089848183095455,
        0.6901000142097473,
        -0.633758544921875,
        -0.5899574160575867,
        0.17074818909168243,
        0.9734656810760498,
        -0.3416438698768616,
        0.3050732910633087,
        1.0541795492172241,
        0.3212589621543884,
        0.22909268736839294,
        0.03650117665529251,
        0.036828238517045975,
        -0.7325596213340759,
        2.1401281356811523,
        2.0198001861572266,
        1.0548911094665527,
        1.1351673603057861,
        -0.6504899263381958,
        1.2082951068878174,
        0.3168076276779175,
        -0.908993661403656,
        -0.028187081217765808,
        -0.08583246916532516,
        0.23075464367866516,
        -0.1810135841369629,
        0.934986412525177,
        0.2889697551727295,
        -0.9346817135810852,
        -0.9134857654571533,
        -0.2210700660943985,
        -0.192105233669281,
        -0.7671796679496765,
        0.07601941376924515,
        0.008838735520839691,
        -0.20181573927402496,
        -0.330454021692276,
        0.19383491575717926,
        -1.8057305812835693,
        -0.28069818019866943,
        -1.304654836654663,
        0.5029457211494446,
        -1.1884835958480835,
        0.7161130309104919,
        -0.3595302104949951,
        0.2729124426841736,
        -0.7834827899932861,
        -0.6621738076210022,
        0.06356482207775116,
        -0.19944192469120026,
        -1.109471321105957,
        -0.19683974981307983,
        -0.46237966418266296,
        0.7138513922691345,
        -0.1656678318977356,
        1.0790364742279053,
        -0.19757182896137238,
        0.009998749010264874,
        0.3477487862110138,
        0.6488944888114929,
        0.6338379383087158,
        -0.4470149576663971,
        0.13880422711372375,
        -0.7759476900100708,
        -1.757300615310669,
        -0.1753847748041153,
        -0.42946237325668335,
        -0.2217670977115631,
        -0.8044957518577576,
        1.1557780504226685,
        -0.36158156394958496,
        1.0754748582839966,
        0.6351344585418701,
        -0.18819473683834076,
        0.1983778178691864,
        -0.2253461331129074,
        -0.5574930310249329,
        -0.4255942702293396,
        0.19450682401657104,
        -0.39619654417037964,
        1.1550514698028564,
        0.9769047498703003,
        -0.7456809282302856,
        -1.3812838792800903,
        0.07662203907966614,
        -1.4201765060424805,
        0.3417740762233734,
        -0.3149818181991577,
        0.18378210067749023,
        -0.6355498433113098,
        0.02634390816092491,
        -0.0017376020550727844,
        -0.8521059155464172,
        -0.23024842143058777,
        -0.1522301733493805,
        -0.7140592932701111,
        0.03488187864422798,
        0.9833220839500427,
        -0.6239383816719055,
        0.6861448287963867,
        0.3349078893661499,
        -0.3160772919654846,
        -0.7955625057220459,
        -0.7610108852386475,
        -0.5100499391555786,
        0.034780535846948624,
        -0.12492454051971436,
        -0.06336754560470581,
        -0.6480013728141785,
        -0.5075410604476929,
        0.9258944988250732,
        -0.23691321909427643,
        0.422885924577713,
        1.1779035329818726,
        -0.29671525955200195,
        0.45856454968452454,
        2.2199974060058594,
        6.049717426300049,
        0.4247852861881256,
        0.03326593339443207,
        0.18774375319480896,
        -3.0617928504943848,
        -0.3538617491722107,
        0.8910201787948608,
        0.06939065456390381,
        -0.45427799224853516,
        0.1145935133099556,
        -1.8876785039901733,
        -0.6034613251686096,
        0.30548396706581116,
        -0.1077260747551918,
        -1.8353358507156372,
        0.6320286393165588,
        0.342405766248703,
        -0.9466231465339661,
        -0.2223590910434723,
        1.633811116218567,
        1.1988320350646973,
        0.5507926344871521,
        0.022187625989317894,
        0.08708589524030685,
        -0.46132567524909973,
        0.9039878845214844,
        1.2047632932662964,
        1.6597583293914795,
        0.3509847819805145,
        -0.7708027362823486,
        -0.00012385949958115816,
        0.02622387185692787,
        0.23687326908111572,
        -0.23483432829380035,
        0.785348653793335,
        1.387162685394287,
        0.05733492970466614,
        0.2834599018096924,
        -1.5549767017364502,
        -1.1869844198226929,
        -0.5001720786094666,
        0.1129777655005455,
        0.2399885356426239,
        -0.36499422788619995,
        0.11703646183013916,
        -0.9573560953140259,
        1.7035828828811646,
        -0.47999221086502075,
        -0.6850872039794922,
        -1.9630496501922607,
        -0.26216503977775574,
        -1.236363172531128,
        1.2131667137145996,
        -1.0335935354232788,
        0.49745985865592957,
        -0.36879947781562805,
        -1.4335675239562988,
        -1.2505924701690674,
        -0.09342067688703537,
        -0.1031719297170639,
        -0.845089316368103,
        -0.25144851207733154,
        0.37272748351097107,
        0.28259360790252686,
        0.1442316174507141,
        1.0918049812316895,
        -0.4222988486289978,
        -0.5521408915519714,
        -0.08937276899814606,
        -0.6735305190086365,
        0.09348229318857193,
        0.8985345363616943,
        0.4040052592754364,
        -0.9244634509086609,
        -0.8207292556762695,
        -0.7995850443840027,
        -0.7133394479751587,
        1.4457272291183472,
        0.6235929727554321,
        -1.1042652130126953,
        -0.18853268027305603,
        -0.030584143474698067,
        0.7769511342048645,
        0.21775323152542114,
        0.6901780962944031,
        0.0605451837182045,
        -0.2597784101963043,
        0.011389640159904957,
        -0.2782304883003235,
        0.5337550044059753,
        0.13002461194992065,
        2.858579635620117,
        2.2165541648864746,
        -1.6957606077194214,
        -1.3484234809875488,
        0.5086627006530762,
        0.5374454259872437,
        0.600132405757904,
        -0.35753437876701355,
        0.9092206358909607,
        0.5283166170120239,
        0.2509515583515167,
        -0.17753618955612183,
        0.6733176112174988,
        0.045449499040842056,
        -0.8949448466300964,
        -0.16959534585475922,
        -0.09252668917179108,
        1.1087617874145508,
        -0.3724087178707123,
        0.3322495222091675,
        0.48003607988357544,
        -0.3687056601047516,
        -1.3073554039001465,
        -0.31638821959495544,
        -0.5134118795394897,
        -0.9235721230506897,
        -0.36659881472587585,
        -0.1856650561094284,
        -0.3007179796695709,
        -0.2346922606229782,
        0.11110550165176392,
        0.5007892847061157,
        -1.2198755741119385,
        -0.3743126690387726,
        -0.3759503960609436,
        0.2762299180030823,
        2.34713077545166,
        6.7869110107421875,
        -0.003611281979829073,
        0.24288053810596466,
        -0.8049203157424927,
        -0.8765348792076111,
        -0.23434144258499146,
        2.1146841049194336,
        0.20689937472343445,
        -0.07038551568984985,
        0.377560019493103,
        -1.0866844654083252,
        -0.1386021077632904,
        0.6897559762001038,
        -0.4297366142272949,
        -0.39869794249534607,
        0.05390496179461479,
        0.7515197396278381,
        -1.0570094585418701,
        0.6468269228935242,
        1.0746899843215942,
        0.7809686660766602,
        0.11603648215532303,
        0.1724310964345932,
        0.10877268761396408,
        -0.5221180319786072,
        1.6492156982421875,
        2.0212700366973877,
        1.0316636562347412,
        1.2041032314300537,
        -0.4907228350639343,
        0.9765589237213135,
        0.3453936278820038,
        0.27868375182151794,
        -0.03254920244216919,
        -0.6575308442115784,
        0.10713095963001251,
        -0.04376266896724701,
        0.23342663049697876,
        -0.7458587288856506,
        -0.6449275016784668,
        -0.856620192527771,
        0.1639704704284668,
        0.01428339071571827,
        0.697925329208374,
        0.060491833835840225,
        -0.32361650466918945,
        -1.05996572971344,
        -0.12626299262046814,
        1.0041906833648682,
        -1.414699673652649,
        -0.13821031153202057,
        -1.269945740699768,
        0.1878547966480255,
        -0.8674177527427673,
        0.46431005001068115,
        0.2327304184436798,
        -1.3188105821609497,
        -0.9720017910003662,
        -0.5720484256744385,
        0.18390484154224396,
        0.16114766895771027,
        -0.8153122663497925,
        -0.3771165907382965,
        -0.061796486377716064,
        0.17708726227283478,
        0.5890390872955322,
        1.07102632522583,
        -0.3739709258079529,
        0.06789779663085938,
        0.3396647572517395,
        1.523109793663025,
        -0.7677808403968811,
        -0.6499733328819275,
        -0.043903522193431854,
        -0.9914679527282715,
        -1.8342633247375488,
        -0.1429421305656433,
        -0.07747496664524078,
        -0.6471755504608154,
        0.018194684758782387,
        0.8335069417953491,
        0.42902958393096924,
        0.7429405450820923,
        -0.07746612280607224,
        0.6400280594825745,
        0.06367930769920349,
        -0.2671132981777191,
        -0.025443751364946365,
        -0.7863265872001648,
        -0.12874262034893036,
        -0.5645725131034851,
        1.3786741495132446,
        0.6260976195335388,
        -0.37118515372276306,
        -1.3806756734848022,
        -0.8428974747657776,
        -0.8127872347831726,
        0.24264702200889587,
        1.0861274003982544,
        -0.6611175537109375,
        -1.5513105392456055,
        0.1873193085193634,
        -0.1576937735080719,
        -0.537710964679718,
        0.33131834864616394,
        0.05729297176003456,
        -0.5702476501464844,
        -0.6196622252464294,
        0.7584403157234192,
        -0.48518112301826477,
        0.10447536408901215,
        1.2382197380065918,
        -0.19881097972393036,
        -0.11184627562761307,
        -1.3794587850570679,
        -0.6244295239448547,
        -1.1302515268325806,
        -0.7000928521156311,
        -0.41212040185928345,
        -0.7566840052604675,
        -0.6411684155464172,
        1.1274703741073608,
        0.6366463303565979,
        -1.3027212619781494,
        0.6816779971122742,
        0.1727447509765625,
        0.25206440687179565,
        1.8186497688293457,
        3.5384328365325928,
        -0.09391462057828903,
        -0.31659889221191406,
        -0.4867439568042755,
        -0.6523049473762512,
        -0.03733422979712486,
        2.59932804107666,
        0.2793954014778137,
        -0.16736052930355072,
        0.2004922330379486,
        -1.0194627046585083,
        -0.2152218073606491,
        0.916074275970459,
        -0.7214213013648987,
        -0.2644707262516022,
        -0.5747984051704407,
        1.2473877668380737,
        -0.9385489821434021,
        0.41783803701400757,
        1.4941927194595337,
        2.101767063140869,
        0.36512520909309387,
        0.1193167120218277,
        0.17989879846572876,
        -0.5801778435707092,
        2.2088305950164795,
        1.5660384893417358,
        1.1682071685791016,
        0.563066303730011,
        -0.6116885542869568,
        1.3193466663360596,
        0.267682820558548,
        -0.62977534532547,
        0.010850176215171814,
        -0.10765872895717621,
        0.39721477031707764,
        -0.02617522142827511,
        -0.33186206221580505,
        -1.3612327575683594,
        -0.3969060480594635,
        -1.0420327186584473,
        0.19964535534381866,
        -0.05370185524225235,
        1.9150503873825073,
        0.10939188301563263,
        -0.812909722328186,
        -0.2523416578769684,
        0.13053910434246063,
        0.927209198474884,
        -2.232102155685425,
        -0.5840155482292175,
        -0.7618265748023987,
        -0.18549108505249023,
        -0.7721548080444336,
        0.30403372645378113,
        0.380097895860672,
        -0.3163313865661621,
        -1.6085025072097778,
        -0.24368107318878174,
        0.13427546620368958,
        0.19827839732170105,
        -0.4600118398666382,
        -0.3990802466869354,
        -0.14053016901016235,
        -0.23337261378765106,
        0.2538866102695465,
        1.2537442445755005,
        -0.7838678956031799,
        -0.10137762874364853,
        0.47031790018081665,
        0.5875551700592041,
        0.2403402030467987,
        -0.2355261594057083,
        -0.6959521770477295,
        -1.1201555728912354,
        -1.9424922466278076,
        0.20083189010620117,
        0.813184380531311,
        -0.3464331030845642,
        0.23560313880443573,
        0.5285450220108032,
        0.09845944494009018,
        0.4913387894630432,
        -0.5413938164710999,
        2.265829086303711,
        -0.003198265330865979,
        -0.24163955450057983,
        0.35089775919914246,
        -1.5483036041259766,
        -0.12350653856992722,
        -0.8283056020736694,
        1.4019381999969482,
        0.42219260334968567,
        0.4508354067802429,
        -1.8289210796356201,
        -0.9122130870819092,
        0.8255137205123901,
        0.7213162779808044,
        0.8427438139915466,
        -0.7342632412910461,
        -1.7567100524902344,
        0.39213860034942627,
        -0.15453149378299713,
        -0.4189983904361725,
        0.579968273639679,
        -0.7543401122093201,
        0.13695837557315826,
        -1.0355637073516846,
        0.13065554201602936,
        -0.2054397016763687,
        0.0388934388756752,
        -0.12198536843061447,
        -0.4686793386936188,
        -0.2934783399105072,
        -1.164990782737732,
        -0.4009651243686676,
        -0.8292542099952698,
        -0.2866267263889313,
        -0.2246156483888626,
        -0.3512592613697052,
        -0.5403620600700378,
        0.8036950826644897,
        0.47769179940223694
      ],
      "parity": {
        "fp32": {
          "maxAbsDiff": 0.008566856384277344,
          "maxAbsReference": 10.59189224243164
        },
        "f16w": {
          "maxAbsDiff": 0.011764049530029297,
          "maxAbsReference": 10.59189224243164
        }
      }
    },
    "regressor": {
      "numClasses": 0,
      "outDim": 999,
      "emptyRowEmbedding": [
        -0.4694979786872864,
        -0.5284363031387329,
        0.7942131161689758,
        -0.1740059107542038,
        0.0819745659828186,
        -1.3124985694885254,
        -1.2073546648025513,
        0.043875839561223984,
        0.4463379681110382,
        -0.06377585232257843,
        -0.6623969078063965,
        1.4375805854797363,
        0.1599443554878235,
        0.06443210691213608,
        -0.07009238004684448,
        0.1406291127204895,
        -0.10016635805368423,
        -0.11019332706928253,
        0.8808096051216125,
        1.9223638772964478,
        0.1885247379541397,
        -0.4930430054664612,
        0.14843468368053436,
        0.05192318931221962,
        0.052645809948444366,
        0.11989498138427734,
        0.7977968454360962,
        -0.15793216228485107,
        0.10221352428197861,
        0.9230976700782776,
        -0.2983953356742859,
        -0.3075169622898102,
        -2.147857904434204,
        -0.14144450426101685,
        0.8169675469398499,
        -0.7120842933654785,
        0.0256796982139349,
        -1.142195463180542,
        -0.11487342417240143,
        2.272392511367798,
        0.06951839476823807,
        -0.1910751760005951,
        -1.095170259475708,
        0.8089185953140259,
        -0.7023834586143494,
        -0.08879584074020386,
        0.17248710989952087,
        -0.7444038987159729,
        0.0691181942820549,
        1.8920302391052246,
        -0.3236050307750702,
        -0.3157457709312439,
        0.8135353326797485,
        -0.7584391236305237,
        -2.9412953853607178,
        0.04221900925040245,
        0.2583717405796051,
        -0.15304900705814362,
        0.11633336544036865,
        -2.524087905883789,
        0.2954463064670563,
        -1.1413064002990723,
        -0.8557645082473755,
        -1.2421867847442627,
        0.6307764053344727,
        0.022662712261080742,
        -0.22013036906719208,
        -0.015391030348837376,
        -0.1214335709810257,
        0.45584020018577576,
        -1.1608262062072754,
        0.0529545322060585,
        0.1444423943758011,
        0.0859779417514801,
        -0.11915931850671768,
        0.2944789528846741,
        -0.12866798043251038,
        0.9841223359107971,
        0.054290398955345154,
        0.47171881794929504,
        0.34118175506591797,
        -2.7080078125,
        0.04803191125392914,
        -2.8414955139160156,
        -1.3152353763580322,
        2.2643065452575684,
        -0.002209432190284133,
        1.4412553310394287,
        -1.8307772874832153,
        -0.09562166780233383,
        0.22316887974739075,
        0.14821569621562958,
        0.13283200562000275,
        0.04302593693137169,
        0.631517767906189,
        -0.22365802526474,
        2.861109733581543,
        -1.69280207157135,
        0.19328664243221283,
        0.10534053295850754,
        0.0377529077231884,
        -0.06221825256943703,
        -0.15376128256320953,
        2.131559133529663,
        0.43501541018486023,
        0.10493450611829758,
        0.7153786420822144,
        0.2233937680721283,
        -0.04615440592169762,
        0.15523748099803925,
        0.9501121640205383,
        0.06562787294387817,
        -4.547725200653076,
        -0.3559969663619995,
        -0.25796622037887573,
        -0.13153403997421265,
        -1.1350716352462769,
        0.42120644450187683,
        0.3186573386192322,
        0.10512907058000565,
        -0.10680767893791199,
        -0.06810915470123291,
        0.5287041068077087,
        -2.7750155925750732,
        0.006905125454068184,
        -2.2520101070404053,
        0.15578068792819977,
        -3.854083776473999,
        -0.42131710052490234,
        0.20317406952381134,
        0.983536958694458,
        1.3437145948410034,
        0.00420773308724165,
        -1.9905420541763306,
        -0.5379229187965393,
        0.0691051259636879,
        0.6821646094322205,
        -0.4331040382385254,
        0.03798132389783859,
        1.9691011905670166,
        1.4221292734146118,
        0.04968034848570824,
        -0.17964068055152893,
        0.15767499804496765,
        -0.04969444498419762,
        -0.37859687209129333,
        1.0235233306884766,
        1.4445098638534546,
        0.07226166874170303,
        -0.3599167466163635,
        0.13257861137390137,
        -0.42844775319099426,
        0.22744068503379822,
        -0.25191816687583923,
        0.7403735518455505,
        -0.14049169421195984,
        0.07010772824287415,
        0.06584154069423676,
        -0.9126791954040527,
        -0.8162152767181396,
        -0.954106330871582,
        -0.5332455039024353,
        1.339992880821228,
        0.05303521081805229,
        0.01032345648854971,
        -1.28511381149292,
        0.07129594683647156,
        0.6310217976570129,
        0.1765824407339096,
        -0.13819557428359985,
        -0.850491464138031,
        0.9061627984046936,
        -0.3883548080921173,
        -0.08780547231435776,
        0.023276332765817642,
        0.2883932590484619,
        0.21321770548820496,
        2.3854525089263916,
        0.009892505593597889,
        -0.18126742541790009,
        0.6580517292022705,
        -0.4252259135246277,
        -0.8437998294830322,
        0.09941366314888,
        -0.7027503848075867,
        -0.5152944326400757,
        0.08010787516832352,
        -3.069200038909912,
        0.10175176709890366,
        -1.429904580116272,
        -0.6756195425987244,
        -1.1351020336151123,
        1.1273844242095947,
        0.42387914657592773,
        0.08681066334247589,
        -1.101196050643921,
        -0.09928438067436218,
        0.5539436340332031,
        -1.188781976699829,
        -0.20027965307235718,
        0.36987486481666565,
        0.04507005959749222,
        -0.3869955539703369,
        0.3004930317401886,
        -0.5720422267913818,
        0.7905634045600891,
        -0.2781822383403778,
        1.0173399448394775,
        0.08428431302309036,
        -1.9608047008514404,
        0.1558610498905182,
        -2.1886048316955566,
        -0.8701810240745544,
        2.7474851608276367,
        0.044570595026016235,
        1.9139198064804077,
        -1.1593388319015503,
        0.27288296818733215,
        -0.03178431838750839,
        0.2022245079278946,
        -0.13363920152187347,
        0.0879179835319519,
        0.6321004629135132,
        -0.16256083548069,
        2.707683801651001,
        -0.46909424662590027,
        -0.20950688421726227,
        -0.27833977341651917,
        0.6684026718139648,
        0.04562697932124138,
        -0.10078641027212143,
        1.2334874868392944,
        0.3106614351272583,
        0.14261791110038757,
        0.7899978756904602,
        0.7311763763427734,
        0.3256126046180725,
        0.04568248614668846,
        1.373286485671997,
        -0.08564990013837814,
        -4.541531085968018,
        -1.6988294124603271,
        -0.14241217076778412,
        -0.4839921295642853,
        -0.5404940247535706,
        -1.046378254890442,
        0.29746195673942566,
        -0.1112639531493187,
        -0.4752063453197479,
        -0.1271742880344391,
        2.1555097103118896,
        -1.3987003564834595,
        0.09240797907114029,
        -2.9889349937438965,
        0.15221719443798065,
        -1.5289369821548462,
        -0.5043174624443054,
        -0.6289622187614441,
        0.36569666862487793,
        1.7360724210739136,
        -0.0156303271651268,
        -1.7020562887191772,
        -1.234953761100769,
        -0.04897680506110191,
        0.0015682547818869352,
        -0.19800569117069244,
        -0.5232179760932922,
        0.4615810215473175,
        0.38175562024116516,
        -0.10689545422792435,
        -0.180118128657341,
        0.41886481642723083,
        -0.16819797456264496,
        -0.13090743124485016,
        0.7626848816871643,
        1.5892177820205688,
        0.22286584973335266,
        -0.37462764978408813,
        0.03223014995455742,
        -1.0433772802352905,
        0.09789539128541946,
        0.0882999524474144,
        0.7384302616119385,
        0.257904589176178,
        0.14308391511440277,
        0.3340730369091034,
        -0.37996524572372437,
        -0.5544859170913696,
        -2.0617072582244873,
        -0.10683762282133102,
        0.8719659447669983,
        -0.0953662171959877,
        0.02857489138841629,
        -1.7508457899093628,
        -0.001652607461437583,
        1.0523642301559448,
        -1.5780483484268188,
        -0.06473027914762497,
        -2.129587411880493,
        0.7441123127937317,
        -0.6347180604934692,
        -0.09376972168684006,
        0.19356223940849304,
        1.0599101781845093,
        0.010516265407204628,
        0.7719600796699524,
        -0.16044656932353973,
        -0.23055405914783478,
        0.5584052205085754,
        -0.6527419686317444,
        -1.7498098611831665,
        -0.010209263302385807,
        0.17469215393066406,
        -0.2665475904941559,
        0.010467955842614174,
        -2.68434739112854,
        0.5477709174156189,
        -1.2088308334350586,
        -0.4581184685230255,
        -1.1672649383544922,
        0.6092084646224976,
        0.4373133182525635,
        -0.03000088408589363,
        -0.3497457206249237,
        -0.19048893451690674,
        0.6278586983680725,
        0.6610175967216492,
        -0.14683549106121063,
        0.8250941634178162,
        0.07790695130825043,
        -0.24170733988285065,
        0.30327358841896057,
        -0.15131603181362152,
        0.8567246198654175,
        -0.32184523344039917,
        0.8064577579498291,
        0.34245675802230835,
        -0.2098846584558487,
        -0.08177467435598373,
        -3.3721365928649902,
        -0.9989380836486816,
        2.252946138381958,
        -0.16334299743175507,
        1.6002048254013062,
        -0.6729496717453003,
        0.10327114164829254,
        0.010235420428216457,
        -0.022049760445952415,
        0.005901553202420473,
        0.06463601440191269,
        0.859628438949585,
        -0.4298275113105774,
        2.4730336666107178,
        -2.4295125007629395,
        0.20150068402290344,
        0.21143126487731934,
        0.03119017370045185,
        0.13222353160381317,
        -0.14819854497909546,
        1.4167088270187378,
        0.5117492079734802,
        0.3221425712108612,
        -0.30901554226875305,
        1.2348562479019165,
        0.08088135719299316,
        -0.10042165964841843,
        0.8788049221038818,
        0.05959128215909004,
        -5.117993354797363,
        -0.912495493888855,
        -0.29120442271232605,
        -0.46083927154541016,
        -1.2243729829788208,
        0.5436981916427612,
        0.420470654964447,
        0.36318808794021606,
        -0.2511226534843445,
        -0.0725305899977684,
        2.0336923599243164,
        -1.3193378448486328,
        0.030161553993821144,
        -2.68609619140625,
        0.22209231555461884,
        -2.029188632965088,
        -0.30868279933929443,
        0.05908827483654022,
        0.793724775314331,
        -0.9478907585144043,
        0.11153782904148102,
        -1.657224416732788,
        -0.5938335061073303,
        0.01591452583670616,
        1.3182036876678467,
        -0.1615467667579651,
        -0.2587709426879883,
        1.1507468223571777,
        0.4555306136608124,
        0.033915795385837555,
        0.06200772896409035,
        0.17856965959072113,
        -0.028422415256500244,
        -0.3590381145477295,
        0.7090999484062195,
        1.7100611925125122,
        0.14070144295692444,
        -0.5500832200050354,
        0.20478008687496185,
        -0.1213708221912384,
        0.20196767151355743,
        -0.12583468854427338,
        0.6796978712081909,
        -0.30199992656707764,
        0.03979099541902542,
        0.5898949503898621,
        0.03335940092802048,
        -0.5206530690193176,
        -0.9246997833251953,
        -0.08036378026008606,
        1.0086718797683716,
        -0.3107143044471741,
        -0.15632975101470947,
        -0.5647521615028381,
        0.0388534851372242,
        1.5328946113586426,
        0.5130590200424194,
        -0.3926909863948822,
        0.46294018626213074,
        0.42535480856895447,
        -0.19435879588127136,
        -0.1934869885444641,
        0.029737945646047592,
        -0.6507677435874939,
        0.15519340336322784,
        2.130070924758911,
        -0.2242933064699173,
        -0.3942379653453827,
        0.8409155011177063,
        -0.5526861548423767,
        -2.8079497814178467,
        0.13277432322502136,
        -0.04973578080534935,
        -0.17795336246490479,
        0.09016696363687515,
        -2.7635128498077393,
        -0.5937476754188538,
        -1.7419863939285278,
        -0.6703135371208191,
        -0.7442510724067688,
        1.0544700622558594,
        -0.614046037197113,
        -0.14987458288669586,
        -0.23138372600078583,
        -0.06697559356689453,
        0.4892526865005493,
        -1.0696855783462524,
        -0.09890517592430115,
        -0.12672454118728638,
        0.08710630238056183,
        -0.14931896328926086,
        0.34537580609321594,
        -0.26897406578063965,
        0.9914587736129761,
        0.0023401814978569746,
        0.3859562575817108,
        0.26136720180511475,
        -3.4191715717315674,
        0.12051139771938324,
        -1.924551248550415,
        -1.0318630933761597,
        2.1765992641448975,
        -0.176780566573143,
        1.68215012550354,
        -2.0660247802734375,
        0.10290340334177017,
        0.23090112209320068,
        0.21758340299129486,
        -0.03828301653265953,
        0.04786951094865799,
        0.6170095801353455,
        0.21853074431419373,
        1.5742398500442505,
        -0.31364452838897705,
        -0.23484764993190765,
        0.11317507922649384,
        0.3531038761138916,
        -0.24483861029148102,
        -0.2011033594608307,
        2.19809627532959,
        0.40214434266090393,
        -0.02699330449104309,
        1.0313245058059692,
        -0.1514972448348999,
        0.3261636197566986,
        0.23002180457115173,
        1.2791661024093628,
        0.07568946480751038,
        -3.383969306945801,
        -1.2306833267211914,
        -0.13302189111709595,
        -0.06789056211709976,
        -0.4665857255458832,
        -0.09816668927669525,
        0.3828674554824829,
        -0.5048136115074158,
        -0.34226295351982117,
        -0.09143981337547302,
        0.2877776324748993,
        -2.8805155754089355,
        0.046335142105817795,
        -1.0330597162246704,
        0.10214007645845413,
        -2.3693912029266357
      ],
      "parity": {
        "fp32": {
          "maxAbsDiff": 0.0016391277313232422,
          "maxAbsReference": 4.916708469390869
        },
        "f16w": {
          "maxAbsDiff": 0.006482720375061035,
          "maxAbsReference": 4.916708469390869
        }
      }
    }
  },
  "files": [
    {
      "task": "classifier",
      "stage": "col",
      "variant": "f16w",
      "file": "classifier-col.f16w.onnx",
      "bytes": 1881044,
      "sha256": "35a0cde45113e557ec25ff7300d94f8c1114f725517cc36654e004e8280faa51",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "X",
          "shape": [
            "columns",
            "rows",
            3
          ]
        },
        {
          "name": "y",
          "shape": [
            "train"
          ]
        },
        {
          "name": "keys",
          "shape": [
            "keys"
          ]
        }
      ],
      "outputs": [
        {
          "name": "E",
          "shape": [
            "columns",
            "rows",
            128
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "hop",
      "variant": "f16w",
      "file": "classifier-hop.f16w.onnx",
      "bytes": 4209838,
      "sha256": "5a9bb35380493127f4a76adf44a19a33a9a69d1b58343f38ecdd4f77792d9b4f",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "x",
          "shape": [
            "nodes",
            512
          ]
        },
        {
          "name": "src",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "dst",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "edge_type",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "noise",
          "shape": [
            "types",
            512
          ]
        },
        {
          "name": "deg",
          "shape": [
            "nodes",
            1
          ]
        }
      ],
      "outputs": [
        {
          "name": "h",
          "shape": [
            "nodes",
            512
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "icl",
      "variant": "f16w",
      "file": "classifier-icl.f16w.onnx",
      "bytes": 52943138,
      "sha256": "e161ab3388e4fb622513c5c8a6b14f73cdba4575b4c16dfb1d9a1b518379f27f",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "Z",
          "shape": [
            "tokens",
            512
          ]
        },
        {
          "name": "y",
          "shape": [
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "out",
          "shape": [
            "queries",
            10
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "out",
      "variant": "f16w",
      "file": "classifier-out.f16w.onnx",
      "bytes": 531196,
      "sha256": "7acf1c5378cb5e2edf67b8040a440914e0cd7d8a2081737b2e863ac8fc566b5a",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "x",
          "shape": [
            "readouts",
            512
          ]
        }
      ],
      "outputs": [
        {
          "name": "z",
          "shape": [
            "readouts",
            512
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "row",
      "variant": "f16w",
      "file": "classifier-row.f16w.onnx",
      "bytes": 854772,
      "sha256": "87b5b9cd288b251bee4b84a056c123941407174a7d0c887a1e386bf326aad3f5",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "E",
          "shape": [
            "columns",
            "rows",
            128
          ]
        }
      ],
      "outputs": [
        {
          "name": "Z",
          "shape": [
            "rows",
            512
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "col",
      "variant": "fp32",
      "file": "classifier-col.fp32.onnx",
      "bytes": 3613403,
      "sha256": "4258b08a30fa4eec00b5804f6c575b4574584ad6e1a311f9b211030a3533ea6f",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "X",
          "shape": [
            "columns",
            "rows",
            3
          ]
        },
        {
          "name": "y",
          "shape": [
            "train"
          ]
        },
        {
          "name": "keys",
          "shape": [
            "keys"
          ]
        }
      ],
      "outputs": [
        {
          "name": "E",
          "shape": [
            "columns",
            "rows",
            128
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "hop",
      "variant": "fp32",
      "file": "classifier-hop.fp32.onnx",
      "bytes": 8403886,
      "sha256": "a52f544ef90eb6de61c517e7e61c7fe63a6e3286a0a0a20356eae2e169616333",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "x",
          "shape": [
            "nodes",
            512
          ]
        },
        {
          "name": "src",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "dst",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "edge_type",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "noise",
          "shape": [
            "types",
            512
          ]
        },
        {
          "name": "deg",
          "shape": [
            "nodes",
            1
          ]
        }
      ],
      "outputs": [
        {
          "name": "h",
          "shape": [
            "nodes",
            512
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "icl",
      "variant": "fp32",
      "file": "classifier-icl.fp32.onnx",
      "bytes": 105355516,
      "sha256": "6a0f29920552b068feb889bbc4aa58ebf3ce8514b8c29c071a6a7baabd6a5b1b",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "Z",
          "shape": [
            "tokens",
            512
          ]
        },
        {
          "name": "y",
          "shape": [
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "out",
          "shape": [
            "queries",
            10
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "out",
      "variant": "fp32",
      "file": "classifier-out.fp32.onnx",
      "bytes": 1055415,
      "sha256": "5bbde304395d3eb511173a82fbc8ac3ffdf11c9d933c5ebc7df7ab439d0e38bf",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "x",
          "shape": [
            "readouts",
            512
          ]
        }
      ],
      "outputs": [
        {
          "name": "z",
          "shape": [
            "readouts",
            512
          ]
        }
      ]
    },
    {
      "task": "classifier",
      "stage": "row",
      "variant": "fp32",
      "file": "classifier-row.fp32.onnx",
      "bytes": 1640460,
      "sha256": "58f6cb77884920c4e92813cb785fa8a95567a94814a2f5d7ae8a4827c214e328",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "E",
          "shape": [
            "columns",
            "rows",
            128
          ]
        }
      ],
      "outputs": [
        {
          "name": "Z",
          "shape": [
            "rows",
            512
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "col",
      "variant": "f16w",
      "file": "regressor-col.f16w.onnx",
      "bytes": 1870902,
      "sha256": "b82598570e8dca41ecb38371f0b80588d82cd115bddc72324f43e138ef27ed1e",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "X",
          "shape": [
            "columns",
            "rows",
            3
          ]
        },
        {
          "name": "y",
          "shape": [
            "train"
          ]
        },
        {
          "name": "keys",
          "shape": [
            "keys"
          ]
        }
      ],
      "outputs": [
        {
          "name": "E",
          "shape": [
            "columns",
            "rows",
            128
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "hop",
      "variant": "f16w",
      "file": "regressor-hop.f16w.onnx",
      "bytes": 4209838,
      "sha256": "d0fe80252bd14032e060beb5d46fc9cb7f5f98458cbd6bfbef90568229c50414",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "x",
          "shape": [
            "nodes",
            512
          ]
        },
        {
          "name": "src",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "dst",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "edge_type",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "noise",
          "shape": [
            "types",
            512
          ]
        },
        {
          "name": "deg",
          "shape": [
            "nodes",
            1
          ]
        }
      ],
      "outputs": [
        {
          "name": "h",
          "shape": [
            "nodes",
            512
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "icl",
      "variant": "f16w",
      "file": "regressor-icl.f16w.onnx",
      "bytes": 54911832,
      "sha256": "63b6a9969914ca2f3aa3c4684fb75c71a84f7d2536a9e87674f4771fcef14b9e",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "Z",
          "shape": [
            "tokens",
            512
          ]
        },
        {
          "name": "y",
          "shape": [
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "out",
          "shape": [
            "queries",
            999
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "out",
      "variant": "f16w",
      "file": "regressor-out.f16w.onnx",
      "bytes": 531196,
      "sha256": "4f7f345e5c0632bb8318aee8756cb470a0fc3888db32ee55e3bb398726d0b13f",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "x",
          "shape": [
            "readouts",
            512
          ]
        }
      ],
      "outputs": [
        {
          "name": "z",
          "shape": [
            "readouts",
            512
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "row",
      "variant": "f16w",
      "file": "regressor-row.f16w.onnx",
      "bytes": 850342,
      "sha256": "18a9cf22082b5c71eb463e20b8a0ff2ca8b2a018afc6492431e869970d924fcc",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "E",
          "shape": [
            "columns",
            "rows",
            128
          ]
        }
      ],
      "outputs": [
        {
          "name": "Z",
          "shape": [
            "rows",
            512
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "col",
      "variant": "fp32",
      "file": "regressor-col.fp32.onnx",
      "bytes": 3600764,
      "sha256": "b0feb762e03f66e381ffb12226b62af4e41b08ad2a628baee193b3bb4a1ab997",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "X",
          "shape": [
            "columns",
            "rows",
            3
          ]
        },
        {
          "name": "y",
          "shape": [
            "train"
          ]
        },
        {
          "name": "keys",
          "shape": [
            "keys"
          ]
        }
      ],
      "outputs": [
        {
          "name": "E",
          "shape": [
            "columns",
            "rows",
            128
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "hop",
      "variant": "fp32",
      "file": "regressor-hop.fp32.onnx",
      "bytes": 8403886,
      "sha256": "f824b39638fe07d8b9aacd6e310be540c31390071c510daec50fedc4abb52f02",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "x",
          "shape": [
            "nodes",
            512
          ]
        },
        {
          "name": "src",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "dst",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "edge_type",
          "shape": [
            "edges"
          ]
        },
        {
          "name": "noise",
          "shape": [
            "types",
            512
          ]
        },
        {
          "name": "deg",
          "shape": [
            "nodes",
            1
          ]
        }
      ],
      "outputs": [
        {
          "name": "h",
          "shape": [
            "nodes",
            512
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "icl",
      "variant": "fp32",
      "file": "regressor-icl.fp32.onnx",
      "bytes": 109339507,
      "sha256": "97c3e0277eec6e9f1f01cd8f0e126c86745b0751f83805124151568112450847",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "Z",
          "shape": [
            "tokens",
            512
          ]
        },
        {
          "name": "y",
          "shape": [
            "train"
          ]
        }
      ],
      "outputs": [
        {
          "name": "out",
          "shape": [
            "queries",
            999
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "out",
      "variant": "fp32",
      "file": "regressor-out.fp32.onnx",
      "bytes": 1055415,
      "sha256": "de9b4ee60e328b8c6cf61924f0e4abb89968e900ed1596e80c3e355a5ad26386",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "x",
          "shape": [
            "readouts",
            512
          ]
        }
      ],
      "outputs": [
        {
          "name": "z",
          "shape": [
            "readouts",
            512
          ]
        }
      ]
    },
    {
      "task": "regressor",
      "stage": "row",
      "variant": "fp32",
      "file": "regressor-row.fp32.onnx",
      "bytes": 1636030,
      "sha256": "5ffa74477d4aa2a33fb2256d8cb053ccb8158e9da330b2317e6e02b113106c11",
      "cpuNodes": [],
      "inputs": [
        {
          "name": "E",
          "shape": [
            "columns",
            "rows",
            128
          ]
        }
      ],
      "outputs": [
        {
          "name": "Z",
          "shape": [
            "rows",
            512
          ]
        }
      ]
    }
  ]
}