OLIX · 8 hours ago
Senior Compiler Engineer
San Francisco, CA
Full-time
Onsite
Senior Level
$233K/yr - $314K/yr
5+ years expOLIX is a company focused on building the next paradigm in AI hardware with their Optical Tensor Processing Unit (OTPU). The Senior Compiler Engineer will be responsible for developing the ML backend for the OPTUs, ensuring seamless integration with major ML frameworks and optimizing performance through collaboration with hardware teams.
Semiconductors
Responsibilities
Own and deliver projects on your team's roadmap that unlock key high-impact technical and business milestones that drive the success of Olix
Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware
Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximise throughput
Lead code reviews, coach peers, and uphold best practices in ML backend and performance engineering for your team
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
Preferred
Personal projects are a key differentiating factor and hold more weight than other requirements
Benefits
Meaningful stock options.
$36k annual Living-Local Bonus if your residence is within 20 minutes of the office.
33 days of paid time off (PTO), including US federal holidays.
Multiple high-quality medical plan options, including family coverage.
A high-deductible medical option with a company-funded health savings account (HSA).
Dental and vision coverage.
Life insurance, plus short- and long-term disability.
Mental health resources, fertility and family-building support.
Pre-tax commuter and parking benefits.
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.
M4 Macs come as standard, with M4 Pro upgrades for our engineering team.
High-spec noise-cancelling headphones and a fully ergonomic workstation designed for deep focus.
Access to our high-performance 3D printing lab for work, experimentation, and personal creative projects.
Chef-prepared meals: if you need to work late.
We offer full UK and international visa sponsorship.
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