On-Device AI Research Engineer — Edge ML & Multimodal LLMs

Plumerai; City Of London; Hybrid; GBP 60,000 - 80,000; Full time

Job summary

A cutting-edge AI company in London is seeking an AI Research Engineer to develop state-of-the-art AI products. The role involves optimizing deep learning models and integrating multimodal LLMs for embedded devices. Candidates should have strong software engineering skills and experience with Python, as well as familiarity with frameworks like PyTorch and TensorFlow. The position offers competitive salary, equity stake, and flexible working hours.

Benefits

Competitive salary

Generous equity stake

Relocation assistance

Choose your own laptop

25 days of paid vacation

Attend top research conferences

Qualifications

• Very strong software engineering skills and proficiency in Python.
• Comfortable with frameworks such as PyTorch, TensorFlow, Keras, or JAX.
• Strong experience with computer vision and multimodal LLMs.
• Experience working in a team on the same software project.

Responsibilities

• Combine Tiny AI with multimodal LLMs to enable advanced AI features.
• Train and design accurate models for low-cost hardware.
• Improve the data pipeline, model architectures, and training software.
• Deploy PyTorch and TensorFlow training jobs using Kubernetes.

Skills

Strong software engineering skills

Proficiency in Python

Comfortable with PyTorch or TensorFlow

Experience with computer vision

Tools

PyTorch

TensorFlow

Kubernetes

Job description

A cutting-edge AI company in London is seeking an AI Research Engineer to develop state-of-the-art AI products. The role involves optimizing deep learning models and integrating multimodal LLMs for embedded devices. Candidates should have strong software engineering skills and experience with Python, as well as familiarity with frameworks like PyTorch and TensorFlow. The position offers competitive salary, equity stake, and flexible working hours.