Pro members applied to this job 36 hours before you saw itGet Pro ›

olix - Senior Compiler Engineer

London, UK+ Equity4d ago

In Office Senior EMEA Artificial Intelligence Telecommunications Senior Software Engineer TensorFlow C++Coaching PyTorch JAX

Requirements

• 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.

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 Olix. • 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 trends; propose and prototype innovations.

Benefits

• Competitive Salary: Commensurate with your experience, skills, and location • Equity & Ownership: Meaningful stock options. You’re not just joining the mission; you’re owning a piece of it • Proximity Bonus: We value your time. To minimise your commute and maximise your life, we offer an annual Living-Local Bonus if your residence is within 20 minutes of the office • Retirement Benefits: Employer-contributed retirement plans to help you build long-term financial security. • Due to U.S. export control regulations, candidates’ eligibility to work at OLIX 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.