Hugging Face's logo

ScriptSmith
/
tabular-data-workbench-onnx

Tabular Classification
ONNX
onnxruntime-web
webgpu
tabular-regression
time-series-forecasting
anomaly-detection
tabular-foundation-model
relational-foundation-model
in-context-learning
Model card Files
xet
 Community
tabular-data-workbench-onnx / rfm /manifest.json
ScriptSmith's picture
ScriptSmith
ONNX exports of TabICLv2, KumoTabular (small, medium, large), KumoRelational, Chronos-2 and Kumo-Anomaly for Tabular Data Workbench
b043bdf verified
Raw History Blame Contribute Delete
45.4 kB
{
"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
]
}
]
}
]
}