Overview
Causilo is a tabular foundation model for classification and regression, pretrained on about 36 million synthetic tables. It takes the rows you already know the answer for as context and predicts the rest in a single call, with no per-task training run or hyperparameter search. Missing values and categorical columns are accepted as they are. It suits research, benchmarking and testing on tabular data without building a training pipeline.
Validation: 95.1% accuracy on 1,000 held-out rows of the Covertype dataset (UCI) with 50,000 context rows, and R-squared 0.878 on California housing with 10,000 context rows, with no training or tuning. The sample notebook reproduces both figures. Latency on ml.g6e.xlarge for a request of 1,000 query rows with 54 feature columns: about 0.7 s with 1,000 context rows and about 10 s with 50,000.
In the Causilo Technical Report (https://arxiv.org/abs/2609.22866 ), Causilo reaches 1785 Elo on TabArena (51 datasets) at a median 0.10 seconds per 1,000 test rows under the benchmark's timing protocol, which places it on the performance-efficiency Pareto frontier. Its row stages attend through a fixed number of summary tokens, so their attention cost grows linearly with the number of columns.
Known limitations: when training and test rows come from different groups (grouped splits), Causilo ranks behind several conventional supervised models. Not intended for medical diagnosis.
This package runs Causilo inside your own AWS account as a SageMaker real-time endpoint or batch transform job, with network isolation: your data stays in your account and the container makes no outbound calls.
Use is governed by the Causilo License v1.0, the end user license agreement for this listing: non-commercial research, testing and evaluation. Commercial or production use of Causilo or its outputs, or offering it as a hosted or API service, paid or free, needs a separate license from Nums AI Inc.; contact api@nums.world .
Highlights
- No training run: known rows in, predictions out, in a single call.
- On the TabArena performance-efficiency Pareto frontier: 1785 Elo at a median 0.10 s per 1,000 test rows (Causilo Technical Report, 2026).
- Runs entirely inside your AWS account; the container makes no outbound calls.
Details
Introducing multi-product solutions
You can now purchase comprehensive solutions tailored to use cases and industries.
Features and programs
Financing for AWS Marketplace purchases
Pricing
Vendor refund policy
Causilo is free on AWS Marketplace: there is no software charge, so there is nothing to refund. SageMaker infrastructure charges are billed by AWS and are not refunded by Nums AI. For questions, contact api@nums.world .
How can we make this page better?
Legal
Vendor terms and conditions
Content disclaimer
Delivery details
Amazon SageMaker model
An Amazon SageMaker model package is a pre-trained machine learning model ready to use without additional training. Use the model package to create a model on Amazon SageMaker for real-time inference or batch processing. Amazon SageMaker is a fully managed platform for building, training, and deploying machine learning models at scale.
Version release notes
Initial release of Causilo for Amazon SageMaker.
Additional details
Inputs
- Summary
One Parquet file per request. Rows with the target filled in are the context; rows with the target left blank are predicted, and predictions come back in the order those rows appear. Parameters go in the Parquet file's key-value metadata under "causilo-params" as JSON: model ("causilo-clf" for classification or "causilo-reg" for regression), target (the name of the target column), output_type, and for output_type "quantiles" a quantiles list such as [0.1, 0.5, 0.9]. Content type: application/vnd.apache.parquet.
- Limitations for input type
- A real-time request is limited to 6 MB and 60 seconds; use batch transform, up to 100 MB per file, for larger tables. Classification accepts at most 64 distinct target values. Date, time, duration, decimal and nested columns, infinite values, and columns that are blank on every query row are refused; convert them first. Data contract and error codes: https://docs.nums.world/reference (it documents the hosted API; its quotas and size limits do not apply to SageMaker).
- Input MIME type
- application/vnd.apache.parquet, application/x-parquet, application/octet-stream
Support
Vendor support
Email api@nums.world . We respond within two business days (KST). The data contract, parameters and error codes are documented at docs.nums.world.
AWS infrastructure support
AWS Support is a one-on-one, fast-response support channel that is staffed 24x7x365 with experienced and technical support engineers. The service helps customers of all sizes and technical abilities to successfully utilize the products and features provided by Amazon Web Services.