Senior Applied ML Engineer | Kimchi (Model Router)

Cast AI; Osterreich; Remote; PEN 412,000 - 618,000; Full time

Benefits offered by this job

Equity options

Remote-first international team

Learning budget

Annual hackathon

Job summary

Cast AI is seeking an experienced ML/Platform engineer to own the routing model end-to-end. You will define features, objectives, training, calibration, and release cadence while building robust production-grade evaluations that compare router performance and cost across models.

You will transform fleet and benchmark data into actionable training signals, manage the online experimentation loop, and push toward calibrated, speed-aware routing with abstention under uncertainty.

Qualifications

• 5+ years in applied ML or ML research with real-time decision models (ranking, pricing, routing).
• Strong Python and SQL at scale; work with millions of request-level records.
• Understand differences between model improvement and evaluation difficulty.
• Have opinions on what LLM benchmarks measure; know serving realities (TTFT, decode speed, batching).
• Instrument before optimizing; use error bars and production-parity evaluation.

Responsibilities

• Own the routing model end to end: features, objective, training, calibration, release cadence.
• Build the evaluation that gates changes: production-parity benchmarks, holdouts by task family, error bars.
• Turn fleet data into training signals from production and benchmark rollouts with labeled feedback.
• Model the user experience: task success × token cost × wall-clock speed under real concurrency.
• Run the online loop: shadow, A/B, full rollout, drift monitoring, retrain.
• Push calibrated decisions, abstention under uncertainty, and speed-aware routing.
• Partner with platform engineers on instrumentation and data-path requirements.

Skills

Applied ML

Python

SQL at scale

Model evaluation

LLM benchmarks

Serving reality

Instrumentation

Problem solving

Tools

ClickHouse

GitLab CI

Job description

Why Kimchi?

Great models are raw ingredients. Kimchi is what happens when you ferment them with good engineering. We started inside CAST AI helping companies run LLMs on their own Kubernetes clusters; today we run our own coding harness on any model you pick, and the router deciding which model answers each request is ours too.

The Auto Router

Our promise is one enforcement layer for your AI: budgets, spend attribution, and model routing, applied before a request ever hits a model. The Auto Router makes the call: for every request, which model serves it. Eight open-source models whose input prices span more than 20×, cached input up to 50× cheaper than fresh, frontier providers on customers' own keys, and a coding agent that fires hundreds of requests before its task is done. Frontier-level task success at a fraction of the token cost: that sentence is the product, and it stays true only if the model inside the router keeps getting better. That model is yours.

Made by Cast AI – a Miami-headquartered, globally distributed team across 34 countries that hit unicorn status in January 2026 with a $50B+ backed investment pushing its valuation past $1 billion.

About The Role

Every request gets a decision: which model serves it.

This is an applied-ML seat, not an infrastructure one. Platform engineers own the proxy, the streaming path, and the fleet. You own the model inside them: the training data, the objective, the evaluation, and the iteration loop that ships a better router every cycle.

The problem:

routing is a prediction problem wearing a product's clothes. For every request you predict the quality each candidate model would deliver, what it would cost, and how fast it would decode, and you have to be right enough, in aggregate, that the savings are real and the quality never dips below frontier. The data is messy: rollout outcomes from benchmark harnesses, per-request records from the live fleet, override signals, and token streams whose speed depends on the serving concurrency at the moment. Speed itself is a correctness variable: on deadline-bound agentic work, a model that decodes three times slower doesn't do the same work slower, it does less work. That's why naive cost-per-token objectives quietly optimize for the wrong thing. And the bar for evidence is high: a router change ships only after production-parity evaluation with error bars, then gets held to its numbers in live traffic.

Stack:

Python; PyTorch; SQL at scale (ClickHouse); OpenAI- and Anthropic-compatible LLM APIs; benchmark and rollout harnesses (RouterBench-class); GitLab CI.

Requirements

• You've spent 5+ years in applied ML or ML research, and you've trained, evaluated, and shipped models that make real-time decisions (ranking, recommendation, bandits, pricing, or LLM routing) and you can tell that story with receipts.
• Strong Python and SQL at scale: you can pull, clean, and interrogate millions of request-level records without waiting on anyone.
• You know the difference between a model that got better and an eval that got easier.
• You have real opinions about what LLM benchmarks measure, agentic harnesses included.
• You know enough serving reality (TTFT, decode speed, batching, prefix caching, concurrency) to model the objective honestly.
• You instrument before you optimize, error bars and all.
• A wide mandate excites you more than it unsettles you.

Nice to haves

• You've worked on LLM routing specifically: RouterBench-class evals, cost/quality/latency Pareto fronts.
• Calibration, abstention, or conformal methods in production decision systems.
• You've run online A/B loops with drift monitoring on a model that touches real money.

Responsibilities

• Own the routing model end to end: features, objective, training, calibration, and release cadence. Iterate it like a product, not a research artifact.
• Build the evaluation that gates every change: production-parity benchmarks, router-vs-router comparisons, holdouts by task family, error bars on pass rates. Nothing ships on vibes.
• Turn fleet data into training signal: per-request outcomes from production, benchmark rollouts, and override feedback, with failure categorization (infra vs task) so poisoned labels never reach the policy.
• Model the objective the user actually experiences: task success × token cost × wall-clock speed under the real concurrency regime, with cached-input economics in the picture.
• Run the online loop: shadow, A/B, full rollout, drift monitoring, retrain. When production disagrees with your offline eval, you find out why before the next ship.
• Push the frontier where it pays: calibrated decisions, abstention under uncertainty, speed-aware routing. We experiment on what a router should be, not just which model it picks today.
• Partner with platform engineers on instrumentation: the features and labels recorded per request (decode speed first-class among them) and the experiments the data path must support.

What’s in it for you?

• Competitive salary (depending on the level of experience).
• Enjoy a flexible, remote-first global environment.
• Collaborate with a global team of cloud experts and innovators, passionate about pushing the boundaries of Kubernetes technology
• Equity options.
• Get quick feedback with a fast-paced workflow. Most feature projects are completed in 1 to 4 weeks.
• Spend 10% of your work time on personal projects or self-improvement.
• Learning budget for professional and personal development - including access to international conferences and courses that elevate your skills.
• Annual hackathon to spark new ideas and strengthen team bonds.
• Team-building budget and company events to connect with your colleagues.
• Equipment budget to ensure you have everything you need.
• Extra days off to help maintain a healthy work-life balance.

Hiring process

• Screening call with Recruiter
• Hiring Manager interview
• Technical interview (system design)
• Live coding
• Culture Check interview with an executive

As part of our standard hiring process, we would like to inform you that a background check may be conducted at the final stage of recruitment through our third-party provider, Checkr.

Please note that Cast AI does not provide any form of visa sponsorship/work permit.