Flux Computing builds optical processors to speed up AI training and inference with lower energy use. They develop Optical Tensor Processing Units and other optical chips that rely on light-based circuits, integrated with supporting electronics and control software, and connected to AI frameworks. This positions them as a hardware provider for data centers and research environments handling large models. Category: B2B hardware for AI workloads; integrates with common AI frameworks and data-center tooling. The company targets scalable deployments in enterprises and labs pursuing advanced AI research and production workloads.
Flux Computing builds optical processors to speed up AI training and inference with lower energy use. They develop Optical Tensor Processing Units and other optical chips that rely on light-based circuits, integrated with supporting electronics and control software, and connected to AI frameworks. This positions them as a hardware provider for data centers and research environments handling large models. Category: B2B hardware for AI workloads; integrates with common AI frameworks and data-center tooling. The company targets scalable deployments in enterprises and labs pursuing advanced AI research and production workloads.
Teeming tracks opportunities at over 24,000 AI startups, then works with you to find (and land) the one you'll love.
Data Scientist
ContractBelgrade, RS
Contract • Belgrade, RS
Data Scientist
InternshipJerusalem
Internship • Jerusalem
Product Designer
Full-timeNew York, US
Full-time • New York, US
Machine Learning Engineer
Part-timeTel Aviv
Part-time • Tel Aviv
DevOps Engineer
Part-timeUtrecht, NL
Part-time • Utrecht, NL
Backend Developer
Full-timeNovi Sad, RS
Full-time • Novi Sad, RS
The Role
We’re searching for Staff Compiler Engineers to architect and build the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs. You will own integration with PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI models, and ensure we are pushing the limits of their performance envelope by closely partnering with hardware and modelling teams to understand device trade-offs.
Responsibilities
Ownership: Define and deliver the technical vision and roadmap for your team that unlocks key strategic technical and business goals that are essential to the success of Flux.
Collaboration: 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.
Architect & Build: Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware.
Optimize Performance: Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximize throughput.
Mentor & Define Standards: Lead code reviews, coach peers, and define best practices in ML backend and performance engineering for your team.
Guide Technology Direction: Stay ahead of GPU, AI accelerator, and optical computing trends; propose and prototype innovations.
Skills & Experience
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.
Compensation & Benefits
Due to U.S. export control regulations, candidates’ eligibility to work at Flux depends on their most recent citizenship or permanent residency status. We are generally unable to consider applicants whose most recent citizenship or permanent residence is in certain restricted countries (currently including Iran, North Korea, Syria, Cuba, Russia, Belarus, China, Hong Kong, Macau, and Venezuela). Applicants who have subsequently obtained citizenship or permanent residency in another country not subject to these restrictions may still be eligible.
We do not accept unsolicited CVs from recruitment agencies, will not be liable for any fees, and prohibit unauthorised use of our company name in recruitment activities.
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 and hold more weight than other requirements.
Starting salary: £208,000+, depending on experience, skills, and location.
Competitive salary and stock options, you’re not just part of the journey, you will own a piece of it.
Work from our HQ in King’s Cross, right in the middle of London’s buzzing AI scene.
Live within 45 minutes of the office? Perfect. Live within 20 minutes? We’ll add an extra location bonus to your salary.
We offer visa sponsorship and full relocation support (UK and abroad), through a dedicated third-party provider who are on hand to make your move to London as seamless as possible.
Full BUPA healthcare and dental cover, medical history disregarded.
High-spec tech for everyone - M4 Macs as standard, M4 Pros for Engineers.
Sony noise-cancelling headphones and ergonomic setups to keep you comfortable and focused.
Personal company card to spend on tools that help you do your job - like ChatGPT Pro or anything else that boosts your workflow.
Healthy, chef-cooked dinners in the office every night, with something for every diet and tastebud.
Monthly off-site team socials.
25 days of paid holiday, plus all the UK bank holidays.
Access to our in-house 3D printer for personal or work projects.
Cycle2work scheme.
Need a caffeine fix? We’ve got you covered with a tab at our favourite local coffee shop.
We offer a pension plan and salary sacrifice options.
