About the job
About OLIX
At OLIX, we are at the forefront of revolutionizing AI technology. As the demand for artificial intelligence accelerates, we face unprecedented challenges in infrastructure development. Traditional hardware designs are no longer sufficient. We are pioneering a transformative approach with our Optical Tensor Processing Unit (OTPU), which sets new benchmarks in performance and energy efficiency, paving the way for the future of computing.
The Role
We are looking for experienced Senior Compiler Engineers to join our innovative team. You will be instrumental in developing the machine learning backend (compiler, runtime, and debugger) for our state-of-the-art OPTUs, ensuring seamless integration with frameworks like PyTorch, TensorFlow, JAX, and MXNet. Your goal will be to enhance the performance capabilities of large-scale AI models while collaborating closely with hardware and modeling teams to optimize device trade-offs.
Key Responsibilities
Project Ownership: Lead and execute projects that are critical to achieving significant technical and business milestones for OLIX.
Architect & Build: Design and implement cutting-edge compiler, runtime, and debugging solutions tailored for custom hardware.
Optimize Performance: Utilize advanced techniques such as layout, fusion, scheduling, and tiling to overcome performance barriers and maximize throughput.
Mentor & Uphold Standards: Conduct code reviews, mentor fellow engineers, and champion best practices in ML backend and performance engineering.
Guide Team Direction: Stay informed about the latest trends in GPU, AI accelerators, and optical computing; innovate and prototype new solutions.
Qualifications & Experience
5+ years of software engineering experience, specializing in C/C++ programming.
Extensive knowledge of machine learning frameworks, compiler design, low-level programming, and optimization methodologies.
Proven expertise in optimizing deep learning models with TensorFlow, PyTorch, or JAX.
Familiarity with custom hardware design and implementation.

