sbi icon indicating copy to clipboard operation
sbi copied to clipboard

Add NPE-PFN

Open jsvetter opened this issue 2 months ago • 5 comments

🚀 Feature Request

Add NPE-PFN to the library (in the same way as autogluon)

NPE-PFN is easy-to-use and has strong performance, particularly for small simulation budgets. But currently it is based on TabPFNv2, which uses a special license. AFAIK, the initial decision was to solve this via an external repo.

However, I recently dabbled with autogluon (developed by Amazon), which also uses the Apache 2.0 license. To conform to the special prior labs license, they simply log once (via a global variable) the necessary attribution line. See the code here The same could be done in the SBI library.

What do you think?

Describe alternatives you've considered

Don't add NPE-PFN (i.e., just keep it in an external repo), or use a different tabular foundation model.

jsvetter avatar Oct 27 '25 08:10 jsvetter

I think this sounds reasonable to me, and if amazon does it then it really should be ok.

michaeldeistler avatar Oct 27 '25 09:10 michaeldeistler

I agree, it's that easy, it would be great to include NPE-PFN. We should still double check whether this is legally bulletproof.

janfb avatar Oct 27 '25 09:10 janfb

I like the idea a lot! Thanks for checking that julius!

dgedon avatar Oct 27 '25 15:10 dgedon

@dgedon or @jsvetter , would you be up for making a PR?

janfb avatar Oct 27 '25 17:10 janfb

Yes, absolutely!

jsvetter avatar Oct 28 '25 06:10 jsvetter