Senior / Staff / Principal Software Engineer

Company Overview

Flux is pioneering a new class of AI accelerators called Optical Tensor Processing Units (OTPUs). We’ve already developed functioning prototypes and are now scaling our operations in London. Our work environment rewards innovation, speed, and bold thinking.

The role

We are seeking highly experienced and motivated Software Engineers to design and build the software architecture for our next-generation OTPUs. This role demands deep expertise in C and C++ programming, low-level programming, compiler construction, and optimisation techniques. The ideal candidate will have a strong background in computer science, electrical engineering, telecoms engineering, mathematics, or a related field, combined with significant experience in machine learning and a passion for high-pace environments. You’ll integrate deeply with the hardware team to ensure smooth interaction between compiler software and the optical hardware. You’ll also help shape our machine learning tools so they run efficiently on the OTPU.

Responsibilities

• Design and implement the software and compiler frameworks that allow AI models to run optimally on our optical hardware.
• Identify bottlenecks in performance and apply advanced compiler techniques, such as code generation, scheduling, and vectorisation.
• Work closely with hardware engineers to align compiler requirements with the OTPU design; ensure software can fully exploit hardware capabilities.
• Extend and maintain relevant machine learning libraries so users can easily leverage OTPU advantages.
• Conduct code reviews, mentor team members, and drive best practices around compiler construction and performance tuning.
• Keep abreast of industry trends in GPU, AI, and optical computing; contribute ideas for future directions and improvements.

Skills & Experience

• 5+ years of experience in software engineering with a focus on C/C++ programming.
• Extensive experience in compilers, low-level programming, and optimisation techniques.
• Proven expertise in machine learning and its applications in high-performance computing.
• Strong problem-solving skills and the ability to think critically and creatively.
• Experience in high-pace, dynamic work environments.
• Bachelor's degree in computer science, electrical engineering, telecoms engineering, mathematics, or a related field.
• Excellent teamwork and communication skills, with the ability to collaborate effectively with cross-functional teams.
• Personal projects are a key differentiating factor and hold more weight than other requirements.

Details

• Competitive salary ranging from £145k+, depending on experience.
• Stock options in a rapidly growing AI company.
• Comprehensive healthcare insurance.
• 25 days PTO policy plus bank holidays.
• Based in our new 5,000 square foot office in the AI hub of Kings Cross, London.
• Bonus additional salary of £12,000 per year if you’re based within a 20-minute commute of the office.
• Private use of our 3D printer.

If you’re passionate about compilers, high-performance computing, and redefining what’s possible in AI, we’d love to talk. Apply now to join Flux and help shape the future of optical computing.