Senior Applied ML Engineer | Kimchi (Model Router)

Kimchi; Deutschland; Remote; EUR 90.000 - 150.000; Vollzeit

Benefits dieser Stelle

Equity options

Learning budget

Flexible remote work

Hackathons

Zusammenfassung

Cast AI, a remote-first global company, seeks an applied ML Platform Engineer to own the routing model end to end. You will define features, objectives, training, calibration, and release cadence, treating it like a product rather than a research artifact.

You will build production-parity benchmarks, run router-vs-router comparisons, and turn fleet data into training signals with careful error-bar analyses.

Qualifikationen

• 5+ years in applied ML or ML research with real-time decision models.
• Strong Python and SQL at scale; ability to pull and interrogate large datasets.
• Understanding of serving realities (TTFT, decode speed, batching, concurrency) and evaluation methods.

Aufgaben

• Own the routing model end to end: features, objective, training, calibration, and release cadence.
• Build production-parity benchmarks and router comparisons; use holdouts by task family with error bars.
• Turn fleet data into training signals with clear failure categorization to avoid poisoned labels.
• Model the user-experienced objective: task success × token cost × wall-clock speed under concurrency.
• Run online loops: shadow, A/B, full rollout, drift monitoring, retrain.
• Push frontier with calibrated decisions and speed-aware routing; collaborate with platform engineers on instrumentation.
• Support experiments and data paths for per-request features and labels.

Kenntnisse

Applied ML

Model deployment

Python

SQL

Data analysis

Experimentation

Tools

PyTorch

ClickHouse

GitLab CI

Jobbeschreibung

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.