Senior Compiler Engineer at Flux

San Francisco, California, United States

Apply Now

£174,000 – £234,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

C++

PyTorch

TensorFlow

JAX

MXNet

LLVM

MLIR

TVM

XLA

compilers

runtime

debugger

ML optimization

low-level programming

performance tuning

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