Senior Compiler Engineer
Job description
Senior Compiler Engineer building the ML backend (compiler, runtime, and debugger) for OLIX's next-generation AI accelerator hardware, connecting frameworks like PyTorch, TensorFlow, JAX, and MXNet down to low-level kernel drivers.
At a glance
- Work mode
- Remote · London, United Kingdom
What you'll do
- Own and deliver projects on your team's roadmap that unlock key high-impact technical and business milestones that drive the success of Olix
- Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware
- Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximise throughput
- Lead code reviews, coach peers, and uphold best practices in ML backend and performance engineering for your team
- Stay ahead of GPU, AI accelerator trends; propose and prototype innovations
What we're looking for
- 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
Nice to have
- Personal projects are a key differentiating factor and hold more weight than other requirements
Benefits
- Competitive Salary: Commensurate with your experience, skills, and location
- Equity & Ownership: Meaningful stock options
- Proximity Bonus: annual Living-Local Bonus if your residence is within 20 minutes of the office
- Retirement Benefits: Employer-contributed retirement plans
Apply
Direct apply at Olix
Opens the job posting at Olix
About Olix
OLIX is building a new paradigm for AI hardware. The OLIX Decode Accelerator 1 (DX-1) is the first accelerator architected specifically for decode, with rack-scale co-design of logic, data movement, packaging, optics and interconnect enabling a step change in system level performance.
Mission
AI is growing faster than any technology in history and the explosion in demand has created a massive infrastructure gap; we can no longer build chips or power stations fast enough to keep up.
Key facts
- OLIX Decode Accelerator 1 (DX-1) is the first accelerator architected specifically for decode
- Rack-scale co-design of logic, data movement, packaging, optics and interconnect