OLIX · 1 day ago
Compiler Engineer
San Francisco, CA
Full-time
Onsite
Senior Level
$233K/yr - $314K/yr
5+ years expOLIX is pioneering the next generation of AI technology with their Optical Tensor Processing Unit (OTPU). They are seeking a Senior Compiler Engineer to develop the ML backend for their OPTUs, ensuring optimal performance for various AI models by collaborating with hardware and modeling teams.
Semiconductors
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, and optical computing trends; propose and prototype innovations
Qualification
Required
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
Benefits
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 a $36k annual Living-Local Bonus if your residence is within 20 minutes of the office.
Healthcare Coverage: Multiple high-quality medical plan options, including family coverage.
Health Savings Account: A high-deductible medical option with a company-funded health savings account (HSA).
Dental & Vision: Dental and vision coverage.
Additional Coverage: Life insurance, plus short- and long-term disability.
Wellbeing Support: Mental health resources, fertility and family-building support.
Commuter Benefits: Pre-tax commuter and parking benefits.
401(k): Access to a 401(k) retirement plan with a 4% employer match, with both traditional and Roth contribution options. Employees also have access to a dedicated financial advisor to support portfolio selection and long-term planning.
Chef-prepared meals: if you need to work late.
Visa Sponsorship: We hire the best in the world. We offer full UK and international visa sponsorship.
Seamless Relocation: Whether you’re moving across the country or across the globe, our dedicated relocation partner provides funding and concierge support to get you settled.
Company
OLIX
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.
51-200 employees
Funding
Current Stage
Growth StageCompany data provided by crunchbase