PriorLabs/TabPFN v7.1.0

on GitHub

Added

• More informative Out-Of-Memory error message. (#805)
• Add multi-GPU DDP support for finetuning via torchrun (auto-detected, no code changes needed) (#812)
• Add task_type to forward. (#844)
• Exclude very recent package release in environment (#847)

Changed

• Switch from Hugging Face to Prior Labs website for model license acceptance (#798)
• "auto" device selection now uses all available CUDA GPUs instead of only the first one (#808)
• Optimize fingerprint hashing in preprocessing: round feature matrix once instead of per-row, avoid redundant SHA-256 calls. Speeds up fit by up to 2x for large datasets. (#818)

Fixed

• Fix the pdf() in FullSupportBarDistribution to actually compute the probability density. (#799)
• Fix float overflow in Yeo-Johnson inverse transform that produced inf values and silently degraded regression border resolution. (#838)
• Fix differentiable input for v2.6 (#843)