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.

