Senior Compiler Engineer

The role

We're searching for Senior Compiler Engineers to join the team building the ML backend (compiler, run-time, and debugger) for our next-generation 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

Compensation & Benefits

Due to U.S. export control regulations, candidates' eligibility to work at Flux depends on their most recent citizenship or permanent residency status. We are generally unable to consider applicants whose most recent citizenship or permanent residence is in certain restricted countries (currently including Iran, North Korea, Syria, Cuba, Russia, Belarus, China, Hong Kong, Macau, and Venezuela). Applicants who have subsequently obtained citizenship or permanent residency in another country not subject to these restrictions may still be eligible.

We do not accept unsolicited CVs from recruitment agencies, will not be liable for any fees, and prohibit unauthorised use of our company name in recruitment activities.