Head of AI Software

Posted Yesterday
Be an Early Applicant
Location, WV, USA
Hybrid
Senior level
Artificial Intelligence • Hardware • Machine Learning • Semiconductor
The Role
Lead and own the AI application software stack architecture for an optical AI accelerator: design SDK and inference runtime, drive framework integration and model optimisation, collaborate with hardware architects, mentor a growing distributed team, hire and deliver production-quality code.
Summary Generated by Built In
The Opportunity

Lumai is redefining how the world computes. We are an ambitious, venture-backed UK startup pioneering a breakthrough AI accelerator for data centers which uses 3D optical compute. Our radical technology uses light to perform computation at orders of magnitude faster speeds and at far greater scales than ever before, all whilst consuming far less energy than traditional approaches.

Lumai is unlocking performance and efficiency gains that could transform the economics of AI and compute infrastructure and reshape how intelligence scales globally.

If you are passionate about bringing groundbreaking technology to market, and want to be part of a team pushing the boundaries of what is physically possible, Lumai is where you can make it happen.

 
About Lumai

Founded in 2022, Lumai is a University of Oxford spinout using optical processing to accelerate large language models (LLMs) and other transformer-based AI systems. The team combines expertise in optical computing, machine learning, and physics.

Lumai has already secured over $15 million in investment from leading deep-tech investors like Constructor Capital, IP Group, PhotonVentures and government grants, and is scaling rapidly to deploy the fastest optical compute currently available globally.

 
The Role

We are looking for a Head of Software for Lumai's AI Application team, who will set the technical direction for how AI workloads are developed, optimised, and deployed on our accelerator hardware. This is a rare opportunity to shape a team's culture and architecture from day one — you will be both a hands-on engineer and the technical anchor for a distributed team of five (growing to ten within 18 months).

 
What You'll Do
  • Own the end-to-end technical architecture of the AI application software stack, from SDK surface to inference runtime

  • Drive engineering decisions around framework integration, model optimisation strategies, and hardware abstraction layers

  • Conduct design reviews, set coding standards, and mentor engineers across inference, SDK, and tooling workstreams

  • Act as the primary technical liaison between the remote European team, HQ architects, and the simulation team

  • Collaborate directly with the System and HW Architects on SW/HW co-design requirements

  • Lead hiring interviews for the team; contribute to technical roadmap planning with engineering leadership

  • Maintain hands-on delivery — writing and reviewing production-quality code alongside the team

 
What We're Looking For

Must-Have

  • 7+ years of software engineering experience, with at least 2 years in a tech lead or equivalent senior individual contributor role

  • Deep expertise in AI/ML systems — inference pipelines, ML framework internals, model optimisation, or accelerator programming

  • Strong Python and C/C++ skills; comfort working across the full stack from hardware abstraction to developer API

  • Demonstrated ability to lead technically in a remote-first, distributed environment.

  • Excellent cross-functional communicator, capable of working with hardware architects and product stakeholders

  • Bachelor's degree or higher in Computer Science or related discipline, or equivalent experience

Strong Preference For

  • Prior experience at an AI accelerator startup or in a semiconductor SW organisation

  • Background in compiler design, MLIR, or custom hardware backend development

  • Track record of open-source leadership or significant community contribution in ML infrastructure

 
Compensation & Benefits
  • Highly Competitive Salary: We are not saying our salary is a blank check, but let's just say it won't be a source of your stress

  • Share Option Scheme: We are all in this together! We believe in shared success while we build the Lumai of tomorrow

  • L&D Allowance: Stay at the forefront of your field with a €500 annual development budget

  • Holidays: Enjoy some deserved "me time" with 25 days paid holiday (plus bank holidays) per year

  • Socials: Be part of an inclusive community enjoying occasional all-company off-sites, lunches and socials

 
Interview Process

Our process is four stages. An initial conversation with our HR team to understand what you want from the role and what we want from it. Two technical sessions with various members of our engineering team. Finally, an HR-team session covering scope, terms, and any final questions. We aim to move fast on candidates we are excited about; expect roughly three to four weeks end to end.

Lumai is an equal opportunity employer. We make hiring decisions on merit, scope-fit, and the strength of the working relationship we expect to build with each hire. Applications welcome from candidates of any background. If you are not sure whether you are a fit, send a note anyway.

Skills Required

  • 7+ years of software engineering experience
  • At least 2 years in a tech lead or senior individual contributor role
  • Deep expertise in AI/ML systems (inference pipelines, model optimisation, accelerator programming)
  • Strong Python skills
  • Strong C/C++ skills
  • Experience working and leading technically in a remote-first, distributed environment
  • Excellent cross-functional communication with hardware architects and product stakeholders
  • Bachelor's degree or higher in Computer Science or related discipline, or equivalent experience
  • Prior experience at an AI accelerator startup or semiconductor software organisation
  • Background in compiler design, MLIR, or custom hardware backend development
  • Track record of open-source leadership or significant ML infrastructure community contribution
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
0 Employees
Year Founded: 2022

What We Do

Lumai is an optical compute company building the next generation of AI infrastructure for the inference era. By utilizing 3D optical computing, the company develops energy-efficient AI processors that surpass the limitations of silicon-based architectures, delivering significantly higher performance and lower power consumption to unlock sustainable intelligence at scale.

Similar Jobs

PNC Bank Logo PNC Bank

Software Engineer

Machine Learning • Payments • Security • Software • Financial Services
Remote or Hybrid
USA
55000 Employees

PNC Bank Logo PNC Bank

Detection and Response Manager, Tempus Technologies

Machine Learning • Payments • Security • Software • Financial Services
Remote or Hybrid
USA
55000 Employees
100K-223K Annually

Enverus Logo Enverus

Account Director

Big Data • Information Technology • Software • Analytics • Energy
In-Office or Remote
2 Locations
1800 Employees

MetLife Logo MetLife

Customer Care Advocate AMS Service - Omaha, NE 9.21.26 - 18275

Fintech • Information Technology • Insurance • Financial Services • Big Data Analytics
Remote or Hybrid
United States
43000 Employees
42K-42K Annually

Similar Companies Hiring

Fairly Even Thumbnail
Hardware • Robotics • Sales • Software • Hospitality
New York, NY
30 Employees
Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account