Staff Engineer — ML Validation & Benchmarking

EngineersOfAI; Bristol; Hybrid; GBP 60,000 - 85,000; Full time

Benefits offered by this job

Unlimited annual leave

Up to 5% matched pension

Phantom equity

True flexibility in work location

Free food and an on-site barista

Health cash plan

Income protection

Life assurance

Job summary

EngineersOfAI in Bristol is looking for a dedicated Machine Learning Validation Engineer to verify the ML stack that translates accelerator hardware into dependable AI performance. In this role, you will ensure ML models function reliably across various frameworks and execution environments while designing intricate benchmarks.

The ideal candidate possesses extensive ML engineering experience, strong Python skills, and hands-on knowledge of ML frameworks like PyTorch and TensorFlow. Benefits include unlimited annual leave, matched pension, equity opportunities, and more.

Qualifications

• Strong experience in Machine Learning or ML-adjacent software engineering roles.
• A solid grasp of neural networks, training, inference, numerical precision and performance trade-offs.
• Hands-on experience with PyTorch, TensorFlow, JAX, Triton, or similar ML frameworks and tools.
• Strong Python skills for automation, experimentation, benchmarking and reporting.
• Experience designing, running and analysing ML benchmarks or model experiments.
• Confident debugging skills in Linux, with curiosity about model behaviour and system performance.

Responsibilities

• Validate the ML stack that turns accelerator hardware into trusted AI performance.
• Test, benchmark and validate complex systems before they reach customers.
• Run open source models and build automated benchmarking pipelines.
• Create targeted tests for low-level ML behaviour.

Skills

Machine Learning experience

Neural networks understanding

Python skills

Debugging in Linux

Experience with ML frameworks

Tools

PyTorch

TensorFlow

JAX

Triton

Job description

EngineersOfAI in Bristol is looking for a dedicated Machine Learning Validation Engineer to verify the ML stack that translates accelerator hardware into dependable AI performance. In this role, you will ensure ML models function reliably across various frameworks and execution environments while designing intricate benchmarks.

The ideal candidate possesses extensive ML engineering experience, strong Python skills, and hands-on knowledge of ML frameworks like PyTorch and TensorFlow. Benefits include unlimited annual leave, matched pension, equity opportunities, and more.