ONNX exports of TabICLv2, KumoTabular (small, medium, large), KumoRelational, Chronos-2 and Kumo-Anomaly for Tabular Data Workbench
b043bdf verified | { | |
| "opset": 18, | |
| "files": [ | |
| { | |
| "model": "chronos-2-small", | |
| "variant": "fp32", | |
| "file": "chronos-2-small.fp32.onnx", | |
| "bytes": 111801066, | |
| "sha256": "d11c8ebf5caeafa27205ba8a5919135e4a75cc25648f81d109e06d39513e4534", | |
| "cpuNodes": [] | |
| }, | |
| { | |
| "model": "chronos-2-small", | |
| "variant": "f16w", | |
| "file": "chronos-2-small.f16w.onnx", | |
| "bytes": 55960044, | |
| "sha256": "56e2743689b8c841d39ec5935d0fd18aa92dd5e8fc83710e08009791e0eff5b2", | |
| "cpuNodes": [] | |
| }, | |
| { | |
| "model": "chronos-2", | |
| "variant": "fp32", | |
| "file": "chronos-2.fp32.onnx", | |
| "bytes": 478045331, | |
| "sha256": "8e606c3d80cc3c1323466b864bb7236842fbe86d639df75598ec69389a0a8221", | |
| "cpuNodes": [] | |
| }, | |
| { | |
| "model": "chronos-2", | |
| "variant": "f16w", | |
| "file": "chronos-2.f16w.onnx", | |
| "bytes": 239161416, | |
| "sha256": "44fa813bd35869f5765234cfae8d0e4a337b7c25c50c15ecb66944fc46e1f5a3", | |
| "cpuNodes": [] | |
| }, | |
| { | |
| "model": "kumo-anomaly", | |
| "variant": "fp32", | |
| "file": "kumo-anomaly.fp32.onnx", | |
| "bytes": 10398173, | |
| "sha256": "3627470deafd7d497b6b37704a9adedf422bed7b070076bdbf54fe21ab9d5210", | |
| "cpuNodes": [] | |
| }, | |
| { | |
| "model": "kumo-anomaly", | |
| "variant": "f16w", | |
| "file": "kumo-anomaly.f16w.onnx", | |
| "bytes": 5327808, | |
| "sha256": "351fa5558baed4b91b10b8f516660b02da3727db61c6fca3e366b901c665851f", | |
| "cpuNodes": [] | |
| } | |
| ], | |
| "models": { | |
| "chronos-2-small": { | |
| "family": "chronos2", | |
| "source": "autogluon/chronos-2-small@ddec01313e50b6bc58ebaa92ede81bc24a3d9f9a", | |
| "parameters": 27934624, | |
| "contextLength": 8192, | |
| "maxOutputPatches": 64, | |
| "patch": 16, | |
| "timeScale": 8192.0, | |
| "dKv": 64, | |
| "quantiles": [ | |
| 0.01, | |
| 0.05, | |
| 0.1, | |
| 0.2, | |
| 0.3, | |
| 0.4, | |
| 0.5, | |
| 0.6, | |
| 0.7, | |
| 0.8, | |
| 0.9, | |
| 0.95, | |
| 0.99 | |
| ], | |
| "exporter": "torch.onnx (dynamo) 2.14.0+cpu, chronos-forecasting 2.3.2", | |
| "checks": { | |
| "air_passengers": { | |
| "fp32": { | |
| "maxAbsDiff": 0.0004534340109785262, | |
| "maxAbsDiffOverSpread": 1.2389919830395411e-06 | |
| }, | |
| "f16w": { | |
| "maxAbsDiff": 0.14048303526010386, | |
| "maxAbsDiffOverSpread": 0.0003838647966982188 | |
| } | |
| }, | |
| "daily_weekly": { | |
| "fp32": { | |
| "maxAbsDiff": 1.936539992186681e-05, | |
| "maxAbsDiffOverSpread": 7.937634073890962e-07 | |
| }, | |
| "f16w": { | |
| "maxAbsDiff": 0.00816448589266372, | |
| "maxAbsDiffOverSpread": 0.00033465201689035035 | |
| } | |
| }, | |
| "daily_weekly_covariate": { | |
| "fp32": { | |
| "maxAbsDiff": 1.4207511100039483e-05, | |
| "maxAbsDiffOverSpread": 9.504695011556084e-07 | |
| }, | |
| "f16w": { | |
| "maxAbsDiff": 0.00725193513707012, | |
| "maxAbsDiffOverSpread": 0.0004851478294551331 | |
| } | |
| }, | |
| "air_passengers_gaps": { | |
| "fp32": { | |
| "maxAbsDiff": 0.0005596565754331095, | |
| "maxAbsDiffOverSpread": 8.044586999305215e-07 | |
| }, | |
| "f16w": { | |
| "maxAbsDiff": 0.12156327028947089, | |
| "maxAbsDiffOverSpread": 0.00017473685590255084 | |
| } | |
| } | |
| } | |
| }, | |
| "chronos-2": { | |
| "family": "chronos2", | |
| "source": "amazon/chronos-2@29ec3766d36d6f73f0696f85560a422f50e8498c", | |
| "parameters": 119477664, | |
| "contextLength": 8192, | |
| "maxOutputPatches": 64, | |
| "patch": 16, | |
| "timeScale": 8192.0, | |
| "dKv": 64, | |
| "quantiles": [ | |
| 0.01, | |
| 0.05, | |
| 0.1, | |
| 0.15, | |
| 0.2, | |
| 0.25, | |
| 0.3, | |
| 0.35, | |
| 0.4, | |
| 0.45, | |
| 0.5, | |
| 0.55, | |
| 0.6, | |
| 0.65, | |
| 0.7, | |
| 0.75, | |
| 0.8, | |
| 0.85, | |
| 0.9, | |
| 0.95, | |
| 0.99 | |
| ], | |
| "exporter": "torch.onnx (dynamo) 2.14.0+cpu, chronos-forecasting 2.3.2", | |
| "checks": { | |
| "air_passengers": { | |
| "fp32": { | |
| "maxAbsDiff": 0.00041321987168885244, | |
| "maxAbsDiffOverSpread": 1.3253376289255459e-06 | |
| }, | |
| "f16w": { | |
| "maxAbsDiff": 0.07163125687907268, | |
| "maxAbsDiffOverSpread": 0.00022974596976921714 | |
| } | |
| }, | |
| "daily_weekly": { | |
| "fp32": { | |
| "maxAbsDiff": 2.2354764610099664e-05, | |
| "maxAbsDiffOverSpread": 9.333428173677023e-07 | |
| }, | |
| "f16w": { | |
| "maxAbsDiff": 0.011134839485819725, | |
| "maxAbsDiffOverSpread": 0.00046489518623411645 | |
| } | |
| }, | |
| "daily_weekly_covariate": { | |
| "fp32": { | |
| "maxAbsDiff": 1.1173277229659107e-05, | |
| "maxAbsDiffOverSpread": 9.343844901163358e-07 | |
| }, | |
| "f16w": { | |
| "maxAbsDiff": 0.0031910641166916776, | |
| "maxAbsDiffOverSpread": 0.00026685821503548776 | |
| } | |
| }, | |
| "air_passengers_gaps": { | |
| "fp32": { | |
| "maxAbsDiff": 0.000852749681143905, | |
| "maxAbsDiffOverSpread": 1.0165590811508801e-06 | |
| }, | |
| "f16w": { | |
| "maxAbsDiff": 0.21954769663511797, | |
| "maxAbsDiffOverSpread": 0.00026172182727855466 | |
| } | |
| } | |
| } | |
| }, | |
| "kumo-anomaly": { | |
| "family": "kumo-anomaly", | |
| "source": "nvidia/Kumo-Anomaly@226c5003b0582adc1cac433123f47324bfae39f2", | |
| "parameters": 2439169, | |
| "channels": 40, | |
| "window": 100, | |
| "trainSteps": 500, | |
| "betas": [ | |
| 0.0001, | |
| 0.01 | |
| ], | |
| "exporter": "torch.onnx (dynamo) 2.14.0+cpu", | |
| "checks": { | |
| "samples4-steps6": { | |
| "maxAbsDiff": 1.0004111891115386e-06, | |
| "maxScore": 1.0615177154541016, | |
| "graphCalls": 12, | |
| "cpuSeconds": 16.310610055923462 | |
| } | |
| } | |
| } | |
| } | |
| } | |