ONNX exports of TabICLv2, KumoTabular (small, medium, large), KumoRelational, Chronos-2 and Kumo-Anomaly for Tabular Data Workbench
b043bdf verified | KumoTabular (small, medium and large; classifier and regressor) converted to | |
| ONNX for Tabular Data Workbench. | |
| This is an unofficial conversion. NVIDIA has not reviewed or endorsed it. | |
| Kumo Tabular | |
| Source: nvidia/Kumo-Tabular, tag v1.0.1 | |
| (revision 3c3e10bbdb590ace29e7026847f92db3c603096d) | |
| https://huggingface.co/nvidia/Kumo-Tabular | |
| Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. | |
| The weights are licensed under the OpenMDW License Agreement, version 1.1 | |
| (LICENSE). | |
| Model code | |
| The graphs were exported from the KumoTabular implementation in | |
| structured-data-models 0.2.1 (https://github.com/NVIDIA/structured-data-models), | |
| Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES, licensed under the | |
| Apache License, Version 2.0 (LICENSE.structured-data-models where it is | |
| distributed with these files). In that implementation the cell embedding is | |
| derived from TabFM (https://github.com/google-research/tabfm, Copyright 2026 | |
| Google LLC, Apache License 2.0), and the attention scaling contains portions | |
| derived from TabICLv2 (https://github.com/soda-inria/tabicl), whose licence | |
| follows at the end of this file. | |
| Changes | |
| - <size>-<head>.f16w.onnx: the KumoTabular network (structured-data-models' | |
| _KumoTabular.forward, without the preprocessing recipe or the ensemble) | |
| exported with torch.onnx (dynamo, opset 20). The cell embedding's column | |
| grouping, imputation and Fourier features are written without int64 | |
| arithmetic or scatters; what they compute is unchanged. Every weight of | |
| 1,024 or more elements is stored as float16 and cast back to float32 on | |
| load; the other weights are unchanged. | |
| TabICLv2 | |
| BSD 3-Clause License | |
| Copyright (c) 2025, Soda team @ Inria | |
| Redistribution and use in source and binary forms, with or without | |
| modification, are permitted provided that the following conditions are met: | |
| 1. Redistributions of source code must retain the above copyright notice, this | |
| list of conditions and the following disclaimer. | |
| 2. Redistributions in binary form must reproduce the above copyright notice, | |
| this list of conditions and the following disclaimer in the documentation | |
| and/or other materials provided with the distribution. | |
| 3. Neither the name of the copyright holder nor the names of its | |
| contributors may be used to endorse or promote products derived from | |
| this software without specific prior written permission. | |
| THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" | |
| AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
| IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE | |
| DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE | |
| FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |
| DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR | |
| SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER | |
| CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, | |
| OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
| OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |