Staff ML Engineer - Production ML Platform Lead (Remote)

Buildkite Pty Ltd; Pacific (MO); On-site; USD 140,000 - 220,000; Full time

Job summary

Buildkite is seeking a Senior ML Engineer to own the end-to-end ML for our Test Engine. You will shape strategy, architecture, and models running in production, enabling faster, smarter test selection at scale.

You will work with Python and SQL on AWS, with containerised workloads and data-at-scale tooling like Spark or Flink. shipping and operating ML in production is essential; sponsorship is not offered.

Qualifications

• Proven ability to ship ML models into production and monitor them.
• Strong programming in Python and SQL with production-grade data pipelines.
• Experience with CI/CD and code analysis is a bonus.

Responsibilities

• Own ML in production for Test Engine, from data features to serving layer.
• Train, evaluate, and retrain models; balance accuracy and latency.
• Build the platform that makes future models quick to production and repeatable.

Skills

Python

SQL

AWS

Machine learning in production

Tools

Spark

Flink

Docker

Job description

Buildkite is seeking a Senior ML Engineer to own the end-to-end ML for our Test Engine. You will shape strategy, architecture, and models running in production, enabling faster, smarter test selection at scale.

You will work with Python and SQL on AWS, with containerised workloads and data-at-scale tooling like Spark or Flink. shipping and operating ML in production is essential; sponsorship is not offered.