Staff Compiler Engineer at Flux

San Francisco, California, United States

Apply Now

£208,000 – £254,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 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 the 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
Architect and build the ML backend (compiler, run-time, and debugger) for next-generation OPTUs
Own integration with PyTorch, Tensorflow, JAX, and MXNet down to low-level kernel drivers
Create seamless support for a broad ecosystem of large AI models
Ensure pushing the limits of performance envelope by partnering with hardware and modelling teams to understand device trade-offs

Skills

C++

PyTorch

TensorFlow

JAX

MXNet

LLVM

MLIR

TVM

OpenXLA

XLA

compilers

runtime

debugger

low-level programming

optimization

ML backend

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