Research Engineer, Latency & Throughput

Mistral

Greater London

Hybrid

GBP 90,000 - 130,000

Full time

6 days ago
Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

Mistral is seeking a Research Engineer for the Latency & Throughput team to push the frontier of LLM inference. You will explore speculative decoding, quantization, and KV compression, turning research ideas into production-ready improvements within our vLLM-based serving stack.

You will work closely with inference engineers and researchers, running rigorous experiments, and contributing to design choices that balance latency gains with model quality.

Qualifications

  • Strong ML background with LLM training concepts (architecture, tokenization, fine-tuning)
  • Hands-on Python and PyTorch; able to write clean, experimental code
  • Experience reading literature and designing controlled experiments
  • Track record of hypothesis testing and drawing conclusions from results

Responsibilities

  • Research and prototype inference techniques (speculative decoding, quantization, KV compression) to reduce latency
  • Design experiments to validate ideas and demonstrate latency/throughput gains
  • Collaborate with Inference Foundation to integrate techniques into the serving stack
  • Coordinate with pretraining/RL teams for co-training or retraining when needed

Skills

Python & PyTorch
JAX familiarity
CUDA basics (GPU)
Research experience

Tools

vLLM serving stack
Experimentation tooling

Job description

About Mistral

Mistral provides full-stack AI solutions: from frontier models to developer tools, applications, and compute. We partner with enterprises tackling the hardest problems across high-stakes industries like finance, manufacturing, defense, healthcare, and the public sector, co-creating customized AI systems that they can run on their terms.

About Mistral

Mistral provides full-stack AI solutions: from frontier models to developer tools, applications, and compute. We partner with enterprises tackling the hardest problems across high-stakes industries like finance, manufacturing, defense, healthcare, and the public sector, co-creating customized AI systems that they can run on their terms. We are a dynamic, collaborative team passionate about AI and its potential to transform society. Our diverse workforce thrives in competitive environments and is committed to driving innovation. Our teams are distributed between Europe, North America, Asia and the Middle East. We are creative, low-ego and team-spirited.

The Role

As a Research Engineer on the Latency & Throughput team, you will push the frontier of LLM inference by researching and prototyping techniques that reduce latency and increase throughput — from speculative decoding and quantization to KV compression and new attention variants. The team sits at the intersection of research and production: you will take ideas from recent literature, form hypotheses, run experiments, and ship the techniques that prove out into our serving stack. This role sits between our inference engineers, who own the production serving stack, and our research scientists, who focus on paper-level contributions. You will be more science-focused than a pure engineer: you are expected to generate new ideas, keep up with the state of the art, and implement proof‑of‑concepts that bridge the gap between academic results and production impact. Some techniques you explore will be lossy or require integration into training — retraining models, tuning architectures, or training auxiliary models to improve inference. You will work closely with the Inference Foundation and Kernels teams, who turn your validated techniques into production recipes, and with pretraining and RL teams when your work touches model training. Success is measured by demonstrated latency or throughput gains, first at small scale, then adopted in production.

What You Will Do
  • Research and prototype inference techniques — speculative decoding, quantization, KV compression, watermarking, and new model architectures — by reading recent literature and forming testable hypotheses.
  • Design and run experiments to validate or refute ideas, iterating quickly on small‑scale setups before proposing production integration. Demonstrate clear latency or throughput gains without unacceptable quality loss.
  • Collaborate with the Inference Foundation team to integrate validated techniques into the vLLM‑based serving stack.
  • Partner with pretraining and RL teams when techniques require co‑training, retraining, or training auxiliary models to improve inference.

Build the right tooling and validation platforms to facilitate AI‑driven inference research.

What We’re Looking For
  • A strong ML background with solid knowledge of LLM training techniques — architecture, tokenization, fine‑tuning, and evaluation.
  • Hands‑on experience with Python and PyTorch; you write clean, experimental code and iterate fast. JAX is acceptable with strong Python proficiency. C++ and Rust are not required, CUDA and GPU knowledge is expected but hands‑on experience is not required.
  • Research experience of some form: a thesis, publications, a dedicated research project, or efficiency‑oriented work in a lab or industry research setting. A PhD is not required, but we are looking for genuine research depth.
  • A track record of conducting experiments: forming hypotheses, designing controlled comparisons, and drawing clear conclusions from results.
  • Familiarity with the state of the art in LLM inference — you know what has been tried, what works, and where the open problems are.
  • Self‑starter, low‑ego, and collaborative. You are comfortable with ambiguity, negative results, and are eager to take matters into your own hands.
  • Inference or serving experience is a plus, not a requirement. Prior work with vLLM, production LLM deployments, or serving systems will help you ramp up faster.
Location and Remote

This role is based in our Paris or London offices (hybrid). We will prioritize candidates who reside in the Paris area or are open to relocating. We strongly believe in the value of in‑person collaboration to foster strong relationships and seamless communication within our team. For senior profiles, we will also consider remote candidates based in France. In that case, we ask all new hires to visit our Paris office: For the first week of their onboarding (accommodation and travel covered) Then at least 3 days per month.

What we Offer

We offer a comprehensive benefits package designed to support your well‑being, growth, and work‑life balance. Benefits vary by country and may include healthcare coverage, parental leave, retirement plans, relocation support, wellness programs, meal and transportation allowances, and other location‑specific perks. For the most up‑to‑date details on benefits available in your location, please refer to our Benefits page.

Privacy Policy

Your privacy matters to us. You can learn more about how we handle your personal data in our Applicant Privacy Policy.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Research Engineer, Latency & Throughput
Research Engineer, Latency & Throughput

Lindus Health Limited • Greater London

Hybrid
GBP 70,000 - 110,000
Research Engineer, Inference Foundation
Research Engineer, Inference Foundation

Mistral AI • Greater London

Hybrid
GBP 120,000 - 180,000
Healthcare coverage
Relocation support
Wellness programs
Research Engineer, Inference Foundation
Research Engineer, Inference Foundation

Mistral • Greater London

Hybrid
GBP 120,000 - 160,000
Research Engineer, Machine Learning
Research Engineer, Machine Learning

Mistral AI • Greater London

On-site
GBP 120,000 - 180,000
Healthcare coverage
Relocation support
Wellness programs
AI Scientist
AI Scientist

Mistral AI • Greater London

On-site
GBP 110,000 - 170,000
Healthcare coverage
Relocation support
Wellness programs
Research Engineer, Machine Learning
Research Engineer, Machine Learning

Mistral • Greater London

On-site
GBP 100,000 - 140,000
Healthcare coverage
Parental leave
Retirement plans
+3
Research Engineer, Data Infrastructure
Research Engineer, Data Infrastructure

Mistral • Greater London

Hybrid
GBP 76,673 - 110,750
Healthcare coverage
Relocation support
Wellness programs
Research Software Engineer
Research Software Engineer

Mistral • Greater London

Hybrid
GBP 90,000 - 120,000
Applied AI, Forward Deployed Machine Learning Engineer - London
Applied AI, Forward Deployed Machine Learning Engineer - London

Mistral • Greater London

On-site
GBP 70,000 - 120,000
Healthcare coverage
Parental leave
Relocation support
+3
Applied AI, Forward Deployed Machine Learning Engineer - London
Applied AI, Forward Deployed Machine Learning Engineer - London

Mistral.ai • Greater London

On-site
GBP 70,000 - 100,000