ABOUT LUMAI
Lumai is an Oxford-based start-up creating the next generation of hardware for AI acceleration using 3D optical computing – a technology that uses light to compute at much faster speeds, higher energy efficiency and at far greater scales than ever before, all whilst consuming far less energy than traditional electronics.
Join us to become part of a multidisciplinary team spanning optics, electronics, software and AI/ML algorithms, and grow with us as we become world leaders in this exciting new era of computing!
JOB DESCRIPTION
Lumai is seeking a Machine Learning Engineer to join our team in a highly influential role. You will be leading the development and optimisation of AI models on the world’s most advanced optical AI processing cluster. Working cross-functionally, you will help to integrate benchmarking and emulation tools into our product development workflow and offerings.
ROLE RESPONSIBILITIES
● Research and implement large-scale, state-of-the-art machine learning algorithms on the optical AI processor.
● Perform comprehensive model optimisation, including quantisation, model compression, memory footprint reduction.
● Collaborate with other software engineers to help develop an accurate emulator of the optical AI processor.
● Collaborate with hardware engineers to improve the optical processor architecture and optimise AI model implementations accordingly.
REQUIREMENTS
● Strong proficiency in machine learning techniques and frameworks such as TensorFlow and PyTorch.
● Experience in model optimisation of AI models, preferably on specialized hardware platforms (NPU/ASIC/FPGA).
● Experience in developing simulation or performance analysis software is a plus.
● A degree in computer science, engineering, mathematics, a science or a related field.
● Excellent communication skills, including the ability to present results orally and in writing.
● Creativity and excellent problem-solving skills. Ability to work independently and as part of a team.
REMUNERATION
Competitive remuneration package across salary, share options and benefits.
HOW TO APPLY
If you are passionate about creating new technology, please apply by submitting your CV along with a cover letter outlining how your skills and experience will bring value to Lumai. Applications, informal inquiries and requests for further information can be directed to careers@lumai.co.uk.
At Lumai, we are passionate about creating a diverse, inclusive, and flexible team. We are committed to providing equal employment opportunities and we welcome applications from people of all ethnicities, cultures, ages, genders, sexual orientation or living with a disability.