ML Tech Lead

The company:

Everything runs on a single Rails app and one Postgres database, deployed on Heroku. Profitable, no outside investors, and no plans to take any. We grow when customers pay us. Founded in 2023, we now have 29 employees.

The role:

We are creating this position because the workload has grown. The Fraud team trains and serves models that run on every customer request. We need someone who cares about latency as much as accuracy. We work hybrid out of Munich, usually three days in, two days remote.

What you'll be doing:

• Document model cards, data sources and known failure modes.
• Productionize data science prototypes without rewriting everything.
• Set up evaluation harnesses and regression tests for model quality.
• Write clean, tested Python and review teammates' code.
• Orchestrate pipelines with Kubeflow, SageMaker Pipelines, Vertex AI or Airflow.

You'll need:

• 7+ years of software engineering experience, with at least two spent on ML systems in production.
• Comfortable with SQL and large-scale data processing in Spark or Ray.
• Hands-on experience with PyTorch, TensorFlow or JAX.
• Knowledge of LLM APIs, prompt design and evaluation methods.
• BS or MS in computer science, engineering or a related field.
• Experience deploying models as services and monitoring them.

Desirable:

• Recommendation or ranking systems at scale.

Pay and benefits:

Compensation: €136,000 to €158,000/year

• Company pension scheme.
• Holiday pay.
• Learning budget.
• Public transport ticket subsidy.
• JobRad bike leasing.
• Meal allowance.
• Flexible working hours.

Applications are reviewed weekly.

Start: Immediately.

Demo job created on 11jobs for testing and demonstrations. It is not an actual opening at KW Data GmbH.