Senior ML Compiler Engineer

Description

Founded in 2022, the company builds chips and systems that generate model output at thousands of tokens per second, including for very long contexts.

It has raised $220M from investors including Founders Fund and Accel.

The company argues that inference speed, not capability alone, now limits progress in areas such as drug discovery, software engineering and materials discovery.

The team is building a full-stack execution engine across hardware, compiler and runtime, aiming to run the largest language models 100x faster than current systems.

The architecture is still taking shape, so you can help decide its IR, lowering approach, optimisation passes and execution model.

Responsibilities

• Build the machine learning compiler stack for the company's AI accelerator hardware.
• Make the latest large language models run at top performance on these accelerators.
• Collaborate closely with hardware, driver runtime and ML engineers in a hardware-software co-design process.

Requirements

• Would be great: experience developing machine learning compilers with standard tools such as MLIR or TVM.
• Would be great: current knowledge of machine learning techniques and technologies.
• Would be great: a solid grasp of computer architecture and performance optimisation.
• Would be great: strong C/C++ skills and experience with common industry development tools.
• Would be great: a creative, self-driven approach, willing to take ownership and deliver results in a quick-moving environment.
• Would be great: a degree in computer science, electronic engineering, maths, physics or a related field, plus 4+ years of industry experience.
• May also have: experience in a startup or small team.

Conditions

Senior role based in the Bristol office or London.

Work involves technologies subject to UK, US and other international export controls, and some roles may need extra eligibility checks, which the company will explain during hiring.

The company is committed to equal employment opportunity and an inclusive team.

Benefits

• A competitive salary that reflects your experience and the specialist nature of the role.
• Meaningful equity, so everyone shares in the value created.
• Private medical, dental and vision cover, a contributory pension, and life and critical illness insurance.
• 25 days of holiday plus bank holidays.