OLIX · 2 hours ago
Staff Compiler Engineer
San Francisco, CA
Full-time
Onsite
Senior Level, Lead/Staff
$279K/yr - $341K/yr
7+ years expOLIX is a company focused on building the next generation of AI infrastructure with their Optical Tensor Processing Unit (OTPU). They are seeking Staff Compiler Engineers to architect and build the ML backend for their OPTUs, integrating with various machine learning frameworks and optimizing performance.
Semiconductors
Responsibilities
Ownership: Define and deliver the technical vision and roadmap for your team that unlocks key strategic technical and business goals that are essential to the success of Olix
Collaboration: Partner closely with hardware engineers to align compiler, runtime, and debugger requirements with the OTPU design; ensure software and hardware are designed together to deliver maximum performance
Architect & Build: Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware
Optimize Performance: Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximize throughput
Mentor & Define Standards: Lead code reviews, coach peers, and define best practices in ML backend and performance engineering for your team
Guide Technology Direction: Stay ahead of GPU, AI accelerator, and optical computing trends; propose and prototype innovations
Qualification
Required
7+ years of experience in software engineering with a focus on C/C++ programming
Extensive experience in ML framework internals, compilers, low-level programming, and optimisation techniques
Extensive experience optimising Tensorflow, PyTorch or JAX deep learning models
Extensive experience 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.
$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
OLIX is building the infrastructure for Frontier AI. (Formerly operating as Flux)
51-200 employees
Funding
Current Stage
Growth StageCompany data provided by crunchbase