Applied Machine Learning Engineer
Applied Machine Learning Engineer
Cast AIStraight from Cast AI’s careers page. Apply on the company site — no recruiter, no middleman.
Senior Applied ML Engineer | Kimchi (Model Router)
Location: Austria; France; Germany; Italy; Netherlands; Poland; Spain; United Kingdom
Department: Technology
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 products 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 doesnt do the same work slower, it does less work. Thats 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
- Youve spent 5+ years in applied ML or ML research, and youve 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
- Youve worked on LLM routing specifically: RouterBench-class evals, cost/quality/latency Pareto fronts.
- Calibration, abstention, or conformal methods in production decision systems.
- Youve 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.
## LI-Remote
Similar remote jobs
More like this →
BAE Systems
AI Application Developer

BAE Systems
Principal Engineer, RF Systems Analyst (Remote)

Aviso Wealth
Data Engineer (Temporary)

VON DER HEIDE
Senior Machine Learning Engineer
Discover More than 100,000 Hidden Remote Jobs Before Everyone Else
Unlock All Remote Jobs Today
Simple pricing. Big savings on Quarterly and Yearly.
Monthly Access
- Instant access to fresh remote jobs from 500+ companies
- New opportunities added hourly, often 3-7 days before anywhere else
- Advanced filtering by role type, stack, pay, and location
- Priority customer support
Yearly Access
- Everything in Monthly
- Save $169 (~74%) vs paying monthly
- Average job search takes ~6 months - get covered for the whole journey
- Less than the cost of one lunch per month for competitive advantage
- Equivalent to just ~$4.92/month