Skip to page content
SAVRN
Search Plan a project

Open-weight model · Tabular classification

tabpfn-mix-1.0-classifier

by Autogluon autogluon/tabpfn-mix-1.0-classifier

tabpfn-mix-1.0-classifier is an open-weight model for tabular classification from Autogluon, released under Apache License 2.0. It has 39M parameters. At 16-bit it needs about 0.1 GB of GPU memory, which fits on 1x MI300X from $1.85 an hour, at the lowest prices in the SAVRN Index. It draws 906 downloads a month.

TabPFNMix classifier is a tabular foundation model that is pre-trained on purely synthetic datasets sampled from a mix of random classifiers. TabPFNMix is based on a 12-layer encoder-decoder Transformer of 37 M parameters.

Parameters39M
Context—
Weights155.8 MB
Licenseapache-2.0
AccessOpen weights
Monthly Downloads906

Runs On

What it takes to serve tabpfn-mix-1.0-classifier (39M parameters): the memory its weights need at each precision, and the cheapest way to rent enough data-center GPUs to hold them.

PrecisionWeightsMemory neededCheapest setupPer hourAlso fits
16-bit 0.1 GB 0.1 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
8-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
4-bit 0.0 GB 0.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00

Memory is the weights at that precision plus 20% for the runtime and a short context; a long context needs more. Prices are the lowest on-demand hourly rates in the SAVRN Index, read Oct 16, 2026.

tabpfn-mix-1.0-classifier on every accelerator the SAVRN Index prices, at every precision

Model Card

By Autogluon, published under apache-2.0, revision 80a938af912a.

TabPFNMix Classifier

TabPFNMix classifier is a tabular foundation model that is pre-trained on purely synthetic datasets sampled from a mix of random classifiers.

Architecture

TabPFNMix is based on a 12-layer encoder-decoder Transformer of 37 M parameters. We use a pre-training strategy incorporating in-context learning, similar to that used by TabPFN and TabForestPFN.

Usage

To use TabPFNMix classifier, install AutoGluon by running:

pip install autogluon

A minimal example showing how to perform fine-tuning and inference using the TabPFNMix classifier:

import pandas as pd

from autogluon.tabular import TabularPredictor


if __name__ == '__main__':
    train_data = pd.read_csv('https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv')
    subsample_size = 5000  
    if subsample_size is not None and subsample_size < len(train_data):
        train_data = train_data.sample(n=subsample_size, random_state=0)
    test_data = pd.read_csv('https://autogluon.s3.amazonaws.com/datasets/Inc/test.csv')

    tabpfnmix_default = {
        "model_path_classifier": "autogluon/tabpfn-mix-1.0-classifier",
        "model_path_regressor": "autogluon/tabpfn-mix-1.0-regressor",
        "n_ensembles": 1,
        "max_epochs": 30,
    }

    hyperparameters = {
        "TABPFNMIX": [
            tabpfnmix_default,
        ],
    }

    label = "class"

    predictor = TabularPredictor(label=label)
    predictor = predictor.fit(
        train_data=train_data,
        hyperparameters=hyperparameters,
        verbosity=3,
    )

    predictor.leaderboard(test_data, display=True)

Citation

Read the full model card (260 words)

Identity and Version

Repository
autogluon/tabpfn-mix-1.0-classifier
Publisher
Autogluon
Task
Tabular classification
Modality
Tabular
Library
Not stated by the source
Parameters
39M parameters
Languages
Not stated by the source
Revision
80a938af912a56967e8a62b8d36efba98adc116c
First published
2024-11-22
Last updated
2024-11-27

Files and Weights

4 files, 155.8 MB in total. The weights are 1 file totalling 155.8 MB in safetensors.

Weights1 file · 155.8 MB
Configuration1 file · 145 B
Documentation1 file · 2.7 KB
Repository1 file · 1.5 KB
Every file
FileTypeSizeSHA-256
model.safetensorsWeights155.8 MB 8c271850e1f9
config.jsonConfiguration145 B —
README.mdDocumentation2.7 KB —
.gitattributesRepository1.5 KB —

License and Download

License
apache-2.0
Access
Open weights, no gate
Download size
155.8 MB
Download from Autogluon

Released by Autogluon through its official repository on Hugging Face. Read the license.

Built From

  • Described by arXiv:2003.06505
  • Described by arXiv:2207.01848
  • Described by arXiv:2405.13396

Memory Requirements

PrecisionWeights in memory
As published155.8 MB
16-bit0.1 GB
8-bit0.0 GB
4-bit0.0 GB

Weights only, from the published parameter count; the key-value cache and runtime add to this.

Questions About tabpfn-mix-1.0-classifier

How much GPU memory does tabpfn-mix-1.0-classifier need?

About 0.1 GB at 16-bit and 0 GB at 4-bit: the weights (39M parameters) plus a working margin. A long context needs more.

What is the cheapest GPU to run tabpfn-mix-1.0-classifier on?

At 16-bit, 1x MI300X from $1.85 an hour; at 4-bit, 1x MI300X from $1.85 an hour, at the lowest on-demand prices the SAVRN Index lists.

Can I use tabpfn-mix-1.0-classifier commercially?

Yes. tabpfn-mix-1.0-classifier is released under Apache License 2.0. The Apache License 2.0 is a permissive open-source license. It permits commercial use, modification and redistribution. It requires keeping the license and copyright notices and any NOTICE file, stating significant changes, and it includes an express patent grant from contributors.

Similar Models

Model · Tabular classification

TabSTAR

Alan Arazi

To fit a pretrained TabSTAR model to your own dataset, install the package: Paper: TabSTAR: A Foundation Tabular Model With Semantically Target-Aware Representations

Open weights cc-by-4.0 47M parameters

Model · Tabular classification

FoMo-0D

YuchenShen

This model has been pushed to the Hub using the PytorchModelHubMixin integration

Open weights mit 5M parameters

Model · Tabular classification

mitra-classifier

Autogluon

Mitra classifier is a tabular foundation model that is pre-trained on purely synthetic datasets sampled from a mix of random classifiers. Mitra is based on a 12-layer Transformer of 72 M parameters, pre-trained by incorporating an in-context learning paradigm. To use Mitra classifier, install AutoGluon by running: A minimal example showing how to perform inference using the Mitra classifier: A minimal example showing how to perform fine-tuning using the Mitra classifier: This project is licensed under the Apache-2.0 License. Amazon Science blog: Mitra: Mixed synthetic priors for enhancing tabular foundation models

Open weights apache-2.0 76M parameters

Model · Tabular classification

mitra-classifier-2

Autogluon

Mitra-v2 classifier is a tabular foundation model that is pre-trained on purely synthetic datasets sampled from a mix of random classifiers, including the new Hybrid SCM prior. It is the second generation of the Mitra classifier (autogluon/mitra-classifier), pre-trained with a 10x longer context, three times as many features, and an improved optimizer. On the TabArena and TALENT benchmarks it delivers state-of-the-art accuracy at the level of TabFM and EXAONE Tabular, while surpassing TabPFN-3 by a wide margin. The regression model is at autogluon/mitra-regressor-2, and the inference and fine-tuning code with our evaluation results is at autogluon/mitra-finetune. Mitra-v2 is based on a…

Open weights apache-2.0 76M parameters

Model · Tabular classification

mitra-classifier-1.1

Autogluon

Mitra classifier is a tabular foundation model that is pre-trained on purely synthetic datasets sampled from a mix of random classifiers. Mitra is based on a 12-layer Transformer of 72 M parameters, pre-trained by incorporating an in-context learning paradigm. To use Mitra classifier, install AutoGluon by running: A minimal example showing how to perform inference using the Mitra classifier: A minimal example showing how to perform fine-tuning using the Mitra classifier: This project is licensed under the Apache-2.0 License. Amazon Science blog: Mitra: Mixed synthetic priors for enhancing tabular foundation models

Open weights apache-2.0 76M parameters

Model · Tabular classification

BAAR4-guard

Isai Lab

이 저장소는 2017~2026 전주기 트래픽(CIC-IDS2017, CICDDoS2019, CICIoT2023, 2024~2026 AI Vectors)으로 훈련된 고속 인라인 AI 보안 방화벽 엔진입니다. 1. model.safetensors: PyTorch SafeTensors 가중치 2. model.onnx: WebAssembly 및 에지 런타임용 고속 ONNX 그래프 3. modelguardultrafp16.baar3: Zero-Copy Direct-mmap FP16 초고속 커널 바이너리 4. config.json: Robust 스케일러 및 제로 트러스트 임계치 메타데이터

Open weights apache-2.0 644,381 parameters