Applied ML Engineer | Kimchi (Model Router)

Cast AI; United States; Remote; USD 180,000 - 240,000; Full time

Benefits offered by this job

Flexible remote environment

Equity options

Learning budget

Annual hackathons

Equipment budget

Job summary

Cast AI, based in Miami with a globally distributed team, is seeking an experienced ML engineer to own the auto-router for LLM workloads. You will design training objectives, implement evaluation gates, and drive a feedback loop from live traffic to continuously improve routing decisions.

You will run production-grade experiments, calibrate models, and collaborate with platform engineers to instrument data paths, ensuring latency-aware, cost-effective routing while maintaining high quality.

Qualifications

• 5+ years in applied ML or ML research with real-time decision models
• Strong Python and SQL at scale with large datasets
• Experience shipping models in production with measurable impact
• Knowledge of LLM benchmarks and routing approaches
• Understanding of serving considerations (TTFT, batching, caching) and instrumentation

Responsibilities

• Own the routing model end to end: features, objective, training, calibration, and release cadence.
• Build evaluation gates: production-parity benchmarks, router-vs-router comparisons, holdouts by task family, error bars on pass rates.
• Turn fleet data into training signal with failure categorization to avoid poisoned labels reaching the policy.
• Model objective to reflect user experience: task success, token cost, and latency under concurrency.
• Run online loops: shadow, A/B, full rollout, drift monitoring, retrain.
• Investigate production offline-eval disagreements before next ship.
• Push frontier with calibrated decisions, abstention, and speed-aware routing.
• Collaborate with platform engineers on instrumentation and data-path support.

Skills

Python

SQL

ML engineering

LLM routing

A/B testing

Observability

Optimization

PyTorch

Tools

ClickHouse

GitLab CI

AWS/GCP

Job description

Задачи:

• 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.

Требования:

• 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.
• 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.

Условия:

• 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.
• 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.

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.