Buildkite logoBuildkite logo

Staff Machine Learning Engineer

BuildkiteANZ Region
Remote Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Experience Level

Senior

Qualifications

Proven experience in machine learning, data science, or a related field. Strong programming skills, particularly in Python or similar languages. Ability to lead complex technical projects and collaborate with cross-functional teams. Familiarity with CI/CD processes and best practices. Excellent problem-solving skills and analytical thinking. Strong communication skills, both verbal and written.

About the job

Buildkite’s Continuous Integration platform supports leading engineering teams worldwide, helping deliver software to over a billion users each day.

Role overview

The Staff Machine Learning Engineer will join the Test Engine team, focusing on the technical direction and hands-on development of machine learning solutions. This role centers on building models and infrastructure for predictive test selection, determining which tests matter most for a given code change. The position is based in the ANZ region.

Staff Engineers at Buildkite act as technical leaders, designing and scaling systems while supporting and mentoring other engineers. In this position, you will be the primary ML expert, guiding the technical approach to test selection and laying the groundwork for broader machine learning efforts across the company.

About the Test Engine team

The Test Engine team helps engineering teams speed up software delivery by improving visibility and control over their test suites. Current features include real-time flaky test detection and management, intelligent test splitting across parallel jobs, and performance analytics that work with any CI/CD platform, not just Buildkite Pipelines.

The Test Engine processes billions of test runs, providing deep insight into test suites, codebases, and their relationships. The next step is to use this data to answer a key question: which tests are most likely to fail when a specific code change is made?

The industry is moving away from running entire test suites for every change. Teams that can shift to a faster, more targeted inner loop, running only the most relevant tests, will deliver value more quickly. For many Buildkite customers, this speed is vital in highly competitive markets where fast feedback cycles make a difference.

What you will do

  • Shape and guide the technical direction for machine learning within the Test Engine team
  • Develop models and infrastructure for predictive test selection based on code changes
  • Act as the lead ML expert, setting frameworks for future machine learning initiatives
  • Collaborate with engineers to design, build, and scale systems that support smarter testing workflows

About Buildkite

Buildkite is at the forefront of innovation in the CI/CD space, empowering engineering teams worldwide to deliver software efficiently and effectively. With a commitment to excellence, Buildkite's platform is designed to streamline the development process, ensuring that teams can focus on what they do best—building great software.

Similar jobs

Browse all companies, explore by city & role, or SEO search pages.

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.