Download Latest Version v8.2.0 source code.zip (2.5 MB)
Email in envelope

Get an email when there's a new version of TabPFN

Home / v8.2.0
Name Modified Size InfoDownloads / Week
Parent folder
README.md 2026-07-28 2.8 kB
v8.2.0 source code.tar.gz 2026-07-28 2.3 MB
v8.2.0 source code.zip 2026-07-28 2.5 MB
Totals: 3 Items   4.9 MB 1

Added

  • Add examples/input_gradients.py showing how to extract gradients of TabPFN predictions with respect to the input data via differentiable_input=True on the breast-cancer dataset. (#1128)

Changed

  • Improved fine-tuning: retuned the example scripts (Higgs test AUC 0.8247 → 0.8322, California housing test MSE 0.1350 → 0.1328), validation_split_ratio=None/0 now disables validation entirely, best-checkpoint saving now only applies with early_stopping=True, and small remainder data chunks no longer crash the context/query split. (#1101)
  • Run the consistency tests with float64 inference precision and regenerate the reference predictions, reducing floating-point divergence across hardware/BLAS backends. (#1111)
  • Expose the kv_cache_dtype as explicit argument. ([5366f7] @stipsan chore(deps): upgrade web-vitals to v6 (#13682) [ffdf25] squiggler-app[bot] chore(deps): dedupe pnpm-lock.yaml (#13680) [45aed7] @stipsan refactor(test-studio): extract presentation preview into preview-iframe app (#13678) [e590fc] squiggler-app[bot] #1120)
  • inference_precision="auto" now uses bfloat16 autocast on CPUs with native bf16 support (Intel AMX / AVX512-BF16, AMD Zen 4+), giving ~2x faster CPU inference at unchanged accuracy. CPUs without fast bf16 keep running in float32. (#1122)
  • Raised the default CPU sample limit from 1000 to 5000 for the v3 model (other versions keep the 1000 limit). With bf16 autocast on modern CPUs, v3 inference on datasets up to 5000 rows now runs in a reasonable time (~30s at 5000 rows). The "may be slow" warning threshold scales with the limit (1000 for v3, 200 otherwise). The override (ignore_pretraining_limits=True / TABPFN_ALLOW_CPU_LARGE_DATASET=1) is unchanged. (#1123)
  • Enabled autocast also for mps. (#1124)
  • Changed the minimal required version of torch to use MPS to 2.6 (fix(deps): update linters (#13660) [669fc7] squiggler-app[bot] chore(deps): update dependency @sentry/react to ^10.67.0 (#13659) [1c9a40] squiggler-app[bot] chore(deps): update dependency oxlint-tsgolint to v7 (#13661) [fe2793] squiggler-app[bot] fix(deps): update dependency groq-js to v2 (#13677) [7ac6d8]#1129)
  • Speed up preprocessing for large number of features (#1136)
  • Speedup preprocessing for large numeric columns. (#1137)

Fixed

  • Fixed a RuntimeWarning: overflow encountered in cast in SafePowerTransformer's Yeo-Johnson inverse transform, caused by the clip bound being computed in the output dtype instead of float64. ( @stipsan chore(deps): update dependency path-to-regexp to v8 (#13588) [5dda1f] @stipsan chore(build): migrate package builds from pkg-utils to tsdown (#13500) [ee8fdb] @jordanl17 fix(releases): keep copy-to sub-menu on the side at narrow widths (#13673) [548713] @juice49 chore(ailf): add require-field-validation ailf task (#13457) [8f582e]#1105)
  • Fix timing on GPU to reflect full GPU work. (#1113)
  • Fix _repair_borders widening the top bar-distribution border downwards for negative targets, which left the borders non-ascending when a quantile target transform collapsed the top gap. (#1127)
Source: README.md, updated 2026-07-28