About the job

Flux Computing is seeking a highly skilled Staff Compiler Engineer to architect and build the ML backend for next-generation optical processors. This role involves ownership of compiler, runtime, and debugger integration with PyTorch, TensorFlow, JAX, and MXNet down to low-level kernel drivers. Responsibilities include designing high-speed digital/mixed signal systems, collaborating closely with hardware teams, optimizing ML model performance, mentoring peers, and guiding technology direction in optical computing. Candidates should have 7+ years of experience in software engineering with C/C++, deep expertise in ML frameworks, compilers, low-level programming, optimization techniques, and practical high-performance ML experience. A Bachelor’s in computer science, electrical engineering, telecoms, mathematics, or related field is required, with personal projects considered a key differentiator. The position includes competitive salary, stock options, relocation support, visa sponsorship for the UK, full healthcare, ergonomic tech setup, 25 days paid holiday, and other benefits. Eligibility is subject to U.S. export control regulations for citizenship/residency. This is a high-impact tech role in a fast-paced AI and computing environment.