PriorLabs/TabPFN v6.0.0 on GitHub

What's Changed

Bump actions/checkout from 4.2.2 to 5.0.0 by @dependabot[bot] in #454
Bump astral-sh/setup-uv from 6.4.3 to 6.5.0 by @dependabot[bot] in #456
Bump astral-sh/ruff-action from 3.5.0 to 3.5.1 by @dependabot[bot] in #455
Finetuning rergression debugging - Renaming bar distributions by @rosenyu304 in #410
Fix infer cat nans by @noahho in #432
Improve column selection in RemoveEmptyFeaturesEncoderStep by @benraha in #468
Ensure tests parameters are in deterministic order. by @oscarkey in #472
Rename public methods to remove "_" prefix. And fix typo in method name. by @oscarkey in #476
Bump astral-sh/setup-uv from 6.5.0 to 6.6.0 by @dependabot[bot] in #467
Use generated data instead of "California housing" in tests. by @oscarkey in #481
Declare joblib dependency. by @oscarkey in #477
Handle arbitrary CI environment variable values by @LeoGrin in #483
Add support for uneven splits for large data by @anuragg1209 in #494
Forward cache_trainset_representation to load_model. by @priorphil in #492
Add note to softmax_temperature for the classifier by @brendan-priorlabs in #497
Add consistency checks for the low_memory and fit_with_cache modes. by @oscarkey in #498
Bump actions/setup-python from 5.6.0 to 6.0.0 by @dependabot[bot] in #489
Bump astral-sh/setup-uv from 6.6.0 to 6.6.1 by @dependabot[bot] in #490
Set up methods for anonymous usage analytics by @safaricd in #499
QKV calculation improvements in attention mechanism by @benraha in #488
Select multiple gpus for inference, if available. Don't use them yet. by @oscarkey in #496
Evaluate the estimators in parallel. by @oscarkey in #484
Increase versioning for TabPFN by @safaricd in #504
refactor preprocessors by @bejaeger in #503
Update classifier model file link in README by @anuragg1209 in #500
Increase versioning from 2.1.4 to 2.2.0 by @safaricd in #505
Update README to test workflow by @bejaeger in #511
Add TELEMETRY.md file by @safaricd in #512
Resolved issue saving & loading the predictor (Regressor/Classifier) model on different devices (cuda / cpu) by @MagnusBuehler in #486
If device="auto" and multiple GPUs are present, only select the first. by @oscarkey in #517
Increase tolerance for test_normalize_data_basic test by @MagnusBuehler in #515
Fix synchronisation bug in multi-gpu inference. by @oscarkey in #524
Run the tests on merge to main in addition to in PRs. by @oscarkey in #519
relax type for features_per_group by @LeoGrin in #529
Disable all telemetry in pytest by @safaricd in #527
Workflow Image by @klemens-floege in #532
[Resolved Merge Conflicts from PR 312] Remove shell download script and fix fallback URLs by @MagnusBuehler in #520
Add global seed fixture for reproducible tests by @MagnusBuehler in #516
Bump max Sklearn version to < 1.8 by @klemens-floege in #534
Switch to uv's "--resolution" option to install minimum dependency set. by @oscarkey in #535
Re-enable onnx tests. by @oscarkey in #526
Manage CI dependencies in pyproject and specify numpy dependency. by @oscarkey in #541
Use a Generator object to specify the seed for the random embeddings. by @oscarkey in #525
Add CODEOWNERS file for review team assignment by @noahho in #540
Bump mypy from 1.17.1 to 1.18.2 by @dependabot[bot] in #522
Fixed a formatting issue in README.md by @azurechen97 in #542
Fix potential race condition in InferenceEngineOnDemand. by @oscarkey in #543
Bump astral-sh/setup-uv from 6.6.1 to 7.1.0 by @dependabot[bot] in #547
Enable MPS devices on the CI, and fix associated bugs. by @oscarkey in #518
Add a note on creating/linking issues before opening a PR in the PR template. by @priorphil in #548
Update Ruff to 0.14. by @oscarkey in #549
Introduce method that returns raw logits for individual estimators by @bejaeger in #550
Add KV cache example for fast predictions by @noahho in #553
Update README.md by @noahho in #552
Explicitly disable Dynamo onnx exports to fix CI. by @oscarkey in #554
Add an n_preprocessing_jobs option to the classifier and regressor. by @oscarkey in #555
Fix build error on Python 3.9 by @brendan-priorlabs in #562
Bugfix: improve logics for handling regression on constant target values by @liam-sbhoo in #564
Bump astral-sh/setup-uv from 7.1.0 to 7.1.2 by @dependabot[bot] in #566
Fix column reordering in process_text_na_dataframe with nan and missing values due to Columntransformer by @MagnusBuehler in #539
Update pyobjc-framework-metal requirement from <12.0 to <13.0 by @dependabot[bot] in #567
Check for incompatible licenses by @brendan-priorlabs in #561
Fix cuda configuration error. by @oscarkey in #570
Fix precision of output in temperature=1.0 case by @bejaeger in #569
Allow to run Classifier and Regressor with multiple models by @bejaeger in #557
Rename _resolve_model_path to resolve_model_path by @bejaeger in #573
Loading files from UCI instead of openml now for stability by @noahho in #572
Fix KDITransformerWithNaN.fit in case of all-nan column. by @priorphil in #574
Add concurrency control to CI workflow by @noahho in #568
Rename ModelInterfaceConfig to InferenceConfig. by @oscarkey in #575
Load the inference config from the checkpoint, if present. by @oscarkey in #576
Add unit tests for inference config. by @oscarkey in #578
Add kdi transform to deps. by @priorphil in #580
Update huggingface-hub requirement from <1,>=0.0.1 to >=0.0.1,<2 by @dependabot[bot] in #581
Changes to support TabPFN 2.5, once it is released. by @oscarkey in #579
Check that inference_config is actually set when loading. by @priorphil in #586
Introduce option to calibrate temperature and tune decision thresholds by @bejaeger in #584
Remove type restriction on REGRESSION_Y_PREPROCESS_TRANSFORMS. by @oscarkey in #587
More robust performance test by @bejaeger in #588
Update 'balance probabilities test' and 'get_embeddings' test by @bejaeger in #589
Fix seed for random embeddings on CPU & MPS by @bejaeger in #592
Changed OrderPreservingColumnTransformer back to ColumnTransformer by @magnus-priorlabs in #596
Increase the default and v2.5 pretraining limits. by @oscarkey in #597
Update the device doc string to include multi-gpu inference. by @oscarkey in #599
Parse model version from model_path by @brendan-priorlabs in #585
Forward HF errors as they explain gating / missing tokens nicely. by @priorphil in #600
Set ensure_y_numeric=True for TabPFNRegressor by @safaricd in #601
update pretraining limits docs by @bejaeger in #598
Add additional classifier checkpoint filename by @LeoGrin in #603
Update classifier consistency test values due to v2.5 checkpoint update. by @oscarkey in #604
Add factory methods to create particular versions of the model. by @oscarkey in #602
Adapt README.md for V2.5 by @priorphil in #606
update to 6.0.0 for TabPFN-2.5 release by @LeoGrin in #607

New Contributors

@MagnusBuehler made their first contribution in #486
@azurechen97 made their first contribution in #542
@liam-sbhoo made their first contribution in #564
@magnus-priorlabs made their first contribution in #596

Full Changelog: v2.1.3...v6.0.0