Senior Compiler Engineer at Flux

San Francisco, California, United States

Apply Now

$230,000 – $282,000Compensation

Senior (5 to 8 years)Experience Level

Full TimeJob Type

UnknownVisa

AI, Machine Learning, Hardware, High-Performance ComputingIndustries

Requirements

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

Responsibilities

Own and deliver projects on your team's roadmap that unlock key high-impact technical and business milestones that drive the success of Flux
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

Skills

Key technologies and capabilities for this role

C++PyTorchTensorFlowJAXMXNetLLVMMLIRTVMOpenXLAXLAcompilersruntimedebuggeroptimizationfusiontilingscheduling

Questions & Answers

Common questions about this position

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

The salary range is $230K - $282K, with a starting salary of $230,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?

Required skills include 5+ years in software engineering with C/C++ focus, deep expertise in ML framework internals, compilers, low-level programming, optimization techniques, optimizing TensorFlow/PyTorch/JAX models, and toolchains like LLVM, OpenXLA/XLA, MLIR, TVM.

What is the company culture like at Flux?

Flux fosters a high-growth environment that emphasizes collaboration, requiring onsite work, and values teamwork, communication, and working in high-pace, dynamic settings.

What makes a strong candidate stand out for this position?

Personal projects are a key differentiating factor and hold more weight than other requirements, along with a Bachelor's degree in computer science or related field.

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