TabFM is a tabular foundation model from Google Research for zero-shot classification and regression on structured datasets. It is designed to work with mixed numerical and categorical columns without requiring a custom training run for every new table. Instead of fitting model weights to the user’s dataset, TabFM uses in-context learning by reading training examples and test rows together at inference time. The library provides scikit-learn-compatible classifier and regressor interfaces, which makes it familiar for data scientists already using Python ML workflows. It supports both JAX and PyTorch backends and can automatically download pretrained TabFM v1.0.0 weights. The project is useful for practitioners who want strong tabular predictions with less manual feature engineering, tuning, and repeated model training.
Features
- Zero-shot tabular prediction
- Classification and regression support
- Mixed numerical and categorical columns
- Scikit-learn-compatible API
- Automatic pretrained weight loading
- JAX and PyTorch backends