DatabricksCyrus Cowley, Tejas Sundaresan, Hanlin Tang, Anshul Gupta, Julia PowellTutorial
Running open-Jev in SQL on Databricks
Summary
System One decision models such as Jev produce well-calibrated decisions from discrete options and are fast and cheap for large-scale applications, with potential use cases ranging from analyzing customer support transcripts to applying complex decisions at scale. Open-weight versions like SemIf-OpenJev are available, and Databricks enables serving them on governed data using serverless GPUs via AI Runtime, which provides on-demand GPU compute without managing infrastructure. The workflow is packaged in an importable notebook that downloads the model files, logs and registers the model using Express Deployments, and creates a GPU Model Serving endpoint automatically. The endpoint can be called from SQL via ai_query, which supports arbitrary custom model APIs, allowing structured request formats directly from SQL to return selected classifications and option probabilities as structured columns. The example classifies hotel reviews as good or bad, and users can import the notebook, select Serverless GPU, and run all to get started.
Context
Applying fast, cheap decision models to large governed datasets requires serverless GPU infrastructure and SQL integration.
Approach / What changed
Use Databricks AI Runtime to serve open-weight decision models on serverless GPUs, register them with Express Deployments, and call them from SQL via ai_query, which supports custom model APIs for structured requests.
Takeaways
“System One” decision models like Jev are fast, cheap, and produce well-calibrated decisions from discrete options, suitable for large-scale data applications.
Databricks AI Runtime provides serverless, on-demand GPU compute without requiring users to manage GPU infrastructure.
The ai_query function in Databricks supports arbitrary custom model APIs, allowing structured request formats directly from SQL to return classifications and probabilities as structured columns.
FAQ
How do I serve an open-weight decision model on Databricks?
Import the provided notebook, select Serverless GPU, and click Run All. The notebook downloads the model, registers it using Express Deployments, and creates a GPU Model Serving endpoint automatically.
Can I call the model from SQL?
Yes, using Databricks ai_query, which supports custom model APIs. You can send structured request formats directly from SQL and get back selected classifications and option probabilities as structured columns.
What hardware is required?
The setup uses serverless GPUs via Databricks AI Runtime, which provides on-demand GPU compute without requiring you to set up or manage GPU infrastructure.
Source and freshness
- Source
- databricks.com
- Published
- Added to Yomu
Summaries are generated from the original by a language model. Check the original before relying on it.
Keep reading
Related reading
The Good Stuff
Paid picks we use ourselves. Links carry a referral code — a subscription through one earns us a commission at no extra cost to you.
The best engineering reads, once a week
One useful email. Unsubscribe anytime.