Enable job alerts via email!

Senior ML Engineer

ZipRecruiter

London

On-site

GBP 70,000 - 100,000

Full time

Yesterday
Be an early applicant

Generate a tailored resume in minutes

Land an interview and earn more. Learn more

Start fresh or import an existing resume

Job summary

A leading company focused on AI development is seeking a Senior ML Compiler Engineer in London. This role involves designing and implementing cutting-edge compiler technology, optimizing performance across various AI frameworks. Ideal candidates will have over 5 years of experience in C/C++ and a strong understanding of ML frameworks. Join the team in shaping the future of computing and enjoy a competitive salary, stock options, and generous benefits.

Benefits

Competitive salary and stock options
Comprehensive healthcare insurance
25 days PTO plus bank holidays
Private access to 3D printer

Qualifications

  • 5+ years of experience in software engineering focused on C/C++.
  • Deep expertise in ML framework internals, compilers, and low-level programming.
  • Experience in high-performance computing contexts.

Responsibilities

  • Own and deliver projects in the ML backend area.
  • Design and implement compiler, runtime, and debugger for AI frameworks.
  • Optimise performance using advanced techniques.

Skills

C/C++ programming
ML framework internals
Optimization techniques
Strong problem-solving skills
Collaboration

Education

Bachelor's degree in computer science, electrical engineering, telecom engineering, mathematics, or related field

Tools

LLVM
OpenXLA/XLA
MLIR
TVM

Job description

Job Description

Company Overview

Flux Computing designs and manufactures optical processors to train and run inference on large AI models. Join us in London to be part of a highly motivated and skilled team that thrives on delivering impact and innovation at speed.

The role

We’re searching for Senior ML Compiler Engineers to join the team building the ML backend (compiler, run-time, and debugger) for our next- OPTUs that connect PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers.Your mission will be to create seamless support for a broad ecosystem of large AI models, and ensure we are pushing the limits of their performance envelope by closely partnering with hardware and modelling teams to understand device trade-offs.

Responsibilities

  • Project Ownership: Own and deliver projects on your team's roadmap that unlock key high-impact technical and business milestones that drive the success of Flux.
  • Architect & Build: Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware.
  • Optimise Performance: Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximise throughput.
  • Mentor & Encourage Standards: Lead code reviews, coach peers, and uphold best practices in ML backend and performance engineering for your team.
  • Guide Team Direction: Stay ahead of GPU, AI accelerator, and optical computing trends; propose and prototype innovations.

Skills & Experience

  • 5+ years of experience in software engineering with a focus on C/C++ programming.
  • Deep expertise in ML framework internals, compilers, low-level programming, and optimisation techniques.
  • Deep expertise in optimising Tensorflow, PyTorch or JAX deep learning models.
  • Deep expertise with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM.
  • Practical experience applying machine learning in high-performance computing contexts.
  • Strong problem-solving skills and the ability to think critically and creatively.
  • Experience in high-pace, dynamic work environments.
  • Excellent teamwork and communication skills, with the ability to collaborate effectively with cross-functional teams.
  • Bachelor's degree in computer science, electrical engineering, telecoms engineering, mathematics, or a related field.
  • Personal projects are a key differentiating factor and hold more weight than other requirements.

Compensation & Benefits

  • Competitive salary and stock options in a rapidly growing AI company.
  • Based in our new 5,000 sq. ft. office in the AI hub of Kings Cross, London.
  • To foster collaboration in our high-growth environment, we require all employees to work from our London HQ and live within a 45-minute commute. We offer an extra £24,000/year incentive for those living within 20 minutes.
  • Comprehensive healthcare insurance.
  • 25 days PTO policy plus bank holidays.
  • Private access to our in-house 3D printer.

If you are passionate about pushing the boundaries of what's possible in AI and thrive in a high-energy, fast-paced environment, we want to hear from you. Apply now to join Flux and be a key player in shaping the future of computing.

Get your free, confidential resume review.
or drag and drop a PDF, DOC, DOCX, ODT, or PAGES file up to 5MB.