Staff Compiler Engineer at Flux

San Francisco, California, United States

Apply Now

$275,000 – $336,000Compensation

Senior (5 to 8 years), Expert & Leadership (9+ years)Experience Level

Full TimeJob Type

UnknownVisa

AI, Machine Learning, Hardware, High-Performance ComputingIndustries

Requirements

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

Responsibilities

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 Flux
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
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 maximize throughput
Lead code reviews, coach peers, and define 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

Skills

Key technologies and capabilities for this role

C++PyTorchTensorFlowJAXMXNetLLVMMLIRTVMOpenXLAXLAcompilersruntimedebuggerML optimizationlow-level programmingkernel drivers

Questions & Answers

Common questions about this position

What is the salary range for the Staff Compiler Engineer position?

The starting salary is $275,000 - $336,000, depending on experience.

Is this role remote or onsite, and what are the location requirements?

This is an onsite role based in the central San Francisco office, requiring all employees to work from the SF office and live within a 45-minute commute, with a $24,000/year incentive for those living within 20 minutes.

What key skills and experience are required for this role?

Candidates need 7+ years in software engineering focused on C/C++, extensive experience in ML framework internals, compilers, optimizing TensorFlow/PyTorch/JAX models, and toolchains like LLVM, MLIR, TVM, plus strong problem-solving and teamwork skills.

What is the work environment and culture like at Flux?

Flux has a high-growth, high-pace, dynamic environment that fosters collaboration, requiring onsite work in SF to support teamwork with hardware and modeling teams.

What makes a strong candidate stand out for this position?

Personal projects are a key differentiating factor and hold more weight than other requirements.

Flux

AI-powered platform for PCB design

About Flux

Flux.ai provides a platform for designing and building printed circuit boards (PCBs) within the electronic design automation (EDA) market. The platform features an AI-powered assistant named Copilot, which helps users streamline the PCB design process, making it more efficient. Users can sign up for free and access basic features, with the option to upgrade to premium subscription plans for additional services. This freemium model allows engineers, designers, and electronics enthusiasts to engage with the platform at their own pace. Flux.ai differentiates itself from competitors by integrating AI assistance directly into the design workflow, enhancing user experience and accessibility in PCB design.

San Francisco, CaliforniaHeadquarters

2019Year Founded

$11.7MTotal Funding

EARLY_VCCompany Stage

Hardware, Consumer Software, AI & Machine LearningIndustries

51-200Employees

Benefits

Remote Work Options

Risks

Over-reliance on AI designs may lead to quality control issues and vulnerabilities.

New features like Smart Vias may increase complexity, potentially confusing users.

Intense competition from GenAI tools like SnapMagic could divert users from Flux.

Differentiation

Flux offers the first AI-powered hardware design assistant integrated into a PCB tool.

Flux's Copilot provides personalized design recommendations, enhancing user experience and efficiency.

Smart Vias technology simplifies high-density PCB designs, setting Flux apart from competitors.

Upsides

Increased adoption of AI tools boosts demand for Flux's innovative design platform.

Remote work trends align with Flux's cloud-based SaaS model, enhancing collaboration.

Generative AI advancements create opportunities for Flux to lead in hardware design.

Land your dream remote job 3x faster with AI

Try Jobo Free