companyOLIX logo

Senior Compiler Engineer

OLIXLondon
On-site Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Senior

Qualifications

QualificationsA minimum of 5 years of experience in software engineering, predominantly in C/C++ programming. In-depth knowledge of ML framework internals, compiler design, low-level programming, and optimization strategies. Extensive experience optimizing deep learning models using TensorFlow, PyTorch, or JAX. Proficiency in multiple tools and methodologies for compiler development and performance enhancement.

About the job

About OLIX

At OLIX, we are at the forefront of revolutionizing AI technology. As the demand for AI continues to skyrocket, the existing infrastructure is struggling to keep pace. Our innovative Optical Tensor Processing Unit (OTPU) is set to redefine performance and energy efficiency, paving the way for groundbreaking advancements in AI hardware.


The Role

We are seeking experienced Senior Compiler Engineers to join our dynamic team focused on developing the machine learning backend (compiler, runtime, and debugger) for our state-of-the-art OPTUs. You will play a crucial role in integrating major ML frameworks such as PyTorch, TensorFlow, JAX, and MXNet with our custom low-level kernel drivers. Your mission will be to ensure optimum performance for a diverse ecosystem of large AI models, collaborating closely with hardware and modeling teams to navigate device trade-offs.

Key Responsibilities

  • Project Ownership: Take charge of and deliver projects within your team's roadmap that are critical to achieving impactful technical and business objectives for OLIX.

  • Architect & Build: Design and implement the compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet tailored for our custom hardware.

  • Optimize Performance: Utilize advanced optimization techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and elevate throughput.

  • Mentor & Uphold Standards: Conduct code reviews, guide peers, and promote best practices in ML backend and performance engineering within your team.

  • Guide Team Direction: Stay ahead of emerging trends in GPU, AI accelerators, and optical computing; propose and prototype innovative solutions.

Qualifications

  • A minimum of 5 years of experience in software engineering, predominantly in C/C++ programming.

  • In-depth knowledge of ML framework internals, compiler design, low-level programming, and optimization strategies.

  • Extensive experience optimizing deep learning models using TensorFlow, PyTorch, or JAX.

  • Proficiency in multiple tools and methodologies for compiler development and performance enhancement.

About OLIX

OLIX is pioneering the future of AI technology, addressing the critical infrastructure gap in the industry with our cutting-edge Optical Tensor Processing Unit (OTPU). We are committed to building a more efficient and powerful hardware ecosystem that meets the ever-growing demands of artificial intelligence.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.