Skip to content

Deploy tabular foundation models to SageMaker endpoints - #273

Merged
shchur merged 12 commits into
masterfrom
tabular-foundation-model-deploy
Sep 18, 2026
Merged

shchur merged 12 commits into
masterfrom
tabular-foundation-model-deploy

Conversation

@shchur

@shchur shchur commented Sep 18, 2026

Copy link
Copy Markdown
Collaborator

Issue #, if available:

Description of changes:

  • implement TabularFoundationModel.deploy using the shared foundation-model deployment path
  • add TabularEndpoint with predict and predict_proba APIs that send both prediction data and labeled train_data
  • add a tabular foundation-model serving handler that caches Hugging Face weights at startup, fits a request-scoped TabularPredictor, and renders predictions/probabilities
  • add serializer support, CPU deployment defaults, public exports, API docs, and unit coverage

Validation:

  • 50 passed in the focused unit suite
  • Ruff and git diff --check pass
  • deployed mitra-classifier with AutoGluon 1.5.0 on ml.m5.4xlarge, invoked predictions and probabilities successfully, and verified endpoint/config/model cleanup

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@shchur
shchur enabled auto-merge (squash) September 18, 2026 15:59
@shchur
shchur disabled auto-merge September 18, 2026 16:20
@shchur
shchur merged commit a70f7ac into master Sep 18, 2026
7 of 8 checks passed
@shchur
shchur deleted the tabular-foundation-model-deploy branch September 18, 2026 16:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant