Senior Software Engineer (ML QA)

United States Digital Space LLC; Cambridge; On-site; GBP 50,000 - 75,000; Full time

Benefits offered by this job

Unlimited annual leave

Up to 5% matched pension

Phantom equity

Flexible working

On-site barista

Health cash plan

Income protection

Life assurance

Job summary

United States Digital Space LLC in Cambridge is looking for a skilled engineer to validate and enhance its machine learning software stack. This role focuses on building test infrastructures and automation necessary for AI compute, providing an opportunity to work at the forefront of technology.

The ideal candidate will possess robust software design skills, significant experience in production Python, and familiarity with CI/CD processes. Benefits include unlimited annual leave, a matched pension plan, and a collaborative office environment.

Qualifications

• Experience building maintainable systems across large codebases.
• Testable, reliable Python code used in production environments.
• Automated testing in Linux-based development workflows.

Responsibilities

• Build test infrastructure and automation for validating ML systems.
• Shape test architecture across large codebases and CI pipelines.
• Mentor engineers and improve technical practices.

Skills

Strong software design skills

Production Python experience

CI/CD experience

Familiarity with C or C++

Understanding of ML frameworks

Mentoring and ownership ability

Job description

About the job

Help prove the ML software stack that future AI systems will depend on. the company is expanding the software systems behind AI compute at datacenter scale. This role focuses on validating a complex machine learning stack before it reaches customers and hardware.You will build the test infrastructure, automation and tooling that engineers trust every day. Your work will help raise reliability, performance and maintainability across frameworks, simulators, emulators and future hardware. You will shape test architecture across large codebases, CI pipelines and representative ML workloads. You will use strong software engineering judgement to turn complex technical goals into scalable validation systems.

This is a role for someone who enjoys depth, ownership and useful engineering. You will make quality easier, faster and more visible for teams building at the edge of AI compute.

The team and culture

The ML QA team sits where the company's software components come together. Engineers work across unit tests, integration tests, full model runs, subgraph validation and benchmarking.

Work is organised through focused squads with clear ownership. People are expected to step up, speak openly and improve how the team builds, tests and learns.

Decisions are grounded in evidence from code, data and system behaviour. The team works closely with software teams to identify gaps, improve tooling and move priorities forward quickly.

What we're looking for

• Strong software design skills, with experience building maintainable systems across large codebases.
• Production Python experience, including testable, reliable code used by other engineers.
• Experience with CI/CD, automated testing and Linux based development workflows.
• Familiarity with C or C++, with confidence reading and debugging low level code.
• Understanding of ML frameworks, workloads or model execution beyond simple API usage.
• Ability to mentor engineers, improve practices and take ownership of complex technical problems.

Benefits

• Unlimited annual leave
• Up to 5% matched pension
• Phantom equity – share in the company's success
• True flexibility in how and where you work
• Office spaces designed for collaboration
• Free food and an on-site barista
• Health cash plan
• Income protection
• Life assurance
• Along with other benefits you can choose from (private medical insurance, dental plan etc)

We welcome people from all backgrounds and experiences and are committed to building an inclusive environment where everyone can do their best work.

We're an equal opportunity employer and recognise that everyone brings different strengths and perspectives. If you need any adjustments during the interview process, just let us know - we're happy to support you.